mm_mysql 1.9.0 → 1.9.2

package/index.js

@@ -11,6 +11,7 @@ const {
 } = require('./db');
 
 const Link_model = require('./link_model');
+const { table } = require('console');
 
 var pool = {};
 
@@ -80,7 +81,7 @@ class Mysql {
 		 * @param {Array} val 替换值
 		 * @return {Promise|Array} 异步构造器, 当await时返回执行结果
 		 */
-		this.run = function(sql, val) {
+		this.run = function (sql, val) {
 			var _this = this;
 			this.sql = sql;
 			if ($this.config.log) {
@@ -89,7 +90,7 @@ class Mysql {
 
 			// 返回一个 Promise
 			return new Promise((resolve, reject) => {
-				$this.conn.getConnection(function(err, db) {
+				$this.conn.getConnection(function (err, db) {
 					if (err) {
 						_this.error = {
 							code: 2003,
@@ -131,7 +132,7 @@ class Mysql {
 		 * @param {Array} val 替换值
 		 * @return {Promise|Array} 异步构造器, 当await时返回执行结果
 		 */
-		this.exec = function(sql, val) {
+		this.exec = function (sql, val) {
 			var _this = this;
 			if (this.task) {
 				this.task_sql += sql + "\r\n";
@@ -153,7 +154,7 @@ class Mysql {
 			}
 			// 返回一个 Promise
 			return new Promise((resolve, reject) => {
-				$this.conn.getConnection(function(err, db) {
+				$this.conn.getConnection(function (err, db) {
 					if (err) {
 						_this.error = {
 							code: 41000,
@@ -182,7 +183,7 @@ class Mysql {
 								if (o.constructor == Array) {
 									if (o.length > 0) {
 										var num = 0;
-										o.map(function(item) {
+										o.map(function (item) {
 											num += item['affectedRows'];
 										});
 										if (num === 0) {
@@ -213,21 +214,169 @@ class Mysql {
 		/**
 		 * @description 获取数据库管理器
 		 */
-		this.db = function() {
+		this.db = function () {
 			return new DB($this);
 		};
 	}
 }
 
+/**
+ * 导入数据库
+ * @param {Object} file sql文件
+ * @param {Function} func 回调函数
+ * @return {Promise} 异步构造器, 当await时返回执行结果
+ */
+Mysql.prototype.load = async function (file, func) {
+	var count = 0;
+	var progress = 0;
+	var errors = [];
+	var index = 0;
+	try {
+		var data = file.loadText();
+		// 将SQL文件内容分割成单独的语句
+		const arr = data.split(';\r\n');
+		count = arr.length;
+		for (var i = 0; i < arr.length; i++) {
+			var sql_str = arr[i].trim();
+			if (sql_str !== '') {
+				await this.run(sql_str + ";");
+				if (this.error) {
+					errors.push({
+						sql_str: sql_str.slice(0, 512),
+						error: this.error
+					});
+					// 如果数据库链接失败(密码错误或无法连接)，则退出循环
+					if (this.error.code == 1045 || this.error.code == 2003) {
+						break;
+					}
+				}
+			}
+			var p = i / arr.length;
+			progress = Math.ceil(p * 100);
+			index = i;
+			if (func) {
+				func(progress, index, this.error, sql_str);
+			}
+		}
+	} catch (err) {
+		$.log.error("导入SQL文件失败！", err);
+	}
+	return {
+		index,
+		count,
+		progress,
+		errors
+	}
+};
+
+/**
+ * 保存数据库为文件
+ * @param {String} file 文件对象
+ * @param {Function} func 回调函数
+ * @return {Promise} 异步构造器, 当await时返回执行结果
+ */
+Mysql.prototype.save = async function (file, func, tables = []) {
+	const fs = require('fs');
+	const stream = fs.createWriteStream(file);
+	let count = 0;
+	let progress = 0;
+	let errors = [];
+	let index = 0;
+	let tableCount = 0;
+
+	try {
+		// 开始导出数据库
+		stream.write('SET FOREIGN_KEY_CHECKS = 0;\r\n\r\n');
+		count++;
+
+		if(!tables.length) {
+			// 获取所有表
+			var tbs = await this.run('SHOW TABLES');
+			tables = tbs.map((item) => {
+				return item[Object.keys(item)[0]];
+			});
+		}
+		
+		tableCount = tables.length;
+		console.log("tables", tables);
+		for (var i = 0; i < tables.length; i++) {
+			var tableName = tables[i];
+			try {
+				// 导出表结构
+				const createTable = await this.run(`SHOW CREATE TABLE ${tableName}`);
+				stream.write(`-- 表结构: ${tableName}\r\n`);
+				stream.write(`DROP TABLE IF EXISTS \`${tableName}\`;\r\n`);
+				count++;
+				stream.write(`${createTable[0]['Create Table']};\r\n\r\n`);
+				count++;
+				// 导出表数据
+				const rows = await this.run(`SELECT * FROM ${tableName}`);
+				if (rows.length > 0) {
+					stream.write(`-- 表数据: ${tableName}\r\n`);
+					for (const row of rows) {
+						const rowValues = Object.values(row)
+							.map(value => {
+								if (value === null) return 'NULL';
+								if (typeof value === 'boolean') return value ? 1 : 0;
+								if (typeof value === 'number') return value;
+								if (value instanceof Date) {
+									return "'" + value.toStr('yyyy-MM-dd hh:mm:ss') + "'";
+								}
+								// 处理字符串，转义特殊字符
+								return "'" + value.toString()
+									.replace(/[\\']/g, '\\$&')
+									.replace(/\n/g, '\\n')
+									.replace(/\r/g, '\\r')
+									.replace(/\t/g, '\\t')
+									.replace(/\u0000/g, '\\0') + "'";
+							})
+							.join(',');
+						stream.write(`INSERT INTO ${tableName} VALUES (${rowValues});\r\n`);
+						count++;
+					}
+					stream.write('\r\n');
+				}
+			} catch (err) {
+				errors.push({
+					table: tableName,
+					error: err.message
+				});
+			}
+
+			index++;
+			progress = Math.ceil((index / tableCount) * 100);
+			if (func) {
+				func(progress, index, this.error, tableName);
+			}
+		}
+
+		// 完成导出
+		stream.write('SET FOREIGN_KEY_CHECKS = 1;\r\n');
+		count++;
+		stream.end();
+	} catch (err) {
+		$.log.error("导出SQL文件失败！", err);
+		stream.end();
+	}
+
+	return {
+		tableCount,
+		index,
+		count,
+		progress,
+		errors
+	}
+};
+
 /**
  * 获取数据库管理器
  * @param {String} key 主键
- * @param {String|Number} value 对象值 
+ * @param {String|Number} value 对象值
  * @param {Boolean} clear_prefix 清除前缀
  * @param {Array} arr_table 关联的数据表
  * @return {Object} 管理模型
  */
-Mysql.prototype.dbs = async function(key, value, clear_prefix, ...arr_table) {
+Mysql.prototype.dbs = async function (key, value, clear_prefix, ...arr_table) {
 	var lm = new Link_model({
 		key,
 		value,
@@ -244,9 +393,9 @@ Mysql.prototype.dbs = async function(key, value, clear_prefix, ...arr_table) {
  * 设置配置参数
  * @param {Object} cg 配置对象或配置路径
  */
-Mysql.prototype.setConfig = function(cg) {
+Mysql.prototype.setConfig = function (cg) {
 	var obj;
-	if (typeof(cg) === "string") {
+	if (typeof (cg) === "string") {
 		obj = cg.loadJson(this.dir);
 	} else {
 		obj = cg;
@@ -259,7 +408,7 @@ Mysql.prototype.setConfig = function(cg) {
  * @description 打开数据库, 如果没有则建立数据库连接再打开
  * @param {boolean} 是否重置
  */
-Mysql.prototype.open = function(reset) {
+Mysql.prototype.open = function (reset) {
 	if (reset || !this.state || !pool[this.identifier]) {
 		this.state = 1;
 		pool[this.identifier] = createPool(this.config);
@@ -270,7 +419,7 @@ Mysql.prototype.open = function(reset) {
 /**
  * @description 关闭连接
  */
-Mysql.prototype.close = function() {
+Mysql.prototype.close = function () {
 	if (pool[this.identifier]) {
 		pool[this.identifier].end();
 		pool[this.identifier] = null;

package/package.json

@@ -1,37 +1,33 @@
-{
-	"name": "mm_mysql",
-	"version": "1.9.0",
-	"description": "这是超级美眉mysql帮助函数模块，用于便捷操作mysql，使用await方式，可以避免嵌套函数",
-	"main": "index.js",
-	"dependencies": {
-		"mm_logs": "^1.1.6",
-		"mysql": "^2.18.1"
-	},
-	"scripts": {
-		"test": "node test.js",
-		"dev": "nodemon test.js"
-	},
-	"author": "邱文武",
-	"license": "ISC",
-	"repository": {
-		"type": "git",
-		"url": "git+https://github.com/qiuwenwu/mm_mysql.git"
-	},
-	"keywords": [
-		"mysql",
-		"async",
-		"await",
-		"promise",
-		"add",
-		"del",
-		"set",
-		"get",
-		"query",
-		"run",
-		"exec"
-	],
-	"bugs": {
-		"url": "https://github.com/qiuwenwu/mm_mysql/issues"
-	},
-	"homepage": "https://github.com/qiuwenwu/mm_mysql#readme"
-}
+{
+  "name": "mm_mysql",
+  "version": "1.9.2",
+  "description": "这是超级美眉mysql帮助函数模块，用于便捷操作mysql，使用await方式，可以避免嵌套函数",
+  "main": "index.js",
+  "scripts": {
+    "test": "node test.js",
+    "dev": "nodemon test.js"
+  },
+  "repository": {
+    "type": "git",
+    "url": "https://gitee.com/qiuwenwu91/mm_mysql.git"
+  },
+  "keywords": [
+    "mysql",
+    "async",
+    "await",
+    "promise",
+    "add",
+    "del",
+    "set",
+    "get",
+    "query",
+    "run",
+    "exec"
+  ],
+  "author": "qww",
+  "license": "ISC",
+  "dependencies": {
+    "mm_logs": "^1.1.7",
+    "mysql": "^2.18.1"
+  }
+}

package/test.js

@@ -1,5 +1,47 @@
 const Mysql = require('./index.js').Mysql;
 
+/**
+ * 测试导入sql文件
+ */
+async function test() {
+	var sql = new Mysql();
+	sql.setConfig({
+		database: 'test',
+		// 连接用户名
+		user: "root",
+		// 连接密码
+		password: "Asd159357",
+		multipleStatements: true
+	});
+	await sql.open();
+	// // 导入
+	// var ret = await sql.load("./data.sql".fullname(__dirname), function(progress, index, error, sql_str) {
+	// 	console.log("导入", progress, index, error, sql_str);
+	// });
+	// console.log("导入结果", ret);
+	
+	// 导出指定表'user_account', 'user_info'
+	// var ret2 = await sql.save("./bat.sql".fullname(__dirname), function(progress, index, error, sql_str) {
+	// 	console.log("导出", progress, index, error, sql_str);
+	// }, ['user_account', 'user_info']);
+	// console.log("导出结果", ret2);
+	
+	// // 导出
+	// var ret2 = await sql.save("./bat.sql".fullname(__dirname), function(progress, index, error, sql_str) {
+	// 	// console.log("导出", progress, index, error, sql_str);
+	// });
+	// console.log("导出结果", ret2);
+
+	// // 导入
+	// var ret = await sql.load("./bat.sql".fullname(__dirname), function(progress, index, error, sql_str) {
+	// 	// console.log("导入", progress, index, error, sql_str);
+	// });
+	// console.log("导入结果", ret);
+
+	sql.close();
+
+}
+test();
 
 // function test() {
 // 	var setSql = "`nickname`='hao,1,2,3',`gm`=5,`mc`=2,`group`='2,3'";
@@ -14,42 +56,42 @@ const Mysql = require('./index.js').Mysql;
 // });
 // sql.open();
 
-var tpl = {
-	"query": {
-		"state_min": "`state` >= '{0}'",
-		"state_max": "`state` <= '{0}'",
-		"vip_min": "`vip` >= '{0}'",
-		"vip_max": "`vip` <= '{0}'",
-		"gm_min": "`gm` >= '{0}'",
-		"gm_max": "`gm` <= '{0}'",
-		"mc_min": "`mc` >= '{0}'",
-		"mc_max": "`mc` <= '{0}'",
-		"create_time_min": "`create_time` >= '{0}'",
-		"create_time_max": "`create_time` <= '{0}'",
-		"login_time_min": "`login_time` >= '{0}'",
-		"login_time_max": "`login_time` <= '{0}'",
-		"salt": "`salt` like '%{0}%'",
-		"invite_code": "`invite_code` like '%{0}%'",
-		"phone": "`phone` like '%{0}%'",
-		"username": "`username` like '%{0}%'",
-		"nickname": "`nickname` like '%{0}%'",
-		"password": "`password` like '%{0}%'",
-		"email": "`email` like '%{0}%'",
-		"user_group": "`user_group` like '%{0}%'",
-		"user_admin": "`user_admin` like '%{0}%'",
-		"login_ip": "`login_ip` like '%{0}%'",
-		"signature": "`signature` like '%{0}%'",
-		"avatar": "`avatar` like '%{0}%'",
-		"friends": "`friends` like '%{0}%'",
-		"admin_group": "`admin_group` like '%{0}%'"
-	},
-	"update": {
-		"state_add": "`state` = `state` + '{0}'",
-		"vip_add": "`vip` = `vip` + '{0}'",
-		"gm_add": "`gm` = `gm` + '{0}'",
-		"mc_add": "`mc` = `mc` + '{0}'"
-	}
-};
+// var tpl = {
+// 	"query": {
+// 		"state_min": "`state` >= '{0}'",
+// 		"state_max": "`state` <= '{0}'",
+// 		"vip_min": "`vip` >= '{0}'",
+// 		"vip_max": "`vip` <= '{0}'",
+// 		"gm_min": "`gm` >= '{0}'",
+// 		"gm_max": "`gm` <= '{0}'",
+// 		"mc_min": "`mc` >= '{0}'",
+// 		"mc_max": "`mc` <= '{0}'",
+// 		"create_time_min": "`create_time` >= '{0}'",
+// 		"create_time_max": "`create_time` <= '{0}'",
+// 		"login_time_min": "`login_time` >= '{0}'",
+// 		"login_time_max": "`login_time` <= '{0}'",
+// 		"salt": "`salt` like '%{0}%'",
+// 		"invite_code": "`invite_code` like '%{0}%'",
+// 		"phone": "`phone` like '%{0}%'",
+// 		"username": "`username` like '%{0}%'",
+// 		"nickname": "`nickname` like '%{0}%'",
+// 		"password": "`password` like '%{0}%'",
+// 		"email": "`email` like '%{0}%'",
+// 		"user_group": "`user_group` like '%{0}%'",
+// 		"user_admin": "`user_admin` like '%{0}%'",
+// 		"login_ip": "`login_ip` like '%{0}%'",
+// 		"signature": "`signature` like '%{0}%'",
+// 		"avatar": "`avatar` like '%{0}%'",
+// 		"friends": "`friends` like '%{0}%'",
+// 		"admin_group": "`admin_group` like '%{0}%'"
+// 	},
+// 	"update": {
+// 		"state_add": "`state` = `state` + '{0}'",
+// 		"vip_add": "`vip` = `vip` + '{0}'",
+// 		"gm_add": "`gm` = `gm` + '{0}'",
+// 		"mc_add": "`mc` = `mc` + '{0}'"
+// 	}
+// };
 
 // // 测试模板修改
 // async function test_tpl_set() {
@@ -76,112 +118,112 @@ var tpl = {
 // }
 // test_tpl_set();
 
-async function test() {
-	console.log('测试开始')
-	var sql = new Mysql();
-	var db = sql.db();
-	// 是否输出语法
-	// sql.config.log = false;
-	// 忽视那些错误的输出，例如[1062]，忽视主键重复插入
-	// sql.config.log_ignore = [];
-
-	sql.setConfig({
-		"user": "root",
-		"password": "Asd159357"
-	});
-	sql.open();
-	// sql.close();
-	// sql.open();
+// async function test() {
+// 	console.log('测试开始')
+// 	var sql = new Mysql();
+// 	var db = sql.db();
+// 	// 是否输出语法
+// 	// sql.config.log = false;
+// 	// 忽视那些错误的输出，例如[1062]，忽视主键重复插入
+// 	// sql.config.log_ignore = [];
 
-	// --- 测试事件 ---
-	$.eventer.on("mysql_add_before:user_account", function(json) {
-		console.log("添加前", json);
-	});
-	$.eventer.on("mysql_add_after:user_account", function(json) {
-		console.log("添加后", json);
-	});
-	$.eventer.on("mysql_del_before:user_account", function(json) {
-		console.log("删除前", json);
-	});
-	$.eventer.on("mysql_del_after:user_account", function(json) {
-		console.log("删除后", json);
-	});
-	$.eventer.on("mysql_set_before:user_account", function(json) {
-		console.log("修改前", json);
-	});
-	$.eventer.on("mysql_set_after:user_account", function(json) {
-		console.log("修改后", json);
-	});
-	$.eventer.on("mysql_addOrSet_before:user_account", function(json) {
-		console.log("添加或修改前", json);
-	});
-	$.eventer.on("mysql_addOrSet_after:user_account", function(json) {
-		console.log("添加或修改后", json);
-	});
-	$.eventer.on("mysql_get_before:user_account", function(json) {
-		console.log("查询列表前", json);
-	});
-	$.eventer.on("mysql_get_after:user_account", function(json) {
-		console.log("查询列表后", json);
-	});
-	$.eventer.on("mysql_getObj_before:user_account", function(json) {
-		console.log("查询对象前", json);
-	});
-	$.eventer.on("mysql_getObj_after:user_account", function(json) {
-		console.log("查询对象后", json);
-	});
+// 	sql.setConfig({
+// 		"user": "root",
+// 		"password": "Asd159357"
+// 	});
+// 	sql.open();
+// 	// sql.close();
+// 	// sql.open();
 
-	db.table = "user_account";
-	db.key = "user_id,username";
-	var obj = await db.getObj({
-		user_id_has: "1"
-	});
-	obj.vip += 1;
+// 	// --- 测试事件 ---
+// 	$.eventer.on("mysql_add_before:user_account", function(json) {
+// 		console.log("添加前", json);
+// 	});
+// 	$.eventer.on("mysql_add_after:user_account", function(json) {
+// 		console.log("添加后", json);
+// 	});
+// 	$.eventer.on("mysql_del_before:user_account", function(json) {
+// 		console.log("删除前", json);
+// 	});
+// 	$.eventer.on("mysql_del_after:user_account", function(json) {
+// 		console.log("删除后", json);
+// 	});
+// 	$.eventer.on("mysql_set_before:user_account", function(json) {
+// 		console.log("修改前", json);
+// 	});
+// 	$.eventer.on("mysql_set_after:user_account", function(json) {
+// 		console.log("修改后", json);
+// 	});
+// 	$.eventer.on("mysql_addOrSet_before:user_account", function(json) {
+// 		console.log("添加或修改前", json);
+// 	});
+// 	$.eventer.on("mysql_addOrSet_after:user_account", function(json) {
+// 		console.log("添加或修改后", json);
+// 	});
+// 	$.eventer.on("mysql_get_before:user_account", function(json) {
+// 		console.log("查询列表前", json);
+// 	});
+// 	$.eventer.on("mysql_get_after:user_account", function(json) {
+// 		console.log("查询列表后", json);
+// 	});
+// 	$.eventer.on("mysql_getObj_before:user_account", function(json) {
+// 		console.log("查询对象前", json);
+// 	});
+// 	$.eventer.on("mysql_getObj_after:user_account", function(json) {
+// 		console.log("查询对象后", json);
+// 	});
 
-	await db.add(obj);
-	console.log("尝试重复插入", db.error);
+// 	db.table = "user_account";
+// 	db.key = "user_id,username";
+// 	var obj = await db.getObj({
+// 		user_id_has: "1"
+// 	});
+// 	obj.vip += 1;
 
-	// obj.vip -= 1;
-	console.log("语法", db.sql, obj);
+// 	await db.add(obj);
+// 	console.log("尝试重复插入", db.error);
 
-	obj.nickname = '管理员';
-	// obj.vip -= 1;
-	console.log("语法2", db.sql, obj);
+// 	// obj.vip -= 1;
+// 	console.log("语法", db.sql, obj);
 
-	var list = await db.get({
-		user_id_has: "1,2,a-3,4,5"
-	});
-	console.log("语法", db.sql, list.length);
+// 	obj.nickname = '管理员';
+// 	// obj.vip -= 1;
+// 	console.log("语法2", db.sql, obj);
 
-	var list = await db.get({
-		user_id: "1"
-	});
-	console.log("查询语法", db.sql, list.length);
+// 	var list = await db.get({
+// 		user_id_has: "1,2,a-3,4,5"
+// 	});
+// 	console.log("语法", db.sql, list.length);
 
-	var list = await db.get({
-		user_id: "1"
-	}, null, null, true);
-	console.log("模糊查询语法", db.sql, list.length);
+// 	var list = await db.get({
+// 		user_id: "1"
+// 	});
+// 	console.log("查询语法", db.sql, list.length);
 
-	var obj = await db.getObj({
-		user_id_has: "1,2,a-3,4,5"
-	});
-	console.log("语法", db.sql, obj);
+// 	var list = await db.get({
+// 		user_id: "1"
+// 	}, null, null, true);
+// 	console.log("模糊查询语法", db.sql, list.length);
 
-	var bl = await db.addOrSet({
-		user_id: obj.user_id
-	}, obj);
-	console.log("语法", db.sql, bl);
-	
-	var obj = await db.getObj({}, "user_id desc");
-	var obj_new = Object.assign({}, obj);
-	obj_new.user_id += 1;
-	var bl = await db.addOrSet({
-		user_id: obj_new.user_id
-	}, obj_new);
-	console.log("语法", db.sql, bl);
-}
-test();
+// 	var obj = await db.getObj({
+// 		user_id_has: "1,2,a-3,4,5"
+// 	});
+// 	console.log("语法", db.sql, obj);
+
+// 	var bl = await db.addOrSet({
+// 		user_id: obj.user_id
+// 	}, obj);
+// 	console.log("语法", db.sql, bl);
+
+// 	var obj = await db.getObj({}, "user_id desc");
+// 	var obj_new = Object.assign({}, obj);
+// 	obj_new.user_id += 1;
+// 	var bl = await db.addOrSet({
+// 		user_id: obj_new.user_id
+// 	}, obj_new);
+// 	console.log("语法", db.sql, bl);
+// }
+// test();
 // setInterval(() => {
 // 	test();
 // }, 3000)