fixed reconnection problem

Conflicts: lib/adapters/sql/mysql.js
geddy · Mar 1, 2014 · b657333 · b657333
1 parent f745064
commit b657333
Showing 1 changed file with 23 additions and 2 deletions.
diff --git a/lib/adapters/sql/mysql.js b/lib/adapters/sql/mysql.js
@@ -39,19 +39,39 @@ utils.mixin(Adapter.prototype, new (function () {
 
   this.COLUMN_NAME_DELIMITER = '`';
 
+  this.handleDisconnect = function () {
+      var self = this;
+      setTimeout(function() {
+        self.client = mysql.createConnection(self.config);
+        self.connect();
+      }, Math.pow(3, reconnectAttempts) * 1000);
+      reconnectAttempts++;
+  };
+
   this.init = function () {
+    var self = this;
     this.client = mysql.createConnection(this.config);
+    this.client.on('error', function(err) {
+      if (err.code === 'PROTOCOL_CONNECTION_LOST' && reconnectAttempts < RECONNECT_TRIES_LIMIT) {
+        self.handleDisconnect();
+      };
+   });
   };
 
   this.connect = function (callback) {
     var self = this
       , cb = callback || function () {};
     this.client.connect(function (err, data) {
       if (err) {
-        self.emit('error', err);
-        cb(err);
+        if (reconnectAttempts < RECONNECT_TRIES_LIMIT) {
+          self.handleDisconnect();
+        } else {
+          self.emit('error', err);
+          cb(err);
+        }
       }
       else {
+        reconnectAttempts = 0;
         self.exec('USE ' + self.config.database, function (err, data) {
           if (err) {
             self.emit('error', err);
@@ -64,6 +84,7 @@ utils.mixin(Adapter.prototype, new (function () {
         });
       }
     });
+
   };
 
   this.disconnect = function (callback) {