wip add batch update

This commit is contained in:
Rizky 2024-03-13 15:04:49 +07:00
parent 95b75d44eb
commit 4ac633d4a2
2 changed files with 99 additions and 52 deletions

BIN
bun.lockb

Binary file not shown.

View File

@ -13,6 +13,35 @@ export type DBArg = {
export const execQuery = async (args: DBArg, prisma: any) => {
const { table, action, params } = args;
if (action === "batch_update") {
const { table, batch } = params as unknown as {
table: string;
batch: { data: any; where: any }[];
};
const promises = [] as any[];
const tableInstance = prisma[table];
if (tableInstance) {
try {
for (const item of batch) {
if (
Object.entries(item.where).length > 0 &&
Object.entries(item.data).length > 0
) {
promises.push(
tableInstance.updateMany({ where: item.where, data: item.data })
);
}
}
await Promise.all(promises);
} catch (e: any) {
throw new Error(e.message);
}
}
return;
}
if (action.startsWith("schema_")) {
const schema_path = dir("app/db/prisma/schema.prisma");
const schema = createPrismaSchemaBuilder(await readAsync(schema_path));
@ -28,73 +57,85 @@ export const execQuery = async (args: DBArg, prisma: any) => {
type: string;
optional: boolean;
db_type: string;
default?: any
default?: any;
}
>;
const rels = {} as Record<string, {
type: 'has-many',
to: { table: string, fields: string[] }
from: { table: string, fields: string[] }
} | {
type: 'has-one',
to: { table: string, fields: string[] }
from: { table: string, fields: string[] }
}>
const rels = {} as Record<
string,
| {
type: "has-many";
to: { table: string; fields: string[] };
from: { table: string; fields: string[] };
}
| {
type: "has-one";
to: { table: string; fields: string[] };
from: { table: string; fields: string[] };
}
>;
if (schema_table) {
if (action === "schema_rels") {
for (const col of schema_table.properties) {
if (
col.type === "field" &&
(!!col.array ||
col.attributes &&
col.attributes?.length > 0)
(!!col.array || (col.attributes && col.attributes?.length > 0))
) {
if (col.array) {
if (typeof col.fieldType === 'string') {
const target = schema.findByType('model', { name: col.fieldType })
if (typeof col.fieldType === "string") {
const target = schema.findByType("model", {
name: col.fieldType,
});
if (target) {
const field = target.properties.find(e => {
if (e.type === 'field'
&& e.fieldType === table) {
const field = target.properties.find((e) => {
if (e.type === "field" && e.fieldType === table) {
return true;
}
});
if (field && field.type === 'field') {
const rel = field.attributes?.find(e => e.kind === 'field');
if (field && field.type === "field") {
const rel = field.attributes?.find(
(e) => e.kind === "field"
);
if (rel && rel.args) {
const { field, ref } = getFieldAndRef(rel, target, table);
const { field, ref } = getFieldAndRef(
rel,
target,
table
);
if (target && ref) {
rels[col.name] = {
type: 'has-many',
type: "has-many",
to: field,
from: ref
}
from: ref,
};
}
}
}
}
}
} else if (col.attributes) {
const rel = col.attributes.find(e => e.type === 'attribute' && e.name === 'relation');
if (rel && typeof col.fieldType === 'string') {
const target = schema.findByType('model', { name: col.fieldType });
const rel = col.attributes.find(
(e) => e.type === "attribute" && e.name === "relation"
);
if (rel && typeof col.fieldType === "string") {
const target = schema.findByType("model", {
name: col.fieldType,
});
const { field, ref } = getFieldAndRef(rel, target, table);
rels[col.name] = {
type: 'has-one',
type: "has-one",
to: {
table: field.table,
fields: ref.fields
fields: ref.fields,
},
from: {
table: ref.table,
fields: field.fields
}
}
fields: field.fields,
},
};
}
}
}
@ -108,10 +149,14 @@ export const execQuery = async (args: DBArg, prisma: any) => {
col.attributes &&
col.attributes?.length > 0
) {
const attr = col.attributes.find(e => e.name !== 'id' && e.name !== 'default');
const attr = col.attributes.find(
(e) => e.name !== "id" && e.name !== "default"
);
const default_val = col.attributes.find(e => e.name === 'default');
const is_pk = col.attributes.find(e => e.name === 'id');
const default_val = col.attributes.find(
(e) => e.name === "default"
);
const is_pk = col.attributes.find((e) => e.name === "id");
if (attr && attr.name !== "relation") {
let type = "String";
@ -122,7 +167,7 @@ export const execQuery = async (args: DBArg, prisma: any) => {
type: type.toLowerCase(),
optional: !!col.optional,
db_type: attr.name.toLowerCase(),
default: default_val
default: default_val,
};
}
}
@ -165,27 +210,29 @@ export const execQuery = async (args: DBArg, prisma: any) => {
};
const getFieldAndRef = (rel: any, target: any, table: string) => {
let field = null as unknown as { table: string, fields: string[] };
let ref = null as unknown as { table: string, fields: string[] };
let field = null as unknown as { table: string; fields: string[] };
let ref = null as unknown as { table: string; fields: string[] };
for (const e of rel.args) {
if (typeof e.value === 'object'
&& !Array.isArray(e.value)
&& e.value.type === 'keyValue'
&& typeof e.value.value === 'object'
&& !Array.isArray(e.value.value)
&& e.value.value.type === 'array') {
if (e.value.key === 'fields') {
if (
typeof e.value === "object" &&
!Array.isArray(e.value) &&
e.value.type === "keyValue" &&
typeof e.value.value === "object" &&
!Array.isArray(e.value.value) &&
e.value.value.type === "array"
) {
if (e.value.key === "fields") {
field = {
table: target.name,
fields: e.value.value.args
}
} else if (e.value.key === 'references') {
fields: e.value.value.args,
};
} else if (e.value.key === "references") {
ref = {
table: table,
fields: e.value.value.args
}
fields: e.value.value.args,
};
}
}
}
return { field, ref };
}
};