177 lines
4.9 KiB
JavaScript
177 lines
4.9 KiB
JavaScript
|
// SQLite3 Query Builder & Compiler
|
||
|
|
||
|
const inherits = require('inherits');
|
||
|
const QueryCompiler = require('../../../query/compiler');
|
||
|
const {
|
||
|
assign,
|
||
|
each,
|
||
|
isEmpty,
|
||
|
isString,
|
||
|
noop,
|
||
|
reduce,
|
||
|
identity,
|
||
|
} = require('lodash');
|
||
|
|
||
|
function QueryCompiler_SQLite3(client, builder) {
|
||
|
QueryCompiler.call(this, client, builder);
|
||
|
|
||
|
const { returning } = this.single;
|
||
|
|
||
|
if (returning) {
|
||
|
this.client.logger.warn(
|
||
|
'.returning() is not supported by sqlite3 and will not have any effect.'
|
||
|
);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
inherits(QueryCompiler_SQLite3, QueryCompiler);
|
||
|
|
||
|
assign(QueryCompiler_SQLite3.prototype, {
|
||
|
// The locks are not applicable in SQLite3
|
||
|
forShare: emptyStr,
|
||
|
|
||
|
forUpdate: emptyStr,
|
||
|
|
||
|
// SQLite requires us to build the multi-row insert as a listing of select with
|
||
|
// unions joining them together. So we'll build out this list of columns and
|
||
|
// then join them all together with select unions to complete the queries.
|
||
|
insert() {
|
||
|
const insertValues = this.single.insert || [];
|
||
|
let sql = this.with() + `insert into ${this.tableName} `;
|
||
|
|
||
|
if (Array.isArray(insertValues)) {
|
||
|
if (insertValues.length === 0) {
|
||
|
return '';
|
||
|
} else if (
|
||
|
insertValues.length === 1 &&
|
||
|
insertValues[0] &&
|
||
|
isEmpty(insertValues[0])
|
||
|
) {
|
||
|
return sql + this._emptyInsertValue;
|
||
|
}
|
||
|
} else if (typeof insertValues === 'object' && isEmpty(insertValues)) {
|
||
|
return sql + this._emptyInsertValue;
|
||
|
}
|
||
|
|
||
|
const insertData = this._prepInsert(insertValues);
|
||
|
|
||
|
if (isString(insertData)) {
|
||
|
return sql + insertData;
|
||
|
}
|
||
|
|
||
|
if (insertData.columns.length === 0) {
|
||
|
return '';
|
||
|
}
|
||
|
|
||
|
sql += `(${this.formatter.columnize(insertData.columns)})`;
|
||
|
|
||
|
// backwards compatible error
|
||
|
if (this.client.valueForUndefined !== null) {
|
||
|
each(insertData.values, (bindings) => {
|
||
|
each(bindings, (binding) => {
|
||
|
if (binding === undefined)
|
||
|
throw new TypeError(
|
||
|
'`sqlite` does not support inserting default values. Specify ' +
|
||
|
'values explicitly or use the `useNullAsDefault` config flag. ' +
|
||
|
'(see docs http://knexjs.org/#Builder-insert).'
|
||
|
);
|
||
|
});
|
||
|
});
|
||
|
}
|
||
|
|
||
|
if (insertData.values.length === 1) {
|
||
|
const parameters = this.formatter.parameterize(
|
||
|
insertData.values[0],
|
||
|
this.client.valueForUndefined
|
||
|
);
|
||
|
return sql + ` values (${parameters})`;
|
||
|
}
|
||
|
|
||
|
const blocks = [];
|
||
|
let i = -1;
|
||
|
while (++i < insertData.values.length) {
|
||
|
let i2 = -1;
|
||
|
const block = (blocks[i] = []);
|
||
|
let current = insertData.values[i];
|
||
|
current = current === undefined ? this.client.valueForUndefined : current;
|
||
|
while (++i2 < insertData.columns.length) {
|
||
|
block.push(
|
||
|
this.formatter.alias(
|
||
|
this.formatter.parameter(current[i2]),
|
||
|
this.formatter.wrap(insertData.columns[i2])
|
||
|
)
|
||
|
);
|
||
|
}
|
||
|
blocks[i] = block.join(', ');
|
||
|
}
|
||
|
return sql + ' select ' + blocks.join(' union all select ');
|
||
|
},
|
||
|
|
||
|
// Compile a truncate table statement into SQL.
|
||
|
truncate() {
|
||
|
const { table } = this.single;
|
||
|
return {
|
||
|
sql: `delete from ${this.tableName}`,
|
||
|
output() {
|
||
|
return this.query({
|
||
|
sql: `delete from sqlite_sequence where name = '${table}'`,
|
||
|
}).catch(noop);
|
||
|
},
|
||
|
};
|
||
|
},
|
||
|
|
||
|
// Compiles a `columnInfo` query
|
||
|
columnInfo() {
|
||
|
const column = this.single.columnInfo;
|
||
|
|
||
|
// The user may have specified a custom wrapIdentifier function in the config. We
|
||
|
// need to run the identifiers through that function, but not format them as
|
||
|
// identifiers otherwise.
|
||
|
const table = this.client.customWrapIdentifier(this.single.table, identity);
|
||
|
|
||
|
return {
|
||
|
sql: `PRAGMA table_info(\`${table}\`)`,
|
||
|
output(resp) {
|
||
|
const maxLengthRegex = /.*\((\d+)\)/;
|
||
|
const out = reduce(
|
||
|
resp,
|
||
|
function(columns, val) {
|
||
|
let { type } = val;
|
||
|
let maxLength = type.match(maxLengthRegex);
|
||
|
if (maxLength) {
|
||
|
maxLength = maxLength[1];
|
||
|
}
|
||
|
type = maxLength ? type.split('(')[0] : type;
|
||
|
columns[val.name] = {
|
||
|
type: type.toLowerCase(),
|
||
|
maxLength,
|
||
|
nullable: !val.notnull,
|
||
|
defaultValue: val.dflt_value,
|
||
|
};
|
||
|
return columns;
|
||
|
},
|
||
|
{}
|
||
|
);
|
||
|
return (column && out[column]) || out;
|
||
|
},
|
||
|
};
|
||
|
},
|
||
|
|
||
|
limit() {
|
||
|
const noLimit = !this.single.limit && this.single.limit !== 0;
|
||
|
if (noLimit && !this.single.offset) return '';
|
||
|
|
||
|
// Workaround for offset only,
|
||
|
// see http://stackoverflow.com/questions/10491492/sqllite-with-skip-offset-only-not-limit
|
||
|
return `limit ${this.formatter.parameter(
|
||
|
noLimit ? -1 : this.single.limit
|
||
|
)}`;
|
||
|
},
|
||
|
});
|
||
|
|
||
|
function emptyStr() {
|
||
|
return '';
|
||
|
}
|
||
|
|
||
|
module.exports = QueryCompiler_SQLite3;
|