70 lines
1.5 KiB
JavaScript
70 lines
1.5 KiB
JavaScript
'use strict';
|
|
|
|
/**
|
|
* Driver for Sqlite databases
|
|
*
|
|
* @module drivers/sqlite
|
|
*/
|
|
module.exports = (function() {
|
|
delete require.cache[require.resolve('../Driver')];
|
|
let driver = require('../Driver'),
|
|
helpers = require('../helpers');
|
|
|
|
// Sqlite doesn't have a truncate command
|
|
driver.hasTruncate = false;
|
|
|
|
/**
|
|
* SQL to insert a group of rows
|
|
* Override default to have better compatibility
|
|
*
|
|
* @param {String} table - The table to insert to
|
|
* @param {Array} [data] - The array of object containing data to insert
|
|
* @return {String} - The generated sql statement
|
|
*/
|
|
driver.insertBatch = function(table, data) {
|
|
|
|
// Get the data values to insert, so they can
|
|
// be parameterized
|
|
let sql = '',
|
|
vals = [],
|
|
cols = [],
|
|
fields = [],
|
|
first = data.shift(),
|
|
params = [],
|
|
paramString = '',
|
|
paramList = [];
|
|
|
|
data.forEach(obj => {
|
|
let row = [];
|
|
Object.keys(obj).forEach(key => {
|
|
row.push(obj[key]);
|
|
});
|
|
vals.push(row);
|
|
});
|
|
|
|
sql += `INSERT INTO ${driver.quoteTable(table)}\n`;
|
|
|
|
// Get the field names from the keys of the first
|
|
// object to be inserted
|
|
fields = Object.keys(first);
|
|
Object.keys(first).forEach(key => {
|
|
cols.push(`'${driver._quote(first[key])}' AS ${driver.quoteIdentifiers(key)}`);
|
|
});
|
|
|
|
sql += `SELECT ${cols.join(', ')}\n`;
|
|
|
|
vals.forEach(rowValues => {
|
|
let quoted = rowValues.map(value => {
|
|
return String(value).replace('\'', '\'\'');
|
|
});
|
|
sql += `UNION ALL SELECT '${quoted.join('\', \'')}'\n`;
|
|
});
|
|
|
|
return {
|
|
sql: sql,
|
|
values: null,
|
|
};
|
|
};
|
|
|
|
return driver;
|
|
}()); |