dennii/dbgate
1
0
Fork 0
mirror of https://github.com/DeNNiiInc/dbgate.git synced 2026-04-21 17:36:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

changeset: support save document to SQL

Browse Source
This commit is contained in:
Jan Prochazka
2024-07-08 15:56:23 +02:00
parent 2199a49126
commit 2a56b562eb
1 changed files with 34 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
packages/datalib/src/ChangeSet.ts
View File

@@ -229,13 +229,23 @@ export function batchUpdateChangeSet(
return changeSet; return changeSet;
} }
function extractFields(item: ChangeSetItem, allowNulls = true): UpdateField[] { function extractFields(item: ChangeSetItem, allowNulls = true, allowedDocumentColumns: string[] = []): UpdateField[] {
return _.keys(item.fields) const allFields = {
.filter(targetColumn => allowNulls || item.fields[targetColumn] != null) ...item.fields,
};
for (const docField in item.document || {}) {
if (allowedDocumentColumns.includes(docField)) {
allFields[docField] = item.document[docField];
}
}
return _.keys(allFields)
.filter(targetColumn => allowNulls || allFields[targetColumn] != null)
.map(targetColumn => ({ .map(targetColumn => ({
targetColumn, targetColumn,
exprType: 'value', exprType: 'value',
value: item.fields[targetColumn], value: allFields[targetColumn],
})); }));
} }
@@ -243,17 +253,19 @@ function changeSetInsertToSql(
item: ChangeSetItem, item: ChangeSetItem,
dbinfo: DatabaseInfo = null dbinfo: DatabaseInfo = null
): [AllowIdentityInsert, Insert, AllowIdentityInsert] { ): [AllowIdentityInsert, Insert, AllowIdentityInsert] {
const fields = extractFields(item, false); const table = dbinfo?.tables?.find(x => x.schemaName == item.schemaName && x.pureName == item.pureName);
const fields = extractFields(
item,
false,
table?.columns?.map(x => x.columnName)
);
if (fields.length == 0) return null; if (fields.length == 0) return null;
let autoInc = false; let autoInc = false;
if (dbinfo) { if (table) {
const table = dbinfo.tables.find(x => x.schemaName == item.schemaName && x.pureName == item.pureName); const autoIncCol = table.columns.find(x => x.autoIncrement);
if (table) { // console.log('autoIncCol', autoIncCol);
const autoIncCol = table.columns.find(x => x.autoIncrement); if (autoIncCol && fields.find(x => x.targetColumn == autoIncCol.columnName)) {
// console.log('autoIncCol', autoIncCol); autoInc = true;
if (autoIncCol && fields.find(x => x.targetColumn == autoIncCol.columnName)) {
autoInc = true;
}
} }
} }
const targetTable = { const targetTable = {
@@ -320,7 +332,9 @@ export function extractChangeSetCondition(item: ChangeSetItem, alias?: string):
}; };
} }
function changeSetUpdateToSql(item: ChangeSetItem): Update { function changeSetUpdateToSql(item: ChangeSetItem, dbinfo: DatabaseInfo = null): Update {
const table = dbinfo?.tables?.find(x => x.schemaName == item.schemaName && x.pureName == item.pureName);
return { return {
from: { from: {
name: { name: {
@@ -329,7 +343,11 @@ function changeSetUpdateToSql(item: ChangeSetItem): Update {
}, },
}, },
commandType: 'update', commandType: 'update',
fields: extractFields(item), fields: extractFields(
item,
true,
table?.columns?.map(x => x.columnName)
),
where: extractChangeSetCondition(item), where: extractChangeSetCondition(item),
}; };
} }
@@ -351,7 +369,7 @@ export function changeSetToSql(changeSet: ChangeSet, dbinfo: DatabaseInfo): Comm
return _.compact( return _.compact(
_.flatten([ _.flatten([
...(changeSet.inserts.map(item => changeSetInsertToSql(item, dbinfo)) as any), ...(changeSet.inserts.map(item => changeSetInsertToSql(item, dbinfo)) as any),
...changeSet.updates.map(changeSetUpdateToSql), ...changeSet.updates.map(item => changeSetUpdateToSql(item, dbinfo)),
...changeSet.deletes.map(changeSetDeleteToSql), ...changeSet.deletes.map(changeSetDeleteToSql),
]) ])
); );