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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
// src/lib/tasks/repository.ts
import db from "@/db/db";
import { ocrRows, users } 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({
// ocrRows의 모든 필드
id: ocrRows.id,
tableId: ocrRows.tableId,
sessionId: ocrRows.sessionId,
rowIndex: ocrRows.rowIndex,
reportNo: ocrRows.reportNo,
no: ocrRows.no,
identificationNo: ocrRows.identificationNo,
tagNo: ocrRows.tagNo,
jointNo: ocrRows.jointNo,
jointType: ocrRows.jointType,
weldingDate: ocrRows.weldingDate,
confidence: ocrRows.confidence,
sourceTable: ocrRows.sourceTable,
sourceRow: ocrRows.sourceRow,
userId: ocrRows.userId,
createdAt: ocrRows.createdAt,
// users 테이블의 필드
userName: users.name,
userEmail: users.email,
})
.from(ocrRows)
.leftJoin(users, eq(ocrRows.userId, users.id))
.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;
}
|
