Added an option to return resultsets as an array type.

Author: ifsnow

Readme.md

@@ -224,6 +224,7 @@ issue [#501](https://github.com/mysqljs/mysql/issues/501). (Default: `false`)
   also possible to blacklist default ones. For more information, check
   [Connection Flags](#connection-flags).
 * `ssl`: object with ssl parameters or a string containing name of ssl profile. See [SSL options](#ssl-options).
+* `rowsAsArray`: If `true`, resultsets return as an array type instead of an object. (Default: `false`)
 
 
 In addition to passing these options as an object, you can also use a url

lib/ConnectionConfig.js

@@ -58,6 +58,8 @@ function ConnectionConfig(options) {
   // Set the client flags
   var defaultFlags = ConnectionConfig.getDefaultFlags(options);
   this.clientFlags = ConnectionConfig.mergeFlags(defaultFlags, options.flags);
+
+  this.rowsAsArray = options.rowsAsArray || false;
 }
 
 ConnectionConfig.mergeFlags = function mergeFlags(defaultFlags, userFlags) {

lib/protocol/Parser.js

@@ -21,6 +21,7 @@ function Parser(options) {
   this._nextPacketNumber  = 0;
   this._encoding          = 'utf-8';
   this._paused            = false;
+  this._globalRowsAsArray = options.config && options.config.rowsAsArray;
 }
 
 Parser.prototype.write = function write(chunk) {
@@ -410,6 +411,14 @@ Parser.prototype.packetLength = function packetLength() {
   return this._packetHeader.length + this._longPacketBuffers.size;
 };
 
+Parser.prototype.setArrayRowMode = function(rowsAsArray) {
+  this._isArrayRowMode = typeof rowsAsArray === 'undefined' ? this._globalRowsAsArray : rowsAsArray;
+};
+
+Parser.prototype.isArrayRowMode = function() {
+  return this._isArrayRowMode;
+};
+
 Parser.prototype._combineNextBuffers = function _combineNextBuffers(bytes) {
   var length = this._buffer.length - this._offset;
 

lib/protocol/packets/RowDataPacket.js

@@ -13,6 +13,22 @@ Object.defineProperty(RowDataPacket.prototype, 'parse', {
   value        : parse
 });
 
+Object.defineProperty(RowDataPacket.prototype, 'isArray', {
+  configurable : true,
+  enumerable   : false,
+  value        : function() {
+    return typeof this._row !== 'undefined';
+  }
+});
+
+Object.defineProperty(RowDataPacket.prototype, 'getArrayValue', {
+  configurable : true,
+  enumerable   : false,
+  value        : function() {
+    return this._row;
+  }
+});
+
 Object.defineProperty(RowDataPacket.prototype, '_typeCast', {
   configurable : true,
   enumerable   : false,
@@ -25,6 +41,11 @@ function parse(parser, fieldPackets, typeCast, nestTables, connection) {
     return self._typeCast(fieldPacket, parser, connection.config.timezone, connection.config.supportBigNumbers, connection.config.bigNumberStrings, connection.config.dateStrings);
   };
 
+  var isArrayRowMode = parser.isArrayRowMode();
+  if (isArrayRowMode) {
+    this._row = [];
+  }
+
   for (var i = 0; i < fieldPackets.length; i++) {
     var fieldPacket = fieldPackets[i];
     var value;
@@ -39,7 +60,9 @@ function parse(parser, fieldPackets, typeCast, nestTables, connection) {
           : parser.parseLengthCodedString() );
     }
 
-    if (typeof nestTables === 'string' && nestTables.length) {
+    if (isArrayRowMode) {
+      this._row.push(value);
+    } else if (typeof nestTables === 'string' && nestTables.length) {
       this[fieldPacket.table + nestTables + fieldPacket.name] = value;
     } else if (nestTables) {
       this[fieldPacket.table] = this[fieldPacket.table] || {};

lib/protocol/sequences/Query.js

@@ -17,6 +17,7 @@ function Query(options, callback) {
     ? true
     : options.typeCast;
   this.nestTables = options.nestTables || false;
+  this.rowsAsArray = options.rowsAsArray;
 
   this._resultSet = null;
   this._results   = [];
@@ -138,12 +139,16 @@ Query.prototype._handleFinalResultPacket = function(packet) {
 };
 
 Query.prototype['RowDataPacket'] = function(packet, parser, connection) {
+  parser.setArrayRowMode(this.rowsAsArray);
+
   packet.parse(parser, this._resultSet.fieldPackets, this.typeCast, this.nestTables, connection);
 
+  var row = packet.isArray() ? packet.getArrayValue() : packet;
+
   if (this._callback) {
-    this._resultSet.rows.push(packet);
+    this._resultSet.rows.push(row);
   } else {
-    this.emit('result', packet, this._index);
+    this.emit('result', row, this._index);
   }
 };
 

test/integration/connection/test-query-rows-as-array.js

@@ -0,0 +1,26 @@
+var assert = require('assert');
+var common = require('../../common');
+
+common.getTestConnection({ rowsAsArray: true }, function (err, connection) {
+  assert.ifError(err);
+
+  connection.query('SELECT 1', function (err, rows) {
+    assert.ifError(err);
+    assert.deepEqual(rows, [[1]]);
+  });
+
+  connection.query({ sql: 'SELECT ?' }, [ 1 ], function (err, rows) {
+    assert.ifError(err);
+    assert.deepEqual(rows, [[1]]);
+  });
+
+  connection.query({
+    sql: 'SELECT ?',
+    rowsAsArray: false
+  }, [ 1 ], function (err, rows) {
+    assert.ifError(err);
+    assert.deepEqual(rows, [{1: 1}]);
+  });
+
+  connection.end(assert.ifError);
+});