mm_mysql 1.9.1 → 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,74 +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时返回执行结果
-		 */
-		this.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();
-					var bl;
-					if (sql_str !== '') {
-						await this.run(sql_str + ";");
-						if (this.error) {
-							errors.push({
-								sql_str: sql_str.slice(0, 512),
-								error: this.error
-							});
-							bl = false;
-							// 如果数据库链接失败(密码错误或无法连接)，则退出循环
-							if (this.error.code == 1045 || this.error.code == 2003) {
-								break;
-							}
-						} else {
-							bl = true;
-						}
+/**
+ * 导入数据库
+ * @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, bl, this.error, sql_str);
+				}
+			}
+			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) {
-				$.log.error("导入SQL文件失败！", err);
+				errors.push({
+					table: tableName,
+					error: err.message
+				});
 			}
-			return {
-				index,
-				count,
-				progress,
-				errors
+
+			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,
@@ -297,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;
@@ -312,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);
@@ -323,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.1",
-	"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

@@ -14,12 +14,32 @@ async function test() {
 		multipleStatements: true
 	});
 	await sql.open();
-	var file = "./data.sql".fullname(__dirname);
-	var ret = await sql.load(file, function(progress, index, bl, error, sql_str) {
-		console.log(progress, index, bl, error, sql_str);
-	});
+	// // 导入
+	// 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();
-	console.log("执行结果", ret);
+
 }
 test();