mm_sqlite 1.0.5 → 1.0.6

package/index_/345/217/202/350/200/203.js

@@ -0,0 +1,299 @@
+/**
+ * @fileOverview Mysql帮助类函数
+ * @author <a href="http://qww.elins.cn">邱文武</a>
+ * @version 1.2
+ */
+const {
+	createPool
+} = require('mysql');
+const {
+	DB
+} = require('./db');
+
+const Link_model = require('./link_model');
+
+var pool = {};
+
+/**
+ * @description 数据库封装
+ */
+class Mysql {
+	/**
+	 * @description 创建Mysql帮助类函数 (构造函数)
+	 * @param {String} scope 作用域
+	 * @param {String} dir 当前路径
+	 * @constructor
+	 */
+	constructor(scope, dir) {
+		// 作用域
+		this.scope;
+		if (scope) {
+			this.scope = scope;
+		} else {
+			this.scope = $.val.scope + '';
+		}
+		// 当前目录
+		this.dir = __dirname;
+		if (dir) {
+			this.dir = dir;
+		}
+		// 错误提示
+		this.error;
+		/**
+		 * sql语句
+		 */
+		this.sql = "";
+		// 连接器
+		this.conn;
+
+		// 数据库配置参数
+		this.config = {
+			// 服务器地址
+			host: "127.0.0.1",
+			// 端口号
+			port: 3306,
+			// 连接用户名
+			user: "root",
+			// 连接密码
+			password: "asd123",
+			// 数据库
+			database: "mm",
+			// 是否支持多个sql语句同时操作
+			multipleStatements: false,
+			// 打印SQL
+			log: false
+		};
+
+		// 唯一标识符
+		this.identifier = this.config.host + "/" + this.config.database;
+
+		// 定义当前类, 用于数据库实例化访问
+		var $this = this;
+
+		/**
+		 * @description 查询sql
+		 * @param {String} sql 查询参
+		 * @param {Array} val 替换值
+		 * @return {Promise|Array} 异步构造器, 当await时返回执行结果
+		 */
+		this.run = function(sql, val) {
+			var _this = this;
+			this.sql = sql;
+			if ($this.config.log) {
+				$.log.debug("SQL：", sql);
+			}
+
+			// 返回一个 Promise
+			return new Promise((resolve, reject) => {
+				$this.conn.getConnection(function(err, db) {
+					if (err) {
+						_this.error = {
+							code: 2003,
+							message: $.info(err).between('Error: ', '\n')
+						};
+						$.log.error("SQL error：", err);
+						resolve([]);
+						reject(err);
+					} else {
+						db.query(sql, val, (err, rows) => {
+							// 结束会话
+							db.release();
+							if (err) {
+								_this.error = {
+									code: err.errno,
+									message: err.sqlMessage
+								};
+								$.log.error("SQL error：", err);
+								resolve([]);
+								reject(err);
+							} else {
+								_this.error = undefined;
+								resolve(rows);
+							}
+						});
+					}
+				});
+			});
+		};
+
+		/**
+		 * @description 增删改sql
+		 * @param {String} sql 查询参
+		 * @param {Array} val 替换值
+		 * @return {Promise|Array} 异步构造器, 当await时返回执行结果
+		 */
+		this.exec = function(sql, val) {
+			var _this = this;
+			if (this.task) {
+				this.task_sql += sql + "\r\n";
+				if (this.task === 1) {
+					return this.task_sql;
+				} else if (this.task === 2) {
+					this.task = 0;
+					sql = this.task_sql.trim();
+					this.task_sql = '';
+				} else {
+					this.task = 0;
+					this.task_sql = '';
+					return;
+				}
+			}
+			this.sql = sql;
+			if ($this.config.log) {
+				$.log.debug("SQL：", sql);
+			}
+			// $this.conn.config.connectionConfig = false;
+			// console.log("测试", $this.conn.config.connectionConfig);
+			// 返回一个 Promise
+			return new Promise((resolve, reject) => {
+				$this.conn.getConnection(function(err, db) {
+					if (err) {
+						_this.error = {
+							code: 41000,
+							message: $.info(err).between('Error: ', '\n')
+						};
+						resolve(-1);
+						reject(err);
+						$.log.error("SQL error：", err);
+					} else {
+						db.query(sql, val, (err, o) => {
+							if (err) {
+								_this.error = {
+									code: err.errno,
+									message: err.sqlMessage
+								};
+								resolve(-1);
+								reject(err);
+								$.log.error("SQL error：", err);
+							} else {
+								_this.error = undefined;
+								if (o.constructor == Array) {
+									if (o.length > 0) {
+										var num = 0;
+										o.map(function(item) {
+											num += item['affectedRows'];
+										});
+										if (num === 0) {
+											resolve(o.length);
+										} else {
+											resolve(num);
+										}
+									} else {
+										resolve(1);
+									}
+								} else {
+									var num = o['affectedRows'];
+									if (num === 0) {
+										resolve(1);
+									} else {
+										resolve(num);
+									}
+								}
+							}
+							// 结束会话
+							db.release();
+						});
+					}
+				});
+			});
+		};
+
+		/**
+		 * @description 获取数据库管理器
+		 */
+		this.db = function() {
+			return new DB($this);
+		};
+	}
+}
+
+/**
+ * 获取数据库管理器
+ * @param {String} key 主键
+ * @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) {
+	var lm = new Link_model({
+		key,
+		value,
+		clear_prefix
+	});
+	lm.sql = this;
+	for (var i = 0; i < arr_table.length; i++) {
+		await lm.add(arr_table[i]);
+	}
+	return lm;
+};
+
+/**
+ * 设置配置参数
+ * @param {Object} cg 配置对象或配置路径
+ */
+Mysql.prototype.setConfig = function(cg) {
+	var obj;
+	if (typeof(cg) === "string") {
+		obj = cg.loadJson(this.dir);
+	} else {
+		obj = cg;
+	}
+	$.push(this.config, obj);
+	this.identifier = this.config.host + "/" + this.config.database;
+};
+
+/**
+ * @description 打开数据库, 如果没有则建立数据库连接再打开
+ */
+Mysql.prototype.open = function() {
+	if (!pool[this.identifier]) {
+		pool[this.identifier] = createPool(this.config);
+	}
+	this.conn = pool[this.identifier];
+};
+
+/**
+ * @description 关闭连接
+ */
+Mysql.prototype.close = function() {
+	if (pool[this.identifier]) {
+		pool[this.identifier] = null;
+	}
+};
+
+/**
+ * @description 导出Mysql帮助类
+ */
+exports.Mysql = Mysql;
+
+
+/**
+ * @description mysql连接池
+ */
+if (!$.pool.mysql) {
+	$.pool.mysql = {};
+}
+
+/**
+ * @description 缓存管理器，用于创建缓存
+ * @param {String} scope 作用域
+ * @param {String} dir 当前路径
+ * @return {Object} 返回一个缓存类
+ */
+function mysql_admin(scope, dir) {
+	if (!scope) {
+		scope = $.val.scope
+	}
+	var obj = $.pool.mysql[scope];
+	if (!obj) {
+		$.pool.mysql[scope] = new Mysql(scope, dir);
+		obj = $.pool.mysql[scope];
+	}
+	return obj;
+}
+
+/**
+ * @description 导出Mysql管理函数
+ */
+exports.mysql_admin = mysql_admin;

package/link_model.js

@@ -0,0 +1,132 @@
+/**
+ *  连接数据库模型类
+ * @class
+ */
+class Link_model {
+	/**
+	 * 构造函数
+	 * @param {Object} config 配置参数
+	 */
+	constructor(config, sql) {
+		/**
+		 * 配置参数
+		 */
+		this.config = Object.assign({
+			key: "user_id",
+			value: 0,
+			clear_prefix: true
+		}, config);
+
+		/**
+		 * 模型集合
+		 */
+		this.model = {};
+
+		/**
+		 * 数据库集合
+		 */
+		this.dbs = {};
+
+		/**
+		 * 数据库集合
+		 */
+		this.sql = sql;
+	}
+}
+
+/**
+ * 添加对象
+ * @param {String} table 表名
+ * @return {Object}
+ */
+Link_model.prototype.add = async function(table) {
+	var {
+		key,
+		value,
+		clear_prefix
+	} = this.config;
+	var db = this.sql.db();
+	db.table = table;
+	db.key = key;
+	db.size = 1;
+	db.page = 1;
+	var query = {};
+	query[key] = value;
+	var obj = await db.getObj(query);
+	if (!obj) {
+		var n = await db.add(query);
+		if (n) {
+			obj = await db.getObj(query);
+		}
+	}
+
+	var name = table;
+	// 是否去除表前缀
+	if (clear_prefix) {
+		var arr = table.split("_");
+		if (arr.length > 1) {
+			name = table.right("_");
+		}
+	}
+	this.model[name] = obj;
+	this.dbs[table] = db;
+	return this;
+};
+
+/**
+ * 初始化
+ * @param {Object} sql 数据管理器
+ * @return {Object} 当前对象
+ */
+Link_model.prototype.init = function(sql) {
+	this.sql = sql;
+	return this;
+};
+
+/**
+ * 更新键值
+ * @param {Object} key
+ * @param {Object} value
+ * @return {Object} 当前对象
+ */
+Link_model.prototype.update = function(key, value) {
+	var m = this.model;
+	for (var k in m) {
+		var o = m[k];
+		if (o[key] !== undefined) {
+			o[key] = value;
+		}
+	}
+	return this;
+};
+
+/**
+ * 合并属性
+ * @param {Array} arr_name 当前对象
+ * @return {Object} 
+ */
+Link_model.prototype.merge = function(...arr_name) {
+	var model = {};
+	if (arr_name.length === 0) {
+		arr_name = Object.keys(this.model);
+	}
+	for (var i = 0; i < arr_name.length; i++) {
+		var name = arr_name[i];
+		var o = this.model[name];
+		if (o) {
+			for (var k in o) {
+				model[k] = o[k]
+			}
+		}
+	}
+	var _this = this;
+	return new Proxy(model, {
+		set: function(obj, prop, value) {
+			_this.update(prop, value);
+			obj[prop] = value;
+			return obj;
+		}
+	});
+};
+
+module.exports = Link_model; 

package/package.json

@@ -1,36 +1,38 @@
-{
-  "name": "mm_sqlite",
-  "version": "1.0.5",
-  "description": "这是超级美眉sqlite帮助函数模块，用于便捷操作sqlite，使用await方式，可以避免嵌套函数",
-  "main": "index.js",
-  "scripts": {
-    "test": "node test.js"
-  },
-  "repository": {
-    "type": "git",
-    "url": "git+https://github.com/qiuwenwu/mm_sqlite.git"
-  },
-  "keywords": [
-    "mysql",
-    "async",
-    "await",
-    "promise",
-    "add",
-    "del",
-    "set",
-    "get",
-    "query",
-    "run",
-    "exec"
-  ],
-  "author": "邱文武",
-  "license": "ISC",
-  "bugs": {
-    "url": "https://github.com/qiuwenwu/mm_sqlite/issues"
-  },
-  "homepage": "https://github.com/qiuwenwu/mm_sqlite#readme",
-  "dependencies": {
-    "mm_expand": "^1.3.6",
-    "sqlite3": "^4.1.0"
-  }
-}
+{
+    "name": "mm_sqlite",
+    "version": "1.0.6",
+    "description": "这是超级美眉sqlite帮助函数模块，用于便捷操作sqlite，使用await方式，可以避免嵌套函数",
+    "main": "index.js",
+    "scripts": {
+        "test": "node test.js",
+        "dev": "nodemon test.js"
+    },
+    "repository": {
+        "type": "git",
+        "url": "git+https://gitee.com/qiuwenwu91/mm_sqlite.git"
+    },
+    "keywords": [
+        "sqlite",
+        "async",
+        "await",
+        "promise",
+        "add",
+        "del",
+        "set",
+        "get",
+        "query",
+        "run",
+        "exec"
+    ],
+    "author": "邱文武",
+    "license": "ISC",
+    "bugs": {
+        "url": "https://gitee.com/qiuwenwu91/mm_sqlite/issues"
+    },
+    "homepage": "https://gitee.com/qiuwenwu91/mm_sqlite#readme",
+    "dependencies": {
+        "mm_logs": "^1.2.0",
+        "sqlite3": "^5.1.7",
+        "sqlstring": "^2.3.3"
+    }
+}