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

feat: myssql callable template

Browse Source
This commit is contained in:
Nybkox
2024-12-05 11:58:58 +01:00
parent 1429c29537
commit 7f639361b8
1 changed files with 40 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
plugins/dbgate-plugin-mysql/src/frontend/Dumper.js
View File

@@ -97,6 +97,46 @@ class Dumper extends SqlDumper {
selectScopeIdentity() {
this.put('^select ^last_insert_id()');
}
callableTemplate(func) {
const parameters = (func.parameters || []).filter(x => x.parameterMode != 'RETURN');
const putParameters = (parameters, delimiter) => {
this.putCollection(delimiter, parameters || [], param => {
if (param.parameterMode == 'IN') {
this.putRaw('@' + param.parameterName);
} else {
this.putRaw('@' + param.parameterName + 'Output');
}
});
};
const putSetParamters = parameters => {
for (const param of parameters || []) {
if (param.parameterMode == 'IN') {
this.put('SET @%s = :%s', param.parameterName, param.parameterName);
this.endCommand();
}
}
this.put('&n');
};
if (func.objectTypeField == 'procedures') {
putSetParamters(func.parameters);
this.put('^call %f(&>&n', func);
putParameters(parameters, ',&n');
this.put('&<&n)');
this.endCommand();
}
if (func.objectTypeField == 'functions') {
putSetParamters(parameters);
this.put('^select %f(&>&n', func);
putParameters(parameters, ',&n');
this.put('&<&n)');
this.endCommand();
}
}
}
module.exports = Dumper;