51 lines
1.2 KiB
JavaScript
51 lines
1.2 KiB
JavaScript
|
const { isString, tail } = require('lodash');
|
||
|
|
|
||
|
|
// Push a new query onto the compiled "sequence" stack,
|
||
|
|
// creating a new formatter, returning the compiler.
|
||
|
|
function pushQuery(query) {
|
||
|
|
if (!query) return;
|
||
|
|
if (isString(query)) {
|
||
|
|
query = { sql: query };
|
||
|
|
}
|
||
|
|
if (!query.bindings) {
|
||
|
|
query.bindings = this.formatter.bindings;
|
||
|
|
}
|
||
|
|
this.sequence.push(query);
|
||
|
|
|
||
|
|
this.formatter = this.client.formatter(this._commonBuilder);
|
||
|
|
}
|
||
|
|
|
||
|
|
// Used in cases where we need to push some additional column specific statements.
|
||
|
|
function pushAdditional(fn) {
|
||
|
|
const child = new this.constructor(
|
||
|
|
this.client,
|
||
|
|
this.tableCompiler,
|
||
|
|
this.columnBuilder
|
||
|
|
);
|
||
|
|
fn.call(child, tail(arguments));
|
||
|
|
this.sequence.additional = (this.sequence.additional || []).concat(
|
||
|
|
child.sequence
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
// Unshift a new query onto the compiled "sequence" stack,
|
||
|
|
// creating a new formatter, returning the compiler.
|
||
|
|
function unshiftQuery(query) {
|
||
|
|
if (!query) return;
|
||
|
|
if (isString(query)) {
|
||
|
|
query = { sql: query };
|
||
|
|
}
|
||
|
|
if (!query.bindings) {
|
||
|
|
query.bindings = this.formatter.bindings;
|
||
|
|
}
|
||
|
|
this.sequence.unshift(query);
|
||
|
|
|
||
|
|
this.formatter = this.client.formatter(this._commonBuilder);
|
||
|
|
}
|
||
|
|
|
||
|
|
module.exports = {
|
||
|
|
pushAdditional,
|
||
|
|
pushQuery,
|
||
|
|
unshiftQuery,
|
||
|
|
};
|