mysql-pool-crud
Version:
mysql-pool-crud
108 lines (104 loc) • 2.9 kB
JavaScript
module.exports = function (mysqlConfig) {
var mysql = require('mysql');
var pool = mysql.createPool(mysqlConfig);
var transAction = require ('./transAction')(mysqlConfig);
return function (sql, callback, isStream) {
if (typeof sql == 'object' && typeof sql.length == 'number') {
transAction(sql, callback);
} else {
pool.getConnection(function(err, connection) {
if (err) {
console.log(err);
} else {
if (sql == 'transAction') {
connection.doneCallback = function (callback) {
if (typeof callback !== 'function') {
callback = function () {};
}
if (connection.released) {
return;
}
connection.release();
connection.released = true;
callback();
};
connection.autoRollback = function (callback) {
connection.rollback(function () {
connection.doneCallback(callback);
return;
});
};
connection.autoCommit = function (callback) {
connection.commit(function () {
connection.doneCallback(callback);
return;
});
};
connection.transQuery = function (sql, callback) {
if (typeof callback != 'function') {
callback = function () {};
}
connection.query({
sql: sql,
timeout: 5000
}, function (err, results) {
if (err) {
return connection.autoRollback(function () {
console.log('MYSQL_ERROR: ' + new Date());
console.log(sql);
console.log(err);
callback(err, results);
return;
});
}
callback(err, results, sql);
});
return connection;
};
connection.beginTransaction(function(err) {
if (err) {
console.log(err);
return;
}
connection.released = false;
if (typeof callback == 'function') {
callback(connection);
}
});
return;
}
if (isStream) {
// callback(err, row, next, success);
connection.query(sql).on('error', function(err) {
console.log('MYSQL_ERROR: ' + new Date());
console.log(sql);
console.log(err);
callback(err, null, function(){}, false, sql);
}).on('fields', function(fields) {
}).on('result', function(row) {
connection.pause();
var next = function () {
connection.resume();
};
callback(null, row, next, false, sql);
}).on('end', function() {
callback(err, null, function(){}, true, sql);
});
return;
}
connection.query(sql, function(err, results) {
connection.release();
if (err) {
console.log('MYSQL_ERROR: ' + new Date());
console.log(sql);
console.log(err);
}
if (typeof callback == 'function') {
callback(err, results, sql);
}
});
}
});
}
};
};