initial test of factory
This commit is contained in:
parent
e6155d9f33
commit
efc10cab3b
96
client.js
96
client.js
|
@ -56,7 +56,7 @@ function getConnection(opts, verbs, mySocket) {
|
||||||
|
|
||||||
ws.on('error', function (err) {
|
ws.on('error', function (err) {
|
||||||
console.error('[ERROR] ws connection failed, retrying');
|
console.error('[ERROR] ws connection failed, retrying');
|
||||||
console.error(err.stack);
|
console.error(err.stack || err);
|
||||||
|
|
||||||
function retry() {
|
function retry() {
|
||||||
// TODO eventually throw up
|
// TODO eventually throw up
|
||||||
|
@ -72,7 +72,7 @@ function getConnection(opts, verbs, mySocket) {
|
||||||
resolve({ masterClient: client });
|
resolve({ masterClient: client });
|
||||||
}, function (err) {
|
}, function (err) {
|
||||||
console.error('[ERROR] failed to connect to sqlite3-cluster service. retrying...');
|
console.error('[ERROR] failed to connect to sqlite3-cluster service. retrying...');
|
||||||
console.error(err);
|
console.error(err.stack || err);
|
||||||
retry();
|
retry();
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
@ -110,6 +110,15 @@ module.exports.createClientFactory = function (conf, verbs, _socket) {
|
||||||
if (_socket && _s) {
|
if (_socket && _s) {
|
||||||
throw new Error("[E_USR_SOCKET] Your parent has decided that you may not choose your own SOCKET. Don't get mad at me, take it up with them.");
|
throw new Error("[E_USR_SOCKET] Your parent has decided that you may not choose your own SOCKET. Don't get mad at me, take it up with them.");
|
||||||
}
|
}
|
||||||
|
if (opts.key && conf.key) {
|
||||||
|
throw new Error("[E_USR_KEY] Your parent has decided that you may not choose your own KEY. Don't get mad at me, take it up with them.");
|
||||||
|
}
|
||||||
|
if (opts.algo && conf.algo) {
|
||||||
|
throw new Error("[E_USR_ALGO] Your parent has decided that you may not choose your own ALGO. Don't get mad at me, take it up with them.");
|
||||||
|
}
|
||||||
|
if (opts.bits && conf.bits) {
|
||||||
|
throw new Error("[E_USR_BITS] Your parent has decided that you may not choose your own BITS. Don't get mad at me, take it up with them.");
|
||||||
|
}
|
||||||
if (opts.dirname && conf.dirname) {
|
if (opts.dirname && conf.dirname) {
|
||||||
throw new Error("[E_USR_TENANT] Your parent has decided that you may not choose your own TENANT. Don't get mad at me, take it up with them.");
|
throw new Error("[E_USR_TENANT] Your parent has decided that you may not choose your own TENANT. Don't get mad at me, take it up with them.");
|
||||||
}
|
}
|
||||||
|
@ -173,6 +182,23 @@ module.exports.create = function (opts, verbs, mySocket) {
|
||||||
}
|
}
|
||||||
|
|
||||||
function create(opts) {
|
function create(opts) {
|
||||||
|
function retryServe() {
|
||||||
|
return startServer(opts, verbs).then(function (client) {
|
||||||
|
// ws.masterClient = client;
|
||||||
|
return { masterClient: client };
|
||||||
|
}, function (err) {
|
||||||
|
console.error('[ERROR] retryServe()');
|
||||||
|
console.error(err.stack || err);
|
||||||
|
retryServe();
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
if (opts.serve) {
|
||||||
|
return retryServe(opts).then(function (servers) {
|
||||||
|
return servers.masterClient;
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
if (!opts.tenant) {
|
if (!opts.tenant) {
|
||||||
opts.tenant = "";
|
opts.tenant = "";
|
||||||
}
|
}
|
||||||
|
@ -201,37 +227,17 @@ module.exports.create = function (opts, verbs, mySocket) {
|
||||||
return require('./wrapper').create(opts, verbs);
|
return require('./wrapper').create(opts, verbs);
|
||||||
}
|
}
|
||||||
|
|
||||||
function retryServe() {
|
|
||||||
return startServer(opts, verbs).then(function (client) {
|
|
||||||
// ws.masterClient = client;
|
|
||||||
return { masterClient: client };
|
|
||||||
}, function (err) {
|
|
||||||
console.error('[ERROR] retryServe()');
|
|
||||||
console.error(err);
|
|
||||||
retryServe();
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!opts.sock) {
|
if (!opts.sock) {
|
||||||
throw new Error("Please specify opts.sock as the path to the master socket. '/tmp/sqlite3-cluster' would do nicely.");
|
throw new Error("Please specify opts.sock as the path to the master socket. '/tmp/sqlite3-cluster' would do nicely.");
|
||||||
}
|
}
|
||||||
|
|
||||||
if (opts.serve) {
|
|
||||||
promise = retryServe(opts);
|
|
||||||
} else {
|
|
||||||
promise = getConnection(opts, verbs, mySocket).then(function (socket) {
|
promise = getConnection(opts, verbs, mySocket).then(function (socket) {
|
||||||
mySocket = socket;
|
mySocket = socket;
|
||||||
return mySocket;
|
return mySocket;
|
||||||
});
|
});
|
||||||
}
|
|
||||||
|
|
||||||
// TODO maybe use HTTP POST instead?
|
// TODO maybe use HTTP POST instead?
|
||||||
return promise.then(function (ws) {
|
return promise.then(function (ws) {
|
||||||
if (ws.masterClient) {
|
|
||||||
// for the server
|
|
||||||
return ws.masterClient;
|
|
||||||
}
|
|
||||||
|
|
||||||
var db = {};
|
var db = {};
|
||||||
var proto = sqlite3real.Database.prototype;
|
var proto = sqlite3real.Database.prototype;
|
||||||
var messages = [];
|
var messages = [];
|
||||||
|
@ -243,15 +249,26 @@ module.exports.create = function (opts, verbs, mySocket) {
|
||||||
return idprefix + idcount;
|
return idprefix + idcount;
|
||||||
}
|
}
|
||||||
|
|
||||||
function init(opts) {
|
function init(iopts) {
|
||||||
return new PromiseA(function (resolve, reject) {
|
console.log('CLIENT INIT');
|
||||||
// TODO needs to reject by a timeout
|
if (db._initPromise) {
|
||||||
|
return db._initPromise;
|
||||||
|
}
|
||||||
|
|
||||||
|
db._initPromise = new PromiseA(function (resolve, reject) {
|
||||||
|
// TODO needs to reject by a timeout
|
||||||
var id = genId();
|
var id = genId();
|
||||||
ws.send(JSON.stringify({
|
ws.send(JSON.stringify({
|
||||||
type: 'init'
|
type: 'init'
|
||||||
, args: [opts]
|
, args: [{
|
||||||
|
// encryption
|
||||||
|
bits: opts.bits || iopts.bits
|
||||||
|
, algorithm: opts.algo || opts.algorithm || iopts.algorithm || iopts.algo
|
||||||
|
, algo: opts.algo || opts.algorithm || iopts.algorithm || iopts.algo
|
||||||
|
, encmode: opts.mode || iopts.mode
|
||||||
|
}]
|
||||||
, func: 'init'
|
, func: 'init'
|
||||||
|
// db
|
||||||
, dirname: opts.dirname
|
, dirname: opts.dirname
|
||||||
, prefix: opts.prefix
|
, prefix: opts.prefix
|
||||||
, subtenant: opts.subtenant
|
, subtenant: opts.subtenant
|
||||||
|
@ -259,34 +276,23 @@ module.exports.create = function (opts, verbs, mySocket) {
|
||||||
, dbname: opts.dbname
|
, dbname: opts.dbname
|
||||||
, suffix: opts.suffix
|
, suffix: opts.suffix
|
||||||
, ext: opts.ext
|
, ext: opts.ext
|
||||||
|
// session
|
||||||
, id: id
|
, id: id
|
||||||
}));
|
}));
|
||||||
|
|
||||||
function onMessage(data) {
|
function onMessage(data) {
|
||||||
var cmd;
|
var cmd;
|
||||||
|
|
||||||
if (
|
|
||||||
(data.dbname !== opts.dbname)
|
|
||||||
|| (data.dirname !== opts.dirname)
|
|
||||||
|| (data.prefix !== opts.prefix)
|
|
||||||
|| (data.subtenant !== opts.subtenant)
|
|
||||||
|| (data.tenant !== opts.tenant)
|
|
||||||
|| (data.suffix !== opts.suffix)
|
|
||||||
|| (data.ext !== opts.ext)
|
|
||||||
) {
|
|
||||||
return reject(new Error("suxors to rejexors"));
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
try {
|
||||||
cmd = JSON.parse(data.toString('utf8'));
|
cmd = JSON.parse(data.toString('utf8'));
|
||||||
} catch(e) {
|
} catch(e) {
|
||||||
console.error('[ERROR] in client, from sql server parse json');
|
console.error('[ERROR] in client, from sql server parse json');
|
||||||
console.error(e);
|
console.error(e.stack || e);
|
||||||
console.error(data);
|
console.error(data);
|
||||||
console.error();
|
console.error();
|
||||||
|
|
||||||
// ignore this message, it came out of order
|
// ignore this message, it came out of order
|
||||||
return reject(new Error("suxors to rejexors"));
|
return reject(new Error("suxors to rejexors parse"));
|
||||||
}
|
}
|
||||||
|
|
||||||
if (cmd.id !== id) {
|
if (cmd.id !== id) {
|
||||||
|
@ -301,17 +307,21 @@ module.exports.create = function (opts, verbs, mySocket) {
|
||||||
messages.splice(messages.indexOf(onMessage), 1);
|
messages.splice(messages.indexOf(onMessage), 1);
|
||||||
|
|
||||||
if ('error' === cmd.type) {
|
if ('error' === cmd.type) {
|
||||||
|
//console.log('ERROR ARGS');
|
||||||
|
//console.log(cmd);
|
||||||
reject(cmd.args[0]);
|
reject(cmd.args[0]);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
//console.log('RESOLVING INIT');
|
console.log('CLIENT RESOLVING INIT');
|
||||||
resolve(cmd.args[0]);
|
resolve(cmd.args[0]);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
messages.push(onMessage);
|
messages.push(onMessage);
|
||||||
});
|
});
|
||||||
|
|
||||||
|
return db._initPromise;
|
||||||
}
|
}
|
||||||
|
|
||||||
function rpcThunk(fname, args) {
|
function rpcThunk(fname, args) {
|
||||||
|
@ -348,7 +358,7 @@ module.exports.create = function (opts, verbs, mySocket) {
|
||||||
cmd = JSON.parse(data.toString('utf8'));
|
cmd = JSON.parse(data.toString('utf8'));
|
||||||
} catch(e) {
|
} catch(e) {
|
||||||
console.error('[ERROR] in client, from sql server parse json');
|
console.error('[ERROR] in client, from sql server parse json');
|
||||||
console.error(e);
|
console.error(e.stack || e);
|
||||||
console.error(data);
|
console.error(data);
|
||||||
console.error();
|
console.error();
|
||||||
|
|
||||||
|
@ -398,7 +408,7 @@ module.exports.create = function (opts, verbs, mySocket) {
|
||||||
fn(data);
|
fn(data);
|
||||||
} catch(e) {
|
} catch(e) {
|
||||||
console.error("[ERROR] ws.on('message', fn) (multi-callback)");
|
console.error("[ERROR] ws.on('message', fn) (multi-callback)");
|
||||||
console.error(e);
|
console.error(e.stack || e);
|
||||||
// ignore
|
// ignore
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|
|
@ -20,3 +20,5 @@ function create(opts) {
|
||||||
module.exports.sanitize = sqlite3.sanitize;
|
module.exports.sanitize = sqlite3.sanitize;
|
||||||
module.exports.escape = sqlite3.escape;
|
module.exports.escape = sqlite3.escape;
|
||||||
module.exports.create = create;
|
module.exports.create = create;
|
||||||
|
module.exports.createServer = sqlite3.createServer;
|
||||||
|
module.exports.createMasterClient = sqlite3.createMasterClient;
|
||||||
|
|
|
@ -0,0 +1,8 @@
|
||||||
|
var cluster = require('cluster');
|
||||||
|
|
||||||
|
if (cluster.isMaster) {
|
||||||
|
require('./master');
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
require('./worker');
|
||||||
|
}
|
|
@ -0,0 +1,51 @@
|
||||||
|
'use strict';
|
||||||
|
|
||||||
|
var cluster = require('cluster');
|
||||||
|
var minCores = 3;
|
||||||
|
var numCores = Math.max(minCores, require('os').cpus().length);
|
||||||
|
var i;
|
||||||
|
var sock = '/tmp/foobar.sqlite3-cluster.test.sock';
|
||||||
|
|
||||||
|
function run(connect, ipcKey) {
|
||||||
|
var sqlite3 = require('../../cluster');
|
||||||
|
|
||||||
|
return sqlite3.create({
|
||||||
|
verbose: null
|
||||||
|
, standalone: null
|
||||||
|
, serve: sock
|
||||||
|
, connect: null
|
||||||
|
, sock: sock
|
||||||
|
, ipcKey: ipcKey
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
var ipcKey = require('crypto').randomBytes(16).toString('hex');
|
||||||
|
// not a bad idea to setup the master before forking the workers
|
||||||
|
|
||||||
|
run(false, ipcKey).then(function () {
|
||||||
|
var w;
|
||||||
|
|
||||||
|
function setupWorker(w) {
|
||||||
|
function sendKey() {
|
||||||
|
w.send({ ipcKey: ipcKey, sock: sock });
|
||||||
|
}
|
||||||
|
w.on('online', sendKey);
|
||||||
|
}
|
||||||
|
|
||||||
|
for (i = 1; i <= numCores; i += 1) {
|
||||||
|
w = cluster.fork();
|
||||||
|
setupWorker(w);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
process.on('beforeExit', function () {
|
||||||
|
console.log("[MASTER] I've got nothing left to live for... ugh... death is upon me...");
|
||||||
|
});
|
||||||
|
// The native Promise implementation ignores errors because... dumbness???
|
||||||
|
process.on('unhandledRejection', function (err) {
|
||||||
|
console.error('Unhandled Promise Rejection');
|
||||||
|
console.error(err);
|
||||||
|
console.error(err.stack);
|
||||||
|
|
||||||
|
process.exit(1);
|
||||||
|
});
|
|
@ -0,0 +1,118 @@
|
||||||
|
'use strict';
|
||||||
|
|
||||||
|
var cluster = require('cluster');
|
||||||
|
|
||||||
|
console.log("[Worker #" + cluster.worker.id + "]");
|
||||||
|
|
||||||
|
function testSelect(client) {
|
||||||
|
var PromiseA = require('bluebird');
|
||||||
|
|
||||||
|
return new PromiseA(function (resolve, reject) {
|
||||||
|
client.run('CREATE TABLE IF NOT EXISTS meta (version TEXT)', function (err) {
|
||||||
|
if (err) {
|
||||||
|
if ('E_NO_INIT' === err.code) {
|
||||||
|
console.error(err.message);
|
||||||
|
resolve(client);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
reject(err);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
client.get("SELECT version FROM meta", [], function (err, result) {
|
||||||
|
if (err) {
|
||||||
|
console.error('[ERROR] create table', cluster.isMaster && '0' || cluster.worker.id);
|
||||||
|
console.error(err.stack || err);
|
||||||
|
reject(err);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
console.log('[this] Worker #', cluster.isMaster && '0' || cluster.worker.id);
|
||||||
|
console.log(this);
|
||||||
|
|
||||||
|
console.log('[result] Worker #', cluster.isMaster && '0' || cluster.worker.id);
|
||||||
|
console.log(result);
|
||||||
|
|
||||||
|
resolve(client);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
function run(ipcKey, sock) {
|
||||||
|
var sqlite3 = require('../../cluster');
|
||||||
|
|
||||||
|
return sqlite3.create({
|
||||||
|
bits: 128
|
||||||
|
, dirname: '/tmp/'
|
||||||
|
, prefix: 'foobar.'
|
||||||
|
, dbname: 'cluster'
|
||||||
|
, suffix: '.test'
|
||||||
|
, ext: '.sqlcipher'
|
||||||
|
, verbose: null
|
||||||
|
, standalone: null
|
||||||
|
, serve: null
|
||||||
|
, connect: sock
|
||||||
|
, sock: sock
|
||||||
|
, ipcKey: ipcKey
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
function init(client) {
|
||||||
|
//console.log('[INIT] begin');
|
||||||
|
return client.init({
|
||||||
|
algorithm: 'aes'
|
||||||
|
, bits: 128
|
||||||
|
, mode: 'cbc'
|
||||||
|
, key: '00000000000000000000000000000000'
|
||||||
|
}).then(function (/*args*/) {
|
||||||
|
//console.log('[INIT]', args);
|
||||||
|
return client;
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
function onMessage(msg) {
|
||||||
|
function loseTheWillToLive() {
|
||||||
|
process.removeListener('message', onMessage);
|
||||||
|
// child processes do not exit when their event loop is empty
|
||||||
|
setTimeout(function () {
|
||||||
|
console.log('#' + cluster.worker.id + ' dead');
|
||||||
|
process.exit(0);
|
||||||
|
}, 100);
|
||||||
|
}
|
||||||
|
|
||||||
|
console.log('New Worker', cluster.worker.id, msg);
|
||||||
|
if (1 === cluster.worker.id) {
|
||||||
|
//setTimeout(function () {
|
||||||
|
run(msg.ipcKey, msg.sock).then(testSelect).then(function (client) {
|
||||||
|
setTimeout(function () {
|
||||||
|
console.log('init worker closing...');
|
||||||
|
client.close(loseTheWillToLive);
|
||||||
|
loseTheWillToLive();
|
||||||
|
}, 1000);
|
||||||
|
// waiting for https://github.com/websockets/ws/issues/613 to land
|
||||||
|
});
|
||||||
|
//}, 1000);
|
||||||
|
} else {
|
||||||
|
setTimeout(function () {
|
||||||
|
run(msg.ipcKey, msg.sock).then(init).then(testSelect).then(function (client) {
|
||||||
|
console.log('#' + cluster.worker.id + ' other working closing...');
|
||||||
|
client.close(loseTheWillToLive);
|
||||||
|
loseTheWillToLive();
|
||||||
|
});
|
||||||
|
}, 100 * cluster.worker.id);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
process.on('message', onMessage);
|
||||||
|
|
||||||
|
// The native Promise implementation ignores errors because... dumbness???
|
||||||
|
process.on('beforeExit', function () {
|
||||||
|
console.log("[WORKER] I've got nothing left to do");
|
||||||
|
});
|
||||||
|
process.on('unhandledRejection', function (err) {
|
||||||
|
console.error('Unhandled Promise Rejection');
|
||||||
|
console.error(err.stack || err);
|
||||||
|
|
||||||
|
process.exit(1);
|
||||||
|
});
|
|
@ -0,0 +1,8 @@
|
||||||
|
var cluster = require('cluster');
|
||||||
|
|
||||||
|
if (cluster.isMaster) {
|
||||||
|
require('./master');
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
require('./worker');
|
||||||
|
}
|
|
@ -0,0 +1,48 @@
|
||||||
|
'use strict';
|
||||||
|
|
||||||
|
var cluster = require('cluster');
|
||||||
|
var minCores = 3;
|
||||||
|
var numCores = Math.max(minCores, require('os').cpus().length);
|
||||||
|
var i;
|
||||||
|
var sock = '/tmp/foobar.sqlite3-cluster.test.sock';
|
||||||
|
|
||||||
|
function run(connect, ipcKey) {
|
||||||
|
var sqlite3 = require('../../server');
|
||||||
|
|
||||||
|
return sqlite3.createServer({
|
||||||
|
verbose: null
|
||||||
|
, sock: sock
|
||||||
|
, ipcKey: ipcKey
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
var ipcKey = require('crypto').randomBytes(16).toString('hex');
|
||||||
|
// not a bad idea to setup the master before forking the workers
|
||||||
|
|
||||||
|
run(false, ipcKey).then(function () {
|
||||||
|
var w;
|
||||||
|
|
||||||
|
function setupWorker(w) {
|
||||||
|
function sendKey() {
|
||||||
|
w.send({ ipcKey: ipcKey, sock: sock });
|
||||||
|
}
|
||||||
|
w.on('online', sendKey);
|
||||||
|
}
|
||||||
|
|
||||||
|
for (i = 1; i <= numCores; i += 1) {
|
||||||
|
w = cluster.fork();
|
||||||
|
setupWorker(w);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
process.on('beforeExit', function () {
|
||||||
|
console.log("[MASTER] I've got nothing left to live for... ugh... death is upon me...");
|
||||||
|
});
|
||||||
|
// The native Promise implementation ignores errors because... dumbness???
|
||||||
|
process.on('unhandledRejection', function (err) {
|
||||||
|
console.error('Unhandled Promise Rejection');
|
||||||
|
console.error(err);
|
||||||
|
console.error(err.stack);
|
||||||
|
|
||||||
|
process.exit(1);
|
||||||
|
});
|
|
@ -0,0 +1,115 @@
|
||||||
|
'use strict';
|
||||||
|
|
||||||
|
var cluster = require('cluster');
|
||||||
|
|
||||||
|
console.log("[Worker #" + cluster.worker.id + "]");
|
||||||
|
|
||||||
|
function testSelect(client) {
|
||||||
|
var PromiseA = require('bluebird');
|
||||||
|
|
||||||
|
return new PromiseA(function (resolve, reject) {
|
||||||
|
client.run('CREATE TABLE IF NOT EXISTS meta (version TEXT)', function (err) {
|
||||||
|
if (err) {
|
||||||
|
if ('E_NO_INIT' === err.code) {
|
||||||
|
console.error(err.message);
|
||||||
|
resolve(client);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
reject(err);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
client.get("SELECT version FROM meta", [], function (err, result) {
|
||||||
|
if (err) {
|
||||||
|
console.error('[ERROR] create table', cluster.isMaster && '0' || cluster.worker.id);
|
||||||
|
console.error(err.stack || err);
|
||||||
|
reject(err);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
console.log('Worker #', (cluster.isMaster && '0' || cluster.worker.id), '[this], [result]:');
|
||||||
|
console.log(this);
|
||||||
|
console.log(result);
|
||||||
|
|
||||||
|
resolve(client);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
function run(clientFactory) {
|
||||||
|
return clientFactory.create({
|
||||||
|
init: false
|
||||||
|
, dbname: 'factory.' + cluster.worker.id
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
function init(client) {
|
||||||
|
return client.init({
|
||||||
|
algorithm: 'aes'
|
||||||
|
, bits: 128
|
||||||
|
, mode: 'cbc'
|
||||||
|
, key: '00000000000000000000000000000000'
|
||||||
|
}).then(function (/*db*/) {
|
||||||
|
return client;
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
function onMessage(msg) {
|
||||||
|
function loseTheWillToLive() {
|
||||||
|
process.removeListener('message', onMessage);
|
||||||
|
// child processes do not exit when their event loop is empty
|
||||||
|
setTimeout(function () {
|
||||||
|
console.log('#' + cluster.worker.id + ' dead');
|
||||||
|
process.exit(0);
|
||||||
|
}, 100);
|
||||||
|
}
|
||||||
|
|
||||||
|
var clientFactory = require('../../client').createClientFactory({
|
||||||
|
algorithm: 'aes'
|
||||||
|
, bits: 128
|
||||||
|
, mode: 'cbc'
|
||||||
|
, dirname: '/tmp/'
|
||||||
|
, prefix: 'foobar.'
|
||||||
|
//, dbname: 'cluster'
|
||||||
|
, suffix: '.test'
|
||||||
|
, ext: '.sqlcipher'
|
||||||
|
, sock: msg.sock
|
||||||
|
, ipcKey: msg.ipcKey
|
||||||
|
});
|
||||||
|
|
||||||
|
console.log('New Worker', cluster.worker.id, msg);
|
||||||
|
if (1 === cluster.worker.id) {
|
||||||
|
//setTimeout(function () {
|
||||||
|
run(clientFactory).then(testSelect).then(function (client) {
|
||||||
|
setTimeout(function () {
|
||||||
|
console.log('init worker closing...');
|
||||||
|
client.close(loseTheWillToLive);
|
||||||
|
loseTheWillToLive();
|
||||||
|
}, 1000);
|
||||||
|
// waiting for https://github.com/websockets/ws/issues/613 to land
|
||||||
|
});
|
||||||
|
//}, 1000);
|
||||||
|
} else {
|
||||||
|
setTimeout(function () {
|
||||||
|
run(clientFactory).then(init).then(testSelect).then(function (client) {
|
||||||
|
console.log('#' + cluster.worker.id + ' other working closing...');
|
||||||
|
client.close(loseTheWillToLive);
|
||||||
|
loseTheWillToLive();
|
||||||
|
});
|
||||||
|
}, 100 * cluster.worker.id);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
process.on('message', onMessage);
|
||||||
|
|
||||||
|
// The native Promise implementation ignores errors because... dumbness???
|
||||||
|
process.on('beforeExit', function () {
|
||||||
|
console.log("[WORKER] I've got nothing left to do");
|
||||||
|
});
|
||||||
|
process.on('unhandledRejection', function (err) {
|
||||||
|
console.error('Unhandled Promise Rejection');
|
||||||
|
console.error(err.stack || err);
|
||||||
|
|
||||||
|
process.exit(1);
|
||||||
|
});
|
149
server.js
149
server.js
|
@ -4,33 +4,25 @@ var PromiseA = require('bluebird').Promise;
|
||||||
var wsses = {};
|
var wsses = {};
|
||||||
|
|
||||||
function createApp(servers, options) {
|
function createApp(servers, options) {
|
||||||
if (wsses[options.sock]) {
|
|
||||||
return PromiseA.resolve(wsses[options.sock]);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
var url = require('url');
|
var url = require('url');
|
||||||
|
var wss = servers.wss;
|
||||||
|
//var server = servers.server;
|
||||||
|
|
||||||
//var express = require('express');
|
//var express = require('express');
|
||||||
//var app = express();
|
//var app = express();
|
||||||
var wss = servers.wss;
|
/*
|
||||||
var server = servers.server;
|
|
||||||
|
|
||||||
function app(req, res) {
|
function app(req, res) {
|
||||||
res.end('NOT IMPLEMENTED');
|
res.end('NOT IMPLEMENTED');
|
||||||
}
|
}
|
||||||
|
*/
|
||||||
|
|
||||||
wss.on('connection', function (ws) {
|
wss.on('connection', function (ws) {
|
||||||
if (!wss.__count) {
|
|
||||||
wss.__count = 0;
|
|
||||||
}
|
|
||||||
wss.__count += 1;
|
|
||||||
|
|
||||||
var location = url.parse(ws.upgradeReq.url, true);
|
var location = url.parse(ws.upgradeReq.url, true);
|
||||||
// you might use location.query.access_token to authenticate or share sessions
|
// you might use location.query.access_token to authenticate or share sessions
|
||||||
// or ws.upgradeReq.headers.cookie (see http://stackoverflow.com/a/16395220/151312
|
// or ws.upgradeReq.headers.cookie (see http://stackoverflow.com/a/16395220/151312
|
||||||
|
|
||||||
if (!options.ipcKey) {
|
if (!options.ipcKey) {
|
||||||
console.warn("[SECURITY] please include { ipcKey: crypto.randomBytes(16).toString('base64') }"
|
console.warn("[S] [SECURITY] please include { ipcKey: crypto.randomBytes(16).toString('base64') }"
|
||||||
+ " in your options and pass it from master to worker processes with worker.send()");
|
+ " in your options and pass it from master to worker processes with worker.send()");
|
||||||
ws._authorized = true;
|
ws._authorized = true;
|
||||||
} else {
|
} else {
|
||||||
|
@ -43,13 +35,33 @@ function createApp(servers, options) {
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
ws.on('close', function () {
|
if (!wss.__count) {
|
||||||
|
wss.__count = 0;
|
||||||
|
}
|
||||||
|
wss.__count += 1;
|
||||||
|
|
||||||
|
function decrWs() {
|
||||||
wss.__count -= 1;
|
wss.__count -= 1;
|
||||||
if (!wss.__count) {
|
if (!wss.__count) {
|
||||||
wss.close();
|
console.log('[S] client count is zero, but server will be left open');
|
||||||
server.close();
|
/*
|
||||||
}
|
wss.close(function () {
|
||||||
|
console.log('wss.closed');
|
||||||
});
|
});
|
||||||
|
server.close(function () {
|
||||||
|
console.log('server closed, but will not exit due to bug? in wss.close()');
|
||||||
|
process.exit(0);
|
||||||
|
});
|
||||||
|
*/
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
ws.on('error', function (err) {
|
||||||
|
console.error('[S] [WebSocket error]');
|
||||||
|
console.error(err.stack);
|
||||||
|
decrWs();
|
||||||
|
});
|
||||||
|
ws.on('close', decrWs);
|
||||||
ws.on('message', function (buffer) {
|
ws.on('message', function (buffer) {
|
||||||
var cmd;
|
var cmd;
|
||||||
var promise;
|
var promise;
|
||||||
|
@ -57,19 +69,22 @@ function createApp(servers, options) {
|
||||||
try {
|
try {
|
||||||
cmd = JSON.parse(buffer.toString('utf8'));
|
cmd = JSON.parse(buffer.toString('utf8'));
|
||||||
} catch(e) {
|
} catch(e) {
|
||||||
console.error('[ERROR] parse json');
|
console.error('[S] [ERROR] parse json');
|
||||||
console.error(e);
|
console.error(e.stack || e);
|
||||||
console.error(buffer);
|
console.error(buffer);
|
||||||
console.error();
|
console.error();
|
||||||
ws.send(JSON.stringify({ type: 'error', value: { message: e.message, code: "E_PARSE_JSON" } }));
|
ws.send(JSON.stringify({ type: 'error', value: { message: e.message, code: "E_PARSE_JSON" } }));
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
//console.log('cmd');
|
||||||
|
//console.log(cmd);
|
||||||
// caching and create logic happens in the wrapper stored here below
|
// caching and create logic happens in the wrapper stored here below
|
||||||
promise = require('./wrapper').create(options, cmd).then(function (db) {
|
promise = require('./wrapper').create(cmd && cmd.dbname && cmd || options).then(function (db) {
|
||||||
|
|
||||||
switch(cmd.type) {
|
switch(cmd.type) {
|
||||||
case 'init':
|
case 'init':
|
||||||
|
console.log('[S] init', cmd);
|
||||||
db[cmd.func].apply(db, cmd.args).then(function () {
|
db[cmd.func].apply(db, cmd.args).then(function () {
|
||||||
var args = Array.prototype.slice.call(arguments);
|
var args = Array.prototype.slice.call(arguments);
|
||||||
var myself;
|
var myself;
|
||||||
|
@ -90,13 +105,13 @@ function createApp(servers, options) {
|
||||||
break;
|
break;
|
||||||
|
|
||||||
case 'rpc':
|
case 'rpc':
|
||||||
if (!db._initialized) {
|
if ('close' !== cmd.func && !db._initialized) {
|
||||||
//console.log('[RPC NOT HAPPENING]');
|
//console.log('[RPC NOT HAPPENING]');
|
||||||
ws.send(JSON.stringify({
|
ws.send(JSON.stringify({
|
||||||
type: 'error'
|
type: 'error'
|
||||||
, id: cmd.id
|
, id: cmd.id
|
||||||
, args: [{ message: 'database has not been initialized' }]
|
, args: [{ message: 'database has not been initialized', code: 'E_NO_INIT' }]
|
||||||
, error: { message: 'database has not been initialized' }
|
, error: { message: 'database has not been initialized', code: 'E_NO_INIT' }
|
||||||
}));
|
}));
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
@ -132,37 +147,77 @@ function createApp(servers, options) {
|
||||||
});
|
});
|
||||||
});
|
});
|
||||||
|
|
||||||
//app.masterClient = db;
|
// wsses[options.sock] = app;
|
||||||
wsses[options.sock] = app;
|
return PromiseA.resolve();
|
||||||
|
}
|
||||||
|
|
||||||
return PromiseA.resolve(app);
|
function newSocket(options) {
|
||||||
|
if (wsses[options.sock]) {
|
||||||
|
return PromiseA.resolve(wsses[options.sock]);
|
||||||
|
}
|
||||||
|
|
||||||
|
wsses[options.sock] = new PromiseA(function (resolve) {
|
||||||
|
var fs = require('fs');
|
||||||
|
fs.unlink(options.sock, function () {
|
||||||
|
var server = require('http').createServer();
|
||||||
|
// ignore error when socket doesn't exist
|
||||||
|
server.listen(options.sock, function () {
|
||||||
|
resolve(server);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
}).then(function (server) {
|
||||||
|
var WebSocketServer = require('ws').Server;
|
||||||
|
var servers = {
|
||||||
|
server: require('http').createServer()
|
||||||
|
, wss: new WebSocketServer({ server: server })
|
||||||
|
};
|
||||||
|
|
||||||
|
return createApp(servers, options).then(function (/*app*/) {
|
||||||
|
// server.on('request', app);
|
||||||
|
wsses[options.sock] = servers;
|
||||||
|
|
||||||
|
return wsses[options.sock];
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
return wsses[options.sock];
|
||||||
|
}
|
||||||
|
|
||||||
|
function createMasterClient(options) {
|
||||||
|
return require('./wrapper').create(options, null).then(function (db) {
|
||||||
|
return db;
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
function createServer(options) {
|
||||||
|
if (!options.sock) {
|
||||||
|
throw new Error("Please provide options.sock as the socket to serve from");
|
||||||
|
}
|
||||||
|
options.server = options.sock;
|
||||||
|
|
||||||
|
return newSocket(options).then(function () {
|
||||||
|
var result = {};
|
||||||
|
|
||||||
|
Object.keys(wsses[options.sock]).forEach(function (key) {
|
||||||
|
result[key] = wsses[options.sock][key];
|
||||||
|
});
|
||||||
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
function create(options) {
|
function create(options) {
|
||||||
var server = require('http').createServer();
|
return createServer(options).then(function (result) {
|
||||||
var WebSocketServer = require('ws').Server;
|
if (!options.dbname) {
|
||||||
var wss = new WebSocketServer({ server: server });
|
return result;
|
||||||
//var port = process.env.PORT || process.argv[0] || 4080;
|
}
|
||||||
|
|
||||||
var fs = require('fs');
|
return createMasterClient(options).then(function (db) {
|
||||||
var ps = [];
|
result.masterClient = db;
|
||||||
|
|
||||||
ps.push(new PromiseA(function (resolve) {
|
return result;
|
||||||
fs.unlink(options.sock, function () {
|
|
||||||
// ignore error when socket doesn't exist
|
|
||||||
|
|
||||||
server.listen(options.sock, resolve);
|
|
||||||
});
|
});
|
||||||
}));
|
|
||||||
|
|
||||||
ps.push(createApp({ server: server, wss: wss }, options).then(function (app) {
|
|
||||||
server.on('request', app);
|
|
||||||
return { masterClient: app.masterClient || true };
|
|
||||||
}));
|
|
||||||
|
|
||||||
return PromiseA.all(ps).then(function (results) {
|
|
||||||
return results[1];
|
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
module.exports.create = create;
|
module.exports.create = create;
|
||||||
|
module.exports.createMasterClient = createMasterClient;
|
||||||
|
module.exports.createServer = createServer;
|
||||||
|
|
99
wrapper.js
99
wrapper.js
|
@ -1,6 +1,7 @@
|
||||||
'use strict';
|
'use strict';
|
||||||
|
|
||||||
var sqlite3 = require('sqlite3');
|
var sqlite3 = require('sqlite3');
|
||||||
|
// TODO expire unused dbs from cache
|
||||||
var dbs = {};
|
var dbs = {};
|
||||||
|
|
||||||
function sanitize(str) {
|
function sanitize(str) {
|
||||||
|
@ -11,52 +12,31 @@ function create(opts, verbs) {
|
||||||
if (!verbs) {
|
if (!verbs) {
|
||||||
verbs = {};
|
verbs = {};
|
||||||
}
|
}
|
||||||
var db;
|
|
||||||
var PromiseA = verbs.Promise || require('bluebird');
|
|
||||||
|
|
||||||
if (!opts) {
|
if (!opts) {
|
||||||
opts = {};
|
opts = {};
|
||||||
}
|
}
|
||||||
|
|
||||||
if (opts.verbose) {
|
var db;
|
||||||
sqlite3.verbose();
|
var PromiseA = verbs.Promise || require('bluebird');
|
||||||
}
|
|
||||||
|
|
||||||
// TODO expire unused dbs from cache
|
|
||||||
var dbname = "";
|
var dbname = "";
|
||||||
if (opts.dirname) {
|
|
||||||
dbname += opts.dirname;
|
dbname += (opts.dirname || '');
|
||||||
}
|
dbname += (opts.prefix || '');
|
||||||
if (opts.prefix) {
|
|
||||||
dbname += opts.prefix;
|
|
||||||
}
|
|
||||||
if (opts.subtenant) {
|
if (opts.subtenant) {
|
||||||
dbname += opts.subtenant + '.';
|
dbname += opts.subtenant + '.';
|
||||||
}
|
}
|
||||||
if (opts.tenant) {
|
if (opts.tenant) {
|
||||||
dbname += opts.tenant + '.';
|
dbname += opts.tenant + '.';
|
||||||
}
|
}
|
||||||
if (opts.dbname) {
|
dbname += (opts.dbname || '');
|
||||||
dbname += opts.dbname;
|
dbname += (opts.suffix || '');
|
||||||
}
|
dbname += (opts.ext || '');
|
||||||
if (opts.suffix) {
|
|
||||||
dbname += opts.suffix;
|
function initDb(newOpts) {
|
||||||
}
|
if (dbs[dbname].initPromise) {
|
||||||
if (opts.ext) {
|
return dbs[dbname].initPromise;
|
||||||
dbname += opts.ext;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if (dbs[dbname]) {
|
|
||||||
return PromiseA.resolve(dbs[dbname]);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
db = new sqlite3.Database(dbname);
|
|
||||||
// dbs[dbname] = db //
|
|
||||||
db.sanitize = sanitize;
|
|
||||||
db.escape = sanitize;
|
|
||||||
|
|
||||||
db.init = function (newOpts) {
|
|
||||||
if (!newOpts) {
|
if (!newOpts) {
|
||||||
newOpts = {};
|
newOpts = {};
|
||||||
}
|
}
|
||||||
|
@ -64,18 +44,19 @@ function create(opts, verbs) {
|
||||||
var key = newOpts.key || opts.key;
|
var key = newOpts.key || opts.key;
|
||||||
var bits = newOpts.bits || opts.bits;
|
var bits = newOpts.bits || opts.bits;
|
||||||
|
|
||||||
return new PromiseA(function (resolve, reject) {
|
dbs[dbname].initPromise = new PromiseA(function (resolve, reject) {
|
||||||
if (db._initialized) {
|
if (dbs[dbname].db._initialized) {
|
||||||
dbs[dbname] = db;
|
|
||||||
resolve(db);
|
resolve(db);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!key) {
|
if (!key) {
|
||||||
if (!bits) {
|
if (!bits) {
|
||||||
db._initialized = true;
|
//console.log("INITIALIZED WITHOUT KEY");
|
||||||
|
//console.log(opts);
|
||||||
|
dbs[dbname].db._initialized = true;
|
||||||
}
|
}
|
||||||
dbs[dbname] = db;
|
dbs[dbname].db = db;
|
||||||
resolve(db);
|
resolve(db);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
@ -107,16 +88,48 @@ function create(opts, verbs) {
|
||||||
|
|
||||||
PromiseA.all(setup).then(function () {
|
PromiseA.all(setup).then(function () {
|
||||||
// restore original functions
|
// restore original functions
|
||||||
db._initialized = true;
|
dbs[dbname].db._initialized = true;
|
||||||
dbs[dbname] = db;
|
dbs[dbname].db = db;
|
||||||
|
|
||||||
resolve(db);
|
resolve(db);
|
||||||
}, reject);
|
}, reject);
|
||||||
});
|
});
|
||||||
});
|
});
|
||||||
};
|
|
||||||
|
|
||||||
dbs[dbname] = db.init(opts);
|
return dbs[dbname].initPromise;
|
||||||
return dbs[dbname];
|
}
|
||||||
|
|
||||||
|
function newDb() {
|
||||||
|
// dbs[dbname] = db //
|
||||||
|
db = new sqlite3.Database(dbname);
|
||||||
|
db.init = initDb;
|
||||||
|
db.sanitize = sanitize;
|
||||||
|
db.escape = sanitize;
|
||||||
|
|
||||||
|
if (opts.verbose) {
|
||||||
|
sqlite3.verbose();
|
||||||
|
}
|
||||||
|
|
||||||
|
return db;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Could be any of:
|
||||||
|
// * db object
|
||||||
|
// * init promise
|
||||||
|
|
||||||
|
if (!dbs[dbname]) {
|
||||||
|
dbs[dbname] = { db: newDb() };
|
||||||
|
}
|
||||||
|
|
||||||
|
if (dbs[dbname].db._initialized) {
|
||||||
|
return PromiseA.resolve(dbs[dbname].db);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (opts.init || ('init' === opts.type) || (opts.bits && opts.key)) {
|
||||||
|
dbs[dbname].initPromise = db.init(opts);
|
||||||
|
}
|
||||||
|
|
||||||
|
return dbs[dbname].initPromise || PromiseA.resolve(dbs[dbname].db);
|
||||||
}
|
}
|
||||||
|
|
||||||
module.exports.sanitize = sanitize;
|
module.exports.sanitize = sanitize;
|
||||||
|
|
Loading…
Reference in New Issue