112 lines
3.9 KiB
JavaScript
112 lines
3.9 KiB
JavaScript
|
'use strict';
|
||
|
|
|
||
|
|
const fs = require('fs');
|
||
|
|
const path = require('path');
|
||
|
|
const AbstractConnectionManager = require('../abstract/connection-manager');
|
||
|
|
const { logger } = require('../../utils/logger');
|
||
|
|
const debug = logger.debugContext('connection:sqlite');
|
||
|
|
const dataTypes = require('../../data-types').sqlite;
|
||
|
|
const sequelizeErrors = require('../../errors');
|
||
|
|
const parserStore = require('../parserStore')('sqlite');
|
||
|
|
const { promisify } = require('util');
|
||
|
|
|
||
|
|
class ConnectionManager extends AbstractConnectionManager {
|
||
|
|
constructor(dialect, sequelize) {
|
||
|
|
super(dialect, sequelize);
|
||
|
|
|
||
|
|
// We attempt to parse file location from a connection uri
|
||
|
|
// but we shouldn't match sequelize default host.
|
||
|
|
if (this.sequelize.options.host === 'localhost') {
|
||
|
|
delete this.sequelize.options.host;
|
||
|
|
}
|
||
|
|
|
||
|
|
this.connections = {};
|
||
|
|
this.lib = this._loadDialectModule('sqlite3');
|
||
|
|
this.refreshTypeParser(dataTypes);
|
||
|
|
}
|
||
|
|
|
||
|
|
async _onProcessExit() {
|
||
|
|
await Promise.all(
|
||
|
|
Object.getOwnPropertyNames(this.connections)
|
||
|
|
.map(connection => promisify(callback => this.connections[connection].close(callback))())
|
||
|
|
);
|
||
|
|
return super._onProcessExit.call(this);
|
||
|
|
}
|
||
|
|
|
||
|
|
// Expose this as a method so that the parsing may be updated when the user has added additional, custom types
|
||
|
|
_refreshTypeParser(dataType) {
|
||
|
|
parserStore.refresh(dataType);
|
||
|
|
}
|
||
|
|
|
||
|
|
_clearTypeParser() {
|
||
|
|
parserStore.clear();
|
||
|
|
}
|
||
|
|
|
||
|
|
async getConnection(options) {
|
||
|
|
options = options || {};
|
||
|
|
options.uuid = options.uuid || 'default';
|
||
|
|
|
||
|
|
if (!!this.sequelize.options.storage !== null && this.sequelize.options.storage !== undefined) {
|
||
|
|
// Check explicitely for the storage option to not be set since an empty string signals
|
||
|
|
// SQLite will create a temporary disk-based database in that case.
|
||
|
|
options.storage = this.sequelize.options.storage;
|
||
|
|
} else {
|
||
|
|
options.storage = this.sequelize.options.host || ':memory:';
|
||
|
|
}
|
||
|
|
|
||
|
|
options.inMemory = options.storage === ':memory:' ? 1 : 0;
|
||
|
|
|
||
|
|
const dialectOptions = this.sequelize.options.dialectOptions;
|
||
|
|
const defaultReadWriteMode = this.lib.OPEN_READWRITE | this.lib.OPEN_CREATE;
|
||
|
|
|
||
|
|
options.readWriteMode = dialectOptions && dialectOptions.mode || defaultReadWriteMode;
|
||
|
|
|
||
|
|
if (this.connections[options.inMemory || options.uuid]) {
|
||
|
|
return this.connections[options.inMemory || options.uuid];
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!options.inMemory && (options.readWriteMode & this.lib.OPEN_CREATE) !== 0) {
|
||
|
|
// automatic path provision for `options.storage`
|
||
|
|
fs.mkdirSync(path.dirname(options.storage), { recursive: true });
|
||
|
|
}
|
||
|
|
|
||
|
|
const connection = await new Promise((resolve, reject) => {
|
||
|
|
this.connections[options.inMemory || options.uuid] = new this.lib.Database(
|
||
|
|
options.storage,
|
||
|
|
options.readWriteMode,
|
||
|
|
err => {
|
||
|
|
if (err) return reject(new sequelizeErrors.ConnectionError(err));
|
||
|
|
debug(`connection acquired ${options.uuid}`);
|
||
|
|
resolve(this.connections[options.inMemory || options.uuid]);
|
||
|
|
}
|
||
|
|
);
|
||
|
|
});
|
||
|
|
|
||
|
|
if (this.sequelize.config.password) {
|
||
|
|
// Make it possible to define and use password for sqlite encryption plugin like sqlcipher
|
||
|
|
connection.run(`PRAGMA KEY=${this.sequelize.escape(this.sequelize.config.password)}`);
|
||
|
|
}
|
||
|
|
if (this.sequelize.options.foreignKeys !== false) {
|
||
|
|
// Make it possible to define and use foreign key constraints unless
|
||
|
|
// explicitly disallowed. It's still opt-in per relation
|
||
|
|
connection.run('PRAGMA FOREIGN_KEYS=ON');
|
||
|
|
}
|
||
|
|
|
||
|
|
return connection;
|
||
|
|
}
|
||
|
|
|
||
|
|
releaseConnection(connection, force) {
|
||
|
|
if (connection.filename === ':memory:' && force !== true) return;
|
||
|
|
|
||
|
|
if (connection.uuid) {
|
||
|
|
connection.close();
|
||
|
|
debug(`connection released ${connection.uuid}`);
|
||
|
|
delete this.connections[connection.uuid];
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
module.exports = ConnectionManager;
|
||
|
|
module.exports.ConnectionManager = ConnectionManager;
|
||
|
|
module.exports.default = ConnectionManager;
|