142 lines
2.9 KiB
JavaScript
142 lines
2.9 KiB
JavaScript
'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;
|