lib/dialects/sqlite3/runner.js

Runner

module.exports = function(client) { var Promise = require('../../promise'); var Runner = require('../../runner'); var helpers = require('../../helpers'); var inherits = require('inherits');

Inherit from the Runner constructor's prototype, so we can add the correct then method.

function Runner_SQLite3() { this.client = client; Runner.apply(this, arguments); } inherits(Runner_SQLite3, Runner); Runner_SQLite3.prototype._beginTransaction = 'begin transaction;';

Runs the query on the specified connection, providing the bindings and any other necessary prep work.

Runner_SQLite3.prototype._query = Promise.method(function(obj) { var method = obj.method; if (this.isDebugging()) this.debug(obj); var callMethod = (method === 'insert' || method === 'update' || method === 'del') ? 'run' : 'all'; var connection = this.connection; return new Promise(function(resolver, rejecter) { if (!connection || !connection[callMethod]) { return rejecter(new Error('Error calling ' + callMethod + ' on connection.')); } connection[callMethod](obj.sql, obj.bindings, function(err, response) { if (err) return rejecter(err); obj.response = response;

We need the context here, as it contains the "this.lastID" or "this.changes"

obj.context = this; return resolver(obj); }); }); });

Grab a connection, run the query via the MySQL streaming interface, and pass that through to the stream we've sent back to the client.

Runner_SQLite3.prototype._stream = Promise.method(function(sql, stream, options) { var runner = this; return new Promise(function(resolver, rejecter) { stream.on('error', rejecter); stream.on('end', resolver); runner.connection.each(sql.sql, sql.bindings, function(err, row) { if (err) return rejecter(err); stream.write(row); }, function() { stream.end(); }); }); });

Ensures the response is returned in the same format as other clients.

Runner_SQLite3.prototype.processResponse = function(obj) { var ctx = obj.context; var response = obj.response; if (obj.output) return obj.output.call(this, response); if (obj.method === 'select') { response = helpers.skim(response); } else if (obj.method === 'insert') { response = [ctx.lastID]; } else if (obj.method === 'del' || obj.method === 'update') { response = ctx.changes; } return response; };

Assign the newly extended Runner constructor to the client object.

client.Runner = Runner_SQLite3; };