summaryrefslogtreecommitdiff
path: root/lib/welding/repository.ts
blob: 10e64f58c7da9f09e92bac80f4566112cd58d0e2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
// src/lib/tasks/repository.ts
import db  from "@/db/db";
import { ocrRows } from "@/db/schema";
import {
  eq,
  inArray,
  not,
  asc,
  desc,
  and,
  ilike,
  gte,
  lte,
  count,
  gt,
} from "drizzle-orm";
import { PgTransaction } from "drizzle-orm/pg-core";

/**
 * 단건/복수 조회 시 공통으로 사용 가능한 SELECT 함수 예시
 *  - 트랜잭션(tx)을 받아서 사용하도록 구현
 */
export async function selectOcrRows(
    tx: PgTransaction<any, any, any>,
    params: {
      where?: any; // drizzle-orm의 조건식 (and, eq...) 등
      orderBy?: (ReturnType<typeof asc> | ReturnType<typeof desc>)[];
      offset?: number;
      limit?: number;
    }
  ) {
    const { where, orderBy, offset = 0, limit = 10 } = params;
  
    return tx
      .select()
      .from(ocrRows)
      .where(where)
      .orderBy(...(orderBy ?? []))
      .offset(offset)
      .limit(limit);
  }
/** 총 개수 count */
export async function countOcrRows(
  tx: PgTransaction<any, any, any>,
  where?: any
) {
  const res = await tx.select({ count: count() }).from(ocrRows).where(where);
  return res[0]?.count ?? 0;
}