fix rel many
This commit is contained in:
Rizky
parent
2b6c39c550
commit
9d9792c56c
|
|
@ -2,7 +2,7 @@ import { createPrismaSchemaBuilder, Field } from "@mrleebo/prisma-ast";
|
||||||
import { get } from "utils/get";
|
import { get } from "utils/get";
|
||||||
import { HasManyType } from "./types";
|
import { HasManyType } from "./types";
|
||||||
|
|
||||||
export const upsertRelMany = (arg: {
|
export const createRelMany = (arg: {
|
||||||
row: any;
|
row: any;
|
||||||
k: string;
|
k: string;
|
||||||
schema: ReturnType<typeof createPrismaSchemaBuilder>;
|
schema: ReturnType<typeof createPrismaSchemaBuilder>;
|
||||||
|
|
@ -57,6 +57,6 @@ export const upsertRelMany = (arg: {
|
||||||
}
|
}
|
||||||
return item;
|
return item;
|
||||||
});
|
});
|
||||||
row[k] = { createMany: { data: create_many } };
|
return create_many
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
@ -3,7 +3,7 @@ import { readAsync } from "fs-jetpack";
|
||||||
import { HasManyType, HasOneType } from "./db/types";
|
import { HasManyType, HasOneType } from "./db/types";
|
||||||
import { dir } from "./dir";
|
import { dir } from "./dir";
|
||||||
import { gunzipAsync } from "./gzip";
|
import { gunzipAsync } from "./gzip";
|
||||||
import { upsertRelMany } from "./db/upsert-rel-many";
|
import { createRelMany } from "./db/create-rel-many";
|
||||||
export type DBArg = {
|
export type DBArg = {
|
||||||
db: string;
|
db: string;
|
||||||
table: string;
|
table: string;
|
||||||
|
|
@ -52,10 +52,109 @@ export const execQuery = async (args: DBArg, prisma: any) => {
|
||||||
const rels = getRels({ schema_table, schema, table, tables });
|
const rels = getRels({ schema_table, schema, table, tables });
|
||||||
if (pks.length > 0) {
|
if (pks.length > 0) {
|
||||||
if (Object.keys(where.length > 0)) {
|
if (Object.keys(where.length > 0)) {
|
||||||
|
const rel_many = {} as Record<
|
||||||
|
string,
|
||||||
|
{
|
||||||
|
to: string;
|
||||||
|
from: string;
|
||||||
|
pk: string[];
|
||||||
|
select: Set<string>;
|
||||||
|
ops: Map<
|
||||||
|
any,
|
||||||
|
{
|
||||||
|
delete: Set<any>;
|
||||||
|
insert: Set<any>;
|
||||||
|
tobe: any[];
|
||||||
|
}
|
||||||
|
>;
|
||||||
|
}
|
||||||
|
>;
|
||||||
|
|
||||||
|
for (const row of data) {
|
||||||
|
for (const [k, v] of Object.entries(row) as any) {
|
||||||
|
const rel = rels[k];
|
||||||
|
if (rel) {
|
||||||
|
if (rel.type === "has-one") {
|
||||||
|
const to = rel.to.fields[0];
|
||||||
|
if (!v.connect && v[to]) {
|
||||||
|
let newv = { connect: { [to]: v[to] } };
|
||||||
|
row[k] = newv;
|
||||||
|
}
|
||||||
|
} else if (rel.type === "has-many") {
|
||||||
|
if (!rel_many[k]) {
|
||||||
|
const schema_table = schema.findByType("model", {
|
||||||
|
name: rel.to.table,
|
||||||
|
});
|
||||||
|
|
||||||
|
let pks: Property[] = [];
|
||||||
|
if (schema_table) {
|
||||||
|
for (const col of schema_table.properties) {
|
||||||
|
if (col.type === "field" && !col.array) {
|
||||||
|
if (
|
||||||
|
col.attributes &&
|
||||||
|
col.attributes?.length > 0
|
||||||
|
) {
|
||||||
|
const is_pk = col.attributes.find(
|
||||||
|
(e) => e.name === "id"
|
||||||
|
);
|
||||||
|
if (is_pk) {
|
||||||
|
pks.push(col);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
rel_many[k] = {
|
||||||
|
to: rel.to.table,
|
||||||
|
select: new Set(),
|
||||||
|
from: rel.from.table,
|
||||||
|
pk: pks.map((e) => e.name),
|
||||||
|
ops: new Map(),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
const tobe = createRelMany({
|
||||||
|
row,
|
||||||
|
k,
|
||||||
|
schema,
|
||||||
|
rel,
|
||||||
|
});
|
||||||
|
|
||||||
|
if (tobe) {
|
||||||
|
for (const row of tobe) {
|
||||||
|
for (const key of Object.keys(row)) {
|
||||||
|
rel_many[k].select.add(key);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
rel_many[k].ops.set(row, {
|
||||||
|
delete: new Set(),
|
||||||
|
insert: new Set(),
|
||||||
|
tobe: tobe || [],
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
const select = {} as any;
|
const select = {} as any;
|
||||||
for (const pk of pks) {
|
for (const pk of pks) {
|
||||||
select[pk.name] = true;
|
select[pk.name] = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (Object.keys(rel_many).length > 0) {
|
||||||
|
for (const [k, v] of Object.entries(rel_many)) {
|
||||||
|
select[k] = { select: {} };
|
||||||
|
Object.values(v.pk).map((e) => (select[k].select[e] = true));
|
||||||
|
for (const e of v.select) {
|
||||||
|
select[k].select[e] = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
const existing: any[] = await prisma[table].findMany({
|
const existing: any[] = await prisma[table].findMany({
|
||||||
where,
|
where,
|
||||||
select,
|
select,
|
||||||
|
|
@ -68,7 +167,6 @@ export const execQuery = async (args: DBArg, prisma: any) => {
|
||||||
const exists_idx = new Set<number>();
|
const exists_idx = new Set<number>();
|
||||||
|
|
||||||
const marker = {} as any;
|
const marker = {} as any;
|
||||||
|
|
||||||
for (const row of data) {
|
for (const row of data) {
|
||||||
const found = existing.find((item, idx) => {
|
const found = existing.find((item, idx) => {
|
||||||
for (const pk of pks) {
|
for (const pk of pks) {
|
||||||
|
|
@ -78,6 +176,55 @@ export const execQuery = async (args: DBArg, prisma: any) => {
|
||||||
return true;
|
return true;
|
||||||
});
|
});
|
||||||
|
|
||||||
|
for (const [k, v] of Object.entries(row) as any) {
|
||||||
|
const rel = rels[k];
|
||||||
|
if (rel) {
|
||||||
|
if (rel.type === "has-many" && rel_many[k]) {
|
||||||
|
delete row[k];
|
||||||
|
const current = rel_many[k].ops.get(row);
|
||||||
|
if (current && found && found[k] && current.tobe) {
|
||||||
|
let cur_matches = new Set<any>();
|
||||||
|
for (const tobe of current.tobe) {
|
||||||
|
let tobe_found = false;
|
||||||
|
for (const cur of found[k]) {
|
||||||
|
let matched_all = true;
|
||||||
|
for (const [k, v] of Object.entries(tobe)) {
|
||||||
|
if (cur[k] !== v) {
|
||||||
|
matched_all = false;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (matched_all) {
|
||||||
|
cur_matches.add(cur);
|
||||||
|
tobe_found = true;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (!tobe_found) {
|
||||||
|
current.insert.add(tobe);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for (const cur of found[k]) {
|
||||||
|
if (!cur_matches.has(cur)) {
|
||||||
|
current.delete.add(cur);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
row[k] = { createMany: { data: [...current.insert] } };
|
||||||
|
|
||||||
|
if (current.delete.size > 0) {
|
||||||
|
for (const d of current.delete) {
|
||||||
|
transactions.push(
|
||||||
|
prisma[rel_many[k].to].delete({ where: d })
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
if (mode === "field") {
|
if (mode === "field") {
|
||||||
if (found) {
|
if (found) {
|
||||||
updates.push({ ...row, ...where });
|
updates.push({ ...row, ...where });
|
||||||
|
|
@ -86,9 +233,9 @@ export const execQuery = async (args: DBArg, prisma: any) => {
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
if (found) {
|
if (found) {
|
||||||
updates.push({ ...row });
|
updates.push(row);
|
||||||
} else {
|
} else {
|
||||||
inserts.push({ ...row });
|
inserts.push(row);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
@ -103,21 +250,6 @@ export const execQuery = async (args: DBArg, prisma: any) => {
|
||||||
|
|
||||||
if (inserts.length > 0) {
|
if (inserts.length > 0) {
|
||||||
for (const row of inserts) {
|
for (const row of inserts) {
|
||||||
for (const [k, v] of Object.entries(row) as any) {
|
|
||||||
const rel = rels[k];
|
|
||||||
if (rel) {
|
|
||||||
if (rel.type === "has-one") {
|
|
||||||
const to = rel.to.fields[0];
|
|
||||||
if (!v.connect && v[to]) {
|
|
||||||
let newv = { connect: { [to]: v[to] } };
|
|
||||||
row[k] = newv;
|
|
||||||
}
|
|
||||||
} else if (rel.type === "has-many") {
|
|
||||||
upsertRelMany({ schema, k, rel, row });
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (typeof row._marker !== "undefined") {
|
if (typeof row._marker !== "undefined") {
|
||||||
marker[transactions.length] = row._marker;
|
marker[transactions.length] = row._marker;
|
||||||
delete row._marker;
|
delete row._marker;
|
||||||
|
|
@ -139,21 +271,6 @@ export const execQuery = async (args: DBArg, prisma: any) => {
|
||||||
delete row[pk.name];
|
delete row[pk.name];
|
||||||
}
|
}
|
||||||
|
|
||||||
for (const [k, v] of Object.entries(row) as any) {
|
|
||||||
const rel = rels[k];
|
|
||||||
if (rel) {
|
|
||||||
if (rel.type === "has-one") {
|
|
||||||
const to = rel.to.fields[0];
|
|
||||||
if (!v.connect && v[to]) {
|
|
||||||
let newv = { connect: { [to]: v[to] } };
|
|
||||||
row[k] = newv;
|
|
||||||
}
|
|
||||||
} else if (rel.type === "has-many") {
|
|
||||||
upsertRelMany({ schema, k, rel, row });
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (typeof row._marker !== "undefined") {
|
if (typeof row._marker !== "undefined") {
|
||||||
marker[transactions.length] = row._marker;
|
marker[transactions.length] = row._marker;
|
||||||
delete row._marker;
|
delete row._marker;
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue