Add support for reset connection packet

dougwilson · dougwilson · commit 08edba2757df · 2014-04-08T16:07:08.000-04:00
diff --git a/lib/Connection.js b/lib/Connection.js
@@ -170,6 +170,11 @@ Connection.prototype.ping = function(cb) {
   this._protocol.ping(bindToCurrentDomain(cb));
 };
 
+Connection.prototype.reset = function reset(callback) {
+  this._implyConnect();
+  this._protocol.resetConnection(bindToCurrentDomain(callback));
+};
+
 Connection.prototype.statistics = function(cb) {
   this._implyConnect();
   this._protocol.stats(bindToCurrentDomain(cb));
diff --git a/lib/protocol/Protocol.js b/lib/protocol/Protocol.js
@@ -54,6 +54,10 @@ Protocol.prototype.ping = function(cb) {
   return this._enqueue(new Sequences.Ping(cb));
 };
 
+Protocol.prototype.resetConnection = function resetConnection(cb) {
+  return this._enqueue(new Sequences.ResetConnection(cb));
+};
+
 Protocol.prototype.stats = function(cb) {
   return this._enqueue(new Sequences.Statistics(cb));
 };
diff --git a/lib/protocol/packets/ComResetConnectionPacket.js b/lib/protocol/packets/ComResetConnectionPacket.js
@@ -0,0 +1,8 @@
+module.exports = ComResetConnectionPacket;
+
+function ComResetConnectionPacket(sql) {
+}
+
+ComResetConnectionPacket.prototype.write = function write(writer) {
+  writer.writeUnsignedNumber(1, 0x1f);
+};
diff --git a/lib/protocol/sequences/ResetConnection.js b/lib/protocol/sequences/ResetConnection.js
@@ -0,0 +1,15 @@
+var Sequence = require('./Sequence');
+var Util     = require('util');
+var Packets  = require('../packets');
+
+module.exports = ResetConnection;
+
+function ResetConnection(callback) {
+  Sequence.call(this, callback);
+}
+
+Util.inherits(ResetConnection, Sequence);
+
+ResetConnection.prototype.start = function start() {
+  this.emit('packet', new Packets.ComResetConnectionPacket);
+};
diff --git a/test/integration/connection/test-reset-connection.js b/test/integration/connection/test-reset-connection.js
@@ -0,0 +1,26 @@
+var common     = require('../../common');
+var connection = common.createConnection();
+var assert     = require('assert');
+
+connection.connect(assertError);
+
+connection.query('SET @user_var = \'apple\'', assertError);
+
+connection.query('SELECT @user_var AS var', function(err, rows) {
+  assert.ifError(err);
+  assert.equal(rows.length, 1);
+  assert.equal(rows[0].var, 'apple');
+});
+
+connection.reset(assertError);
+
+connection.query('SELECT @user_var AS var', function(err, _rows) {
+  assert.ifError(err);
+  assert.equal(rows.length, 0);
+});
+
+connection.end(assertError);
+
+function assertError(err) {
+  assert.ifError(err);
+}
