webio-node/libs/mysql_cls.js
2017-03-23 16:42:12 +08:00

104 lines
2.5 KiB
JavaScript

var mysql = require('mysql');
class database {
constructor() {
this._user = '';
this._password = '';
this._host = '';
this._port = 3306;
this._database = '';
this._socketPath = '/var/lib/mysql/mysql.sock'
this._con = null;
this.autoclose = false;
}
connect() {
if (!this._user || !this._host || !this._password || !this._database) throw 'mysql connect arg is empty';
this._con = mysql.createConnection({
user: this._user,
password: this._password,
host: this._host,
database: this._database,
port: this._port,
socketPath: this._socketPath
});
}
checkConnect() {
if (!this._con) {
this.connect();
}
}
formatQuery(query, arg) {
return mysql.format(query, arg);
}
escape(val) {
return mysql.escape(val);
}
query(query, params, cb) {
this.checkConnect();
let self = this;
if (typeof params == 'function') {
cb = params;
params = [];
};
if (typeof params == 'object' && !Array.isArray(params)) params = [];
this._con.query(query, params, (err, rows, field) => {
if (self.autoclose) {
self.close(() => {
cb.apply(this, [err, rows, field]);
});
} else {
cb.apply(this, [err, rows, field]);
}
});
}
recordPage(rows, page, maxPage) {
if (!page || !isFinite(page) || page < 1) page = 1;
let totalpage = Math.ceil(rows / maxPage);
let prevpage = page - 1;
let nextpage = page + 1;
if (prevpage < 1) prevpage = 1;
if (nextpage > totalpage) nextpage = totalpage;
let rec_start = (page - 1) * maxPage + 1
let rec_end = (rec_start + maxPage - 1);
if (rec_end > rows) rec_end = rows;
let json = {
rec_start,
rec_end,
total: rows,
prevpage,
nextpage,
totalpage,
page
};
return json;
}
close(cb) {
let self = this;
this._con.end((err) => {
self._con = null;
if (cb) cb();
});
}
set user(str) { this._user = str; }
set host(str) { this._host = str; }
set password(str) { this._password = str; }
set port(str) { this._port = str; }
set database(str) { this._database = str; }
}
module.exports = database;