// src/lib/items/repository.ts import db from "@/db/db"; import { Item, items } from "@/db/schema/items"; import { eq, inArray, not, asc, desc, and, ilike, gte, lte, count, gt, } from "drizzle-orm"; import { PgTransaction } from "drizzle-orm/pg-core"; export type NewItem = typeof items.$inferInsert /** * 단건/복수 조회 시 공통으로 사용 가능한 SELECT 함수 예시 * - 트랜잭션(tx)을 받아서 사용하도록 구현 */ export async function selectItems( tx: PgTransaction, params: { where?: any; // drizzle-orm의 조건식 (and, eq...) 등 orderBy?: (ReturnType | ReturnType)[]; offset?: number; limit?: number; } ) { const { where, orderBy, offset = 0, limit = 10 } = params; return tx .select() .from(items) .where(where) .orderBy(...(orderBy ?? [])) .offset(offset) .limit(limit); } /** 총 개수 count */ export async function countItems( tx: PgTransaction, where?: any ) { const res = await tx.select({ count: count() }).from(items).where(where); return res[0]?.count ?? 0; } /** 단건 Insert 예시 */ export async function insertItem( tx: PgTransaction, data: NewItem // DB와 동일한 insert 가능한 타입 ) { // returning() 사용 시 배열로 돌아오므로 [0]만 리턴 return tx .insert(items) .values(data) .returning({ id: items.id, createdAt: items.createdAt }); } /** 복수 Insert 예시 */ export async function insertItems( tx: PgTransaction, data: Item[] ) { return tx.insert(items).values(data).onConflictDoNothing(); } /** 단건 삭제 */ export async function deleteItemById( tx: PgTransaction, itemId: number ) { return tx.delete(items).where(eq(items.id, itemId)); } /** 복수 삭제 */ export async function deleteItemsByIds( tx: PgTransaction, ids: number[] ) { return tx.delete(items).where(inArray(items.id, ids)); } /** 전체 삭제 */ export async function deleteAllItems( tx: PgTransaction, ) { return tx.delete(items); } /** 단건 업데이트 */ export async function updateItem( tx: PgTransaction, itemId: number, data: Partial ) { return tx .update(items) .set(data) .where(eq(items.id, itemId)) .returning({ id: items.id, createdAt: items.createdAt }); } /** 복수 업데이트 */ export async function updateItems( tx: PgTransaction, ids: number[], data: Partial ) { return tx .update(items) .set(data) .where(inArray(items.id, ids)) .returning({ id: items.id, createdAt: items.createdAt }); } export async function findAllItems(): Promise { return db.select().from(items).orderBy(asc(items.itemCode)); }