69 lines
2.5 KiB
JavaScript
69 lines
2.5 KiB
JavaScript
function quoteIdentifier(identifier) {
|
|
return `\`${identifier.replace(/`/g, "``")}\``;
|
|
}
|
|
/**
|
|
* # SQL Insert Generator
|
|
*/
|
|
export default function sqlInsertGenerator({ tableName, data, dbFullName, }) {
|
|
const finalDbName = dbFullName ? `${quoteIdentifier(dbFullName)}.` : "";
|
|
try {
|
|
if (Array.isArray(data) && data?.[0]) {
|
|
let insertKeys = [];
|
|
data.forEach((dt) => {
|
|
const kys = Object.keys(dt);
|
|
kys.forEach((ky) => {
|
|
if (!insertKeys.includes(ky)) {
|
|
insertKeys.push(ky);
|
|
}
|
|
});
|
|
});
|
|
let queryBatches = [];
|
|
let queryValues = [];
|
|
data.forEach((item) => {
|
|
queryBatches.push(`(${insertKeys
|
|
.map((ky) => {
|
|
const value = item[ky];
|
|
const is_embedding = Array.isArray(value) &&
|
|
typeof value[0] == "number";
|
|
const finalValue = typeof value == "string" ||
|
|
typeof value == "number"
|
|
? value
|
|
: typeof value == "function"
|
|
? value().value
|
|
: is_embedding
|
|
? JSON.stringify(value)
|
|
: value
|
|
? value
|
|
: null;
|
|
if (!finalValue && typeof value !== "number") {
|
|
queryValues.push(null);
|
|
return "?";
|
|
}
|
|
queryValues.push(finalValue);
|
|
const placeholder = typeof value == "function"
|
|
? value().placeholder
|
|
: is_embedding
|
|
? "VEC_FromText(?)"
|
|
: "?";
|
|
return placeholder;
|
|
})
|
|
.filter((k) => Boolean(k))
|
|
.join(",")})`);
|
|
});
|
|
const insertColumns = insertKeys.map(quoteIdentifier).join(",");
|
|
let query = `INSERT INTO ${finalDbName}${quoteIdentifier(tableName)} (${insertColumns}) VALUES ${queryBatches.join(",")}`;
|
|
return {
|
|
query: query,
|
|
values: queryValues,
|
|
};
|
|
}
|
|
else {
|
|
return undefined;
|
|
}
|
|
}
|
|
catch ( /** @type {any} */error) {
|
|
console.log(`SQL insert gen ERROR: ${error.message}`);
|
|
return undefined;
|
|
}
|
|
}
|