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

feat: pssql callable template

Browse Source
This commit is contained in:
Nybkox
2024-12-05 15:32:33 +01:00
parent 216614a4b1
commit e231a3a41a
1 changed files with 51 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
plugins/dbgate-plugin-postgres/src/frontend/Dumper.js
View File

@@ -109,6 +109,57 @@ class Dumper extends SqlDumper {
const column = table.columns && table.columns.find(x => x.autoIncrement); const column = table.columns && table.columns.find(x => x.autoIncrement);
this.put("^SELECT currval(pg_get_serial_sequence('%f','%s'))", table, column ? column.columnName : null); this.put("^SELECT currval(pg_get_serial_sequence('%f','%s'))", table, column ? column.columnName : null);
} }
callableTemplate(func) {
const putDeclareParamters = parameters => {
for (const param of parameters.filter(i => i.parameterMode != 'RETURN')) {
if (param.parameterMode == 'IN') {
this.put('%s %s := :%s', param.parameterName, param.dataType, param.parameterName);
this.endCommand();
} else {
this.put('%s %s', param.parameterName, param.dataType);
this.endCommand();
}
}
this.put('&n');
};
const putParameters = (parameters, delimiter) => {
this.putCollection(delimiter, parameters || [], param => {
this.putRaw(param.parameterName);
});
};
if (func.objectTypeField == 'procedures') {
this.put('^do $$&n');
this.put('^declare&n');
putDeclareParamters(func.parameters);
this.put('^begin&n');
this.put('^call %f(&>&n', func);
putParameters(func.parameters, ',&n');
this.put('&<&n)');
this.endCommand();
this.put('&n');
this.put('^end $$');
this.endCommand();
}
if (func.objectTypeField == 'functions') {
this.put('^do $$&n');
this.put('^declare&n');
this.put('result %s', func.returnType);
this.endCommand();
putDeclareParamters(func.parameters);
this.put('^begin&n');
this.put('result := %f(&>&n', func);
putParameters(func.parameters, ',&n');
this.put('&<&n)');
this.endCommand();
this.put('&n');
this.put('^end $$');
this.endCommand();
}
}
} }
module.exports = Dumper; module.exports = Dumper;