Initial work on the server

2026-07-24 12:50:10 +08:00 · 2013-03-25 06:08:51 +05:30
parent 93400554fd
commit 0485272d93
626 changed files with 121311 additions and 1 deletions
@@ -0,0 +1,164 @@
+var Net              = require('net');
+var ConnectionConfig = require('./ConnectionConfig');
+var Pool             = require('./Pool');
+var Protocol         = require('./protocol/Protocol');
+var SqlString        = require('./protocol/SqlString');
+var Query            = require('./protocol/sequences/Query');
+var EventEmitter     = require('events').EventEmitter;
+var Util             = require('util');
+
+module.exports = Connection;
+Util.inherits(Connection, EventEmitter);
+function Connection(options) {
+  EventEmitter.call(this);
+
+  this.config = options.config;
+
+  this._socket        = options.socket;
+  this._protocol      = new Protocol({config: this.config, connection: this});
+  this._connectCalled = false;
+}
+
+Connection.createQuery = function(sql, values, cb) {
+  if (sql instanceof Query) {
+    return sql;
+  }
+
+  var options = {};
+
+  if (typeof sql === 'object') {
+    // query(options, cb)
+    options = sql;
+    cb      = values;
+  } else if (typeof values === 'function') {
+    // query(sql, cb)
+    cb             = values;
+    options.sql    = sql;
+    options.values = undefined;
+  } else {
+    // query(sql, values, cb)
+    options.sql    = sql;
+    options.values = values;
+  }
+  return new Query(options, cb);
+};
+
+Connection.prototype.connect = function(cb) {
+  if (!this._connectCalled) {
+    this._connectCalled = true;
+
+    this._socket = (this.config.socketPath)
+      ? Net.createConnection(this.config.socketPath)
+      : Net.createConnection(this.config.port, this.config.host);
+
+    this._socket.pipe(this._protocol);
+    this._protocol.pipe(this._socket);
+
+    this._socket.on('error', this._handleNetworkError.bind(this));
+    this._protocol.on('unhandledError', this._handleProtocolError.bind(this));
+    this._protocol.on('drain', this._handleProtocolDrain.bind(this));
+    this._protocol.on('end', this._handleProtocolEnd.bind(this));
+  }
+
+  this._protocol.handshake(cb);
+};
+
+Connection.prototype.changeUser = function(options, cb){
+  this._implyConnect();
+
+  if (typeof options === 'function') {
+    cb      = options;
+    options = {};
+  }
+
+  var charsetNumber = (options.charset)
+    ? Config.getCharsetNumber(options.charset)
+    : this.config.charsetNumber;
+
+  return this._protocol.changeUser({
+    user          : options.user || this.config.user,
+    password      : options.password || this.config.password,
+    database      : options.database || this.config.database,
+    charsetNumber : charsetNumber,
+    currentConfig : this.config
+  }, cb);
+};
+
+Connection.prototype.query = function(sql, values, cb) {
+  this._implyConnect();
+
+  var query = Connection.createQuery(sql, values, cb);
+
+  if (!(typeof sql == 'object' && 'typeCast' in sql)) {
+    query.typeCast = this.config.typeCast;
+  }
+
+  query.sql = this.format(query.sql, query.values || []);
+  delete query.values;
+
+  return this._protocol._enqueue(query);
+};
+
+Connection.prototype.ping = function(cb) {
+  this._implyConnect();
+  this._protocol.ping(cb);
+};
+
+Connection.prototype.statistics = function(cb) {
+  this._implyConnect();
+  this._protocol.stats(cb);
+};
+
+Connection.prototype.end = function(cb) {
+  this._implyConnect();
+  this._protocol.quit(cb);
+};
+
+Connection.prototype.destroy = function() {
+  this._implyConnect();
+  this._socket.destroy();
+  this._protocol.destroy();
+};
+
+Connection.prototype.pause = function() {
+  this._socket.pause();
+  this._protocol.pause();
+};
+
+Connection.prototype.resume = function() {
+  this._socket.resume();
+  this._protocol.resume();
+};
+
+Connection.prototype.escape = function(value) {
+  return SqlString.escape(value, false, this.config.timezone);
+};
+
+Connection.prototype.format = function(sql, values) {
+  if (typeof this.config.queryFormat == "function") {
+    return this.config.queryFormat.call(this, sql, values, this.config.timezone);
+  }
+  return SqlString.format(sql, values, this.config.timezone);
+};
+
+Connection.prototype._handleNetworkError = function(err) {
+  this._protocol.handleNetworkError(err);
+};
+
+Connection.prototype._handleProtocolError = function(err) {
+  this.emit('error', err);
+};
+
+Connection.prototype._handleProtocolDrain = function() {
+  this.emit('drain');
+};
+
+Connection.prototype._handleProtocolEnd = function(err) {
+  this.emit('end', err);
+};
+
+Connection.prototype._implyConnect = function() {
+  if (!this._connectCalled) {
+    this.connect();
+  }
+};