feat: wrap generator function to promise (#19)

ali-sdk · Mar 31, 2017 · fe1b4a3 · fe1b4a3
1 parent e8d8056
commit fe1b4a3
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 1 deletion.
diff --git a/lib/client.js b/lib/client.js
@@ -1,7 +1,8 @@
 'use strict';
 
-const mysql = require('mysql');
 const util = require('util');
+const mysql = require('mysql');
+const wrap = require('co-wrap-all');
 const Operator = require('./operator');
 const RDSConnection = require('./connection');
 const RDSTransaction = require('./transaction');
@@ -115,3 +116,5 @@ proto.end = function(callback) {
     });
   });
 };
+
+wrap(proto);
diff --git a/lib/transaction.js b/lib/transaction.js
@@ -5,6 +5,7 @@
  */
 
 const util = require('util');
+const wrap = require('co-wrap-all');
 const Operator = require('./operator');
 
 module.exports = RDSTransaction;
@@ -49,3 +50,5 @@ proto._check = function () {
     throw new Error('transaction was commit or rollback');
   }
 };
+
+wrap(proto);
diff --git a/package.json b/package.json
@@ -12,6 +12,7 @@
     "lint": "jshint ."
   },
   "dependencies": {
+    "co-wrap-all": "^1.0.0",
     "debug": "^2.2.0",
     "mysql": "^2.10.2"
   },