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

tsql call

Browse Source
This commit is contained in:
SPRINX0\prochazka
2024-12-04 13:27:20 +01:00
parent 0fde8c49a7
commit edf3a072c5
4 changed files with 33 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
packages/tools/src/SqlDumper.ts
View File

@@ -785,7 +785,11 @@ export class SqlDumper implements AlterProcessor {
}
declareVariable(name: string, type: string, defaultValueLiteral?: string) {}
executeCallable(func: CallableObjectInfo, argLiterals: string[]) {
this.putCmd('^call %f(%,s)', func, argLiterals);
executeCallable(func: CallableObjectInfo, argLiteralsByName: { [name: string]: string }) {
this.putCmd(
'^call %f(%,s)',
func,
(func.parameters || []).map(x => argLiteralsByName[x.parameterName])
);
}
}

2
packages/types/dumper.d.ts vendored
View File

@@ -18,7 +18,7 @@ export interface SqlDumper extends AlterProcessor {
dropDatabase(name: string);
declareVariable(name: string, type: string, defaultValueLiteral?: string);
executeCallable(func: CallableObjectInfo, argLiterals: string[]);
executeCallable(func: CallableObjectInfo, argLiteralsByName: { [name: string]: string });
endCommand();
allowIdentityInsert(table: NamedObjectInfo, allow: boolean);

6
packages/web/src/utility/applyScriptTemplate.ts
View File

@@ -77,7 +77,7 @@ export default async function applyScriptTemplate(
const driver = findEngineDriver(connection, extensions) || driverBase;
const dmp = driver.createDumper();
if (procedureInfo) {
const sqlVars = [];
const argLiteralsByName = {};
for (const param of procedureInfo.parameters || []) {
const sqlVarName = param.parameterName?.startsWith('@')
? param.parameterName?.substring(1)
@@ -88,9 +88,9 @@ export default async function applyScriptTemplate(
param.dataType,
param.parameterMode == 'OUT' ? null : `:${sqlVarName}`
);
sqlVars.push(param.parameterName);
argLiteralsByName[param.parameterName] = param.parameterName;
}
dmp.executeCallable(procedureInfo, sqlVars);
dmp.executeCallable(procedureInfo, argLiteralsByName);
}
// if (procedureInfo) dmp.put('^execute %f', procedureInfo);
return dmp.s;

32
plugins/dbgate-plugin-mssql/src/frontend/MsSqlDumper.js
View File

@@ -1,5 +1,4 @@
const { SqlDumper, testEqualColumns, arrayToHexString } = global.DBGATE_PACKAGES['dbgate-tools'];
const _zip = require('lodash/zip');
class MsSqlDumper extends SqlDumper {
constructor(driver, options) {
@@ -170,18 +169,33 @@ class MsSqlDumper extends SqlDumper {
this.endCommand();
}
executeCallable(func, argLiterals) {
console.log('executeCallable', func, argLiterals);
if (func.objectTypeField == 'procedures') {
this.put('^execute %f&>&n', func, argLiterals);
executeCallable(func, argLiteralsByName) {
const putParameters = (parameters, delimiter) => {
this.putCollection(
delimiter,
(parameters || []), param => {
this.putRaw(argLiteralsByName[param]);
if (param?.parameterMode == 'OUT') this.put(' ^output');
}
);
};
this.putCollection(',&n', _zip(func.parameters || [], argLiterals || []), ([param, value]) => {
this.putRaw(value);
if (param?.parameterMode == 'OUT') this.put(' ^output');
});
if (func.objectTypeField == 'procedures') {
this.put('^execute %f&>&n', func);
putParameters(func.parameters, ',&n');
this.put('&<&n');
this.endCommand();
}
if (func.objectTypeField == 'functions') {
this.put('^select %f(', func);
putParameters(
(func.parameters || []).filter(x => x.parameterMode != 'OUT'),
', '
);
this.put(')');
this.endCommand();
}
}
}