autostart server, return wrapped
This commit is contained in:
parent
4e3b9b88f7
commit
696ac1ed1c
|
@ -23,7 +23,7 @@ var sqlite = require('sqlite3-server');
|
||||||
var opts = {
|
var opts = {
|
||||||
key: '1892d335081d8d346e556c9c3c8ff2c3'
|
key: '1892d335081d8d346e556c9c3c8ff2c3'
|
||||||
, bits: 128
|
, bits: 128
|
||||||
, storage: path.join('/tmp/authn.sqlcipher')
|
, filename: path.join('/tmp/authn.sqlcipher')
|
||||||
, verbose: false
|
, verbose: false
|
||||||
, port: 3232 // default random
|
, port: 3232 // default random
|
||||||
, forceServer: true // default false
|
, forceServer: true // default false
|
||||||
|
|
138
client.js
138
client.js
|
@ -1,3 +1,141 @@
|
||||||
'use strict';
|
'use strict';
|
||||||
|
|
||||||
|
/*global Promise*/
|
||||||
|
|
||||||
|
// TODO iterate over the prototype
|
||||||
|
// translate request / response
|
||||||
|
var sqlite3real = require('sqlite3');
|
||||||
|
|
||||||
|
/*
|
||||||
|
function createConnection(opts) {
|
||||||
|
var server = ;
|
||||||
|
|
||||||
|
return server.create(opts).then(function () {
|
||||||
|
// created and listening
|
||||||
|
});
|
||||||
|
}
|
||||||
|
*/
|
||||||
|
|
||||||
|
function startServer(opts) {
|
||||||
|
return require('./server').create(opts).then(function (server) {
|
||||||
|
// this process doesn't need to connect to itself
|
||||||
|
// through a socket
|
||||||
|
return server.masterClient;
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
function getConnection(opts) {
|
||||||
|
if (!opts.sock) {
|
||||||
|
opts.sock = opts.filename + '.sock';
|
||||||
|
}
|
||||||
|
|
||||||
|
return new Promise(function (resolve) {
|
||||||
|
setTimeout(function () {
|
||||||
|
var WebSocket = require('ws');
|
||||||
|
var ws = new WebSocket('ws+unix:' + opts.sock);
|
||||||
|
|
||||||
|
if (opts.server) {
|
||||||
|
return startServer(opts);
|
||||||
|
}
|
||||||
|
|
||||||
|
ws.on('error', function (err) {
|
||||||
|
console.error('[ERROR] ws connection failed, retrying');
|
||||||
|
console.error(err);
|
||||||
|
|
||||||
|
function retry() {
|
||||||
|
setTimeout(function () {
|
||||||
|
getConnection(opts).then(resolve);
|
||||||
|
}, 100 + (Math.random() * 250));
|
||||||
|
}
|
||||||
|
|
||||||
|
if ('ENOENT' === err.code || 'ECONNREFUSED' === err.code) {
|
||||||
|
return startServer(opts).then(function (client) {
|
||||||
|
// ws.masterClient = client;
|
||||||
|
resolve({ masterClient: client });
|
||||||
|
}, function () {
|
||||||
|
retry();
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
retry();
|
||||||
|
});
|
||||||
|
|
||||||
|
ws.on('open', function () {
|
||||||
|
resolve(ws);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
}, 100 + (Math.random() * 250));
|
||||||
|
}
|
||||||
|
|
||||||
|
function create(opts) {
|
||||||
|
// TODO maybe use HTTP POST instead?
|
||||||
|
return getConnection(opts).then(function (ws) {
|
||||||
|
if (ws.masterClient) {
|
||||||
|
return ws.masterClient;
|
||||||
|
}
|
||||||
|
|
||||||
|
var db = {};
|
||||||
|
var proto = sqlite3real.Database.prototype;
|
||||||
|
|
||||||
|
function rpc(fname, args) {
|
||||||
|
var id;
|
||||||
|
var cb;
|
||||||
|
|
||||||
|
if ('function' === typeof args[args.length - 1]) {
|
||||||
|
id = Math.random();
|
||||||
|
cb = args.pop();
|
||||||
|
}
|
||||||
|
|
||||||
|
ws.send({
|
||||||
|
type: 'rpc'
|
||||||
|
, func: fname
|
||||||
|
, args: args
|
||||||
|
, filename: opts.filename
|
||||||
|
, id: id
|
||||||
|
});
|
||||||
|
|
||||||
|
if (!cb) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
function onMessage(data) {
|
||||||
|
if (!data || 'object' !== typeof data) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (data.id !== id) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
cb.apply(data.this, data.args);
|
||||||
|
}
|
||||||
|
|
||||||
|
ws.on('message', onMessage);
|
||||||
|
}
|
||||||
|
|
||||||
|
db.sanitize = require('./wrapper').sanitize;
|
||||||
|
|
||||||
|
Object.keys(sqlite3real.Database.prototype).forEach(function (key) {
|
||||||
|
|
||||||
|
if ('function' === typeof proto[key]) {
|
||||||
|
db[key] = function () {
|
||||||
|
rpc(key, Array.prototype.slice.call(arguments));
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
});
|
||||||
|
|
||||||
|
|
||||||
|
// serialize
|
||||||
|
// parallel
|
||||||
|
db.serialize = db.parallel = function () {
|
||||||
|
throw new Error('NOT IMPLEMENTED in SQLITE3-remote');
|
||||||
|
};
|
||||||
|
|
||||||
|
return db;
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
module.exports.sanitize = require('./wrapper').sanitize;
|
||||||
|
module.exports.create = create;
|
||||||
|
|
2
index.js
2
index.js
|
@ -3,7 +3,7 @@
|
||||||
var numcpus = require('os').cpus().length;
|
var numcpus = require('os').cpus().length;
|
||||||
var sqlite3;
|
var sqlite3;
|
||||||
|
|
||||||
if (false && numcpus >= 2) {
|
if (numcpus >= 2) {
|
||||||
sqlite3 = require('./client');
|
sqlite3 = require('./client');
|
||||||
} else {
|
} else {
|
||||||
sqlite3 = require('./wrapper');
|
sqlite3 = require('./wrapper');
|
||||||
|
|
52
server.js
52
server.js
|
@ -1,10 +1,20 @@
|
||||||
'use strict';
|
'use strict';
|
||||||
|
/*global Promise*/
|
||||||
|
|
||||||
function create(options) {
|
var wsses = {};
|
||||||
|
|
||||||
|
function createApp(server, options) {
|
||||||
|
console.log('Create App');
|
||||||
|
|
||||||
|
if (wsses[options.filename]) {
|
||||||
|
return Promise.resolve(wsses[options.filename]);
|
||||||
|
}
|
||||||
|
|
||||||
|
return require('./wrapper').create(options).then(function (db) {
|
||||||
var url = require('url');
|
var url = require('url');
|
||||||
var express = require('express');
|
var express = require('express');
|
||||||
var app = express();
|
var app = express();
|
||||||
var wss = options.wss;
|
var wss = server.wss;
|
||||||
|
|
||||||
wss.on('connection', function (ws) {
|
wss.on('connection', function (ws) {
|
||||||
var location = url.parse(ws.upgradeReq.url, true);
|
var location = url.parse(ws.upgradeReq.url, true);
|
||||||
|
@ -38,17 +48,37 @@ function create(options) {
|
||||||
ws.send(JSON.stringify({ type: 'session', value: ws.__session_id }));
|
ws.send(JSON.stringify({ type: 'session', value: ws.__session_id }));
|
||||||
});
|
});
|
||||||
|
|
||||||
/*
|
app.masterClient = db;
|
||||||
var tablename = 'authn';
|
wsses[options.filename] = app;
|
||||||
if (tablename) {
|
|
||||||
setup.push(db.runAsync("CREATE TABLE IF NOT EXISTS '" + sanitize(tablename)
|
return app;
|
||||||
+ "' (id TEXT, secret TEXT, json TEXT, PRIMARY KEY(id))"));
|
});
|
||||||
}
|
}
|
||||||
*/
|
|
||||||
|
function create(options) {
|
||||||
|
console.log('Create Server');
|
||||||
|
|
||||||
/*global Promise*/
|
|
||||||
return new Promise(function (resolve) {
|
return new Promise(function (resolve) {
|
||||||
resolve(app);
|
var server = require('http').createServer();
|
||||||
|
var WebSocketServer = require('ws').Server;
|
||||||
|
var wss = new WebSocketServer({ server: server });
|
||||||
|
//var port = process.env.PORT || process.argv[0] || 4080;
|
||||||
|
|
||||||
|
console.log('options.sock');
|
||||||
|
console.log(options.sock);
|
||||||
|
var fs = require('fs');
|
||||||
|
fs.unlink(options.sock, function () {
|
||||||
|
// ignore error when socket doesn't exist
|
||||||
|
|
||||||
|
server.listen(options.sock, function () {
|
||||||
|
console.log('Listening');
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
createApp({ server: server, wss: wss }, options).then(function (app) {
|
||||||
|
server.on('request', app);
|
||||||
|
resolve({ masterClient: app.masterClient });
|
||||||
|
});
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -19,11 +19,12 @@ function create(opts) {
|
||||||
sqlite3.verbose();
|
sqlite3.verbose();
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!dbs[opts.storage] || dbs[opts.storage].__key !== opts.key) {
|
if (!dbs[opts.filename] || dbs[opts.filename].__key !== opts.key) {
|
||||||
dbs[opts.storage] = new sqlite3.Database(opts.storage);
|
dbs[opts.filename] = new sqlite3.Database(opts.filename);
|
||||||
}
|
}
|
||||||
|
|
||||||
db = dbs[opts.storage];
|
db = dbs[opts.filename];
|
||||||
|
db.sanitize = sanitize;
|
||||||
db.__key = opts.key;
|
db.__key = opts.key;
|
||||||
|
|
||||||
return new Promise(function (resolve, reject) {
|
return new Promise(function (resolve, reject) {
|
||||||
|
|
Loading…
Reference in New Issue