midsuit
/
prasi-api
6
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix batch upsert no return deleted

This commit is contained in:
Rizky 2024-08-12 11:12:42 +07:00
parent 78a4141dc8
commit 700cc0887c
1 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
pkgs/utils/query.ts
View File

@ -29,6 +29,7 @@ export const execQuery = async (args: DBArg, prisma: any) => {
const mode = arg.mode || "field"; const mode = arg.mode || "field";
if (table && where && data) { if (table && where && data) {
const transactions = []; const transactions = [];
const tx_delete_index = new Set<number>();
const schema_path = dir("app/db/prisma/schema.prisma"); const schema_path = dir("app/db/prisma/schema.prisma");
const schema = createPrismaSchemaBuilder(await readAsync(schema_path)); const schema = createPrismaSchemaBuilder(await readAsync(schema_path));
@ -88,7 +89,7 @@ export const execQuery = async (args: DBArg, prisma: any) => {
>; >;
} }
>; >;
if (mode !== "raw") { if (mode !== "raw") {
for (const row of data) { for (const row of data) {
for (const [k, v] of Object.entries(row) as any) { for (const [k, v] of Object.entries(row) as any) {
@ -332,16 +333,26 @@ export const execQuery = async (args: DBArg, prisma: any) => {
} }
transactions.push(prisma[table].delete({ where })); transactions.push(prisma[table].delete({ where }));
tx_delete_index.add(transactions.length - 1);
} }
} }
const result = await prisma.$transaction(transactions); const raw_result = await prisma.$transaction(transactions);
const result: any = [];
if (Object.keys(marker).length > 0) { if (Object.keys(marker).length > 0) {
for (const [k, v] of Object.entries(marker)) { for (const [k, v] of Object.entries(marker)) {
result[k]._marker = v; raw_result[k]._marker = v;
} }
} }
for (const [k, v] of Object.entries(raw_result)) {
if (tx_delete_index.has(parseInt(k))) {
continue;
}
result.push(v);
}
return result; return result;
} }
} }