mm_mongodb 1.3.8 → 1.4.1

package/cache.js

@@ -0,0 +1,359 @@
+/**
+ * @fileOverview mongodb_cache缓存帮助类函数
+ * @author <a href="http://qww.elins.cn">邱文武</a>
+ * @version 1.2
+ */
+const {
+	MongoDB
+} = require('./index.js');
+
+/* 创建Mongodb_cache帮助类函数 */
+/// 构造函数
+class Mongodb_cache extends MongoDB {
+	/**
+	 * @description 创建Mongodb_cache帮助类函数 (构造函数)
+	 * @param {String} scope 作用域
+	 * @param {String} dir 当前路径
+	 * @constructor
+	 */
+	constructor(scope, dir, config) {
+		super(scope, dir, config);
+		
+		// 消息订阅器
+		this.conn_rss;
+
+		// 订阅回调函数
+		this.func_list = {};
+	}
+}
+
+/**
+ * 消息订阅
+ * @param {String} channel 订阅的频道
+ * @param {Function} func 操作回调函数，可判断成功与否
+ */
+Mongodb_cache.prototype.subscribe = function(channel, func) {
+	this.conn_rss.subscribe(channel);
+	this.func_list[channel] = func;
+};
+
+/**
+ * 取消订阅
+ * @param {String} channel 订阅的频道
+ */
+Mongodb_cache.prototype.unsubscribe = function(channel) {
+	this.conn_rss.unsubscribe(channel);
+	delete this.func_list[channel];
+};
+
+/**
+ * 发布消息
+ * @param {String} channel 频道
+ * @param {String} message 订阅的消息
+ * @param {Function} func 消息发布成功回调函数
+ */
+Mongodb_cache.prototype.publish = function(channel, message, func) {
+	this.conn.publish(channel, message, func);
+};
+
+/**
+ * 订阅消息处理, 如果没有订阅函数，则其他的订阅消息会进入该函数
+ * @param {String} channel
+ * @param {String} message
+ */
+Mongodb_cache.prototype.main = async function(channel, message) {};
+
+/**
+ * 消息通知
+ * @param {String} channel 频道
+ * @param {String} message 消息
+ */
+Mongodb_cache.prototype.message = function(channel, message) {
+	// console.log("我接收到信息了" + message);
+	var func = this.func_list[channel];
+	if (func) {
+		func(message);
+	} else {
+		this.main(channel, message);
+	}
+};
+
+/**z
+ * @description 增加整数值(负数为减)
+ * @param {String} key 键
+ * @param {Number} num 数值
+ * @return {Promise|Number} 计算后的结果
+ */
+Mongodb_cache.prototype.addInt = function(key, num) {
+	if (num > 0) {
+		return this.conn.incrby(key, num);
+	} else {
+		return this.conn.decrby(key, -num);
+	}
+};
+
+/**
+ * @description 增加浮点数值(负数为减)
+ * @param {String} key 键
+ * @param {Number} num 数值
+ * @return {Promise|Number} 计算后的结果
+ */
+Mongodb_cache.prototype.addFloat = function(key, num) {
+	if (num > 0) {
+		return this.conn.incrbyfloat(key, num);
+	} else {
+		return this.conn.decrby(key, -num);
+	};
+};
+
+/**
+ * @description 增加字符串值到指定缓存
+ * @param {String} key 键
+ * @param {String} str 添加的字符串
+ * @return {Promise|String} 添加后的字符串
+ */
+Mongodb_cache.prototype.addStr = function(key, str) {
+	return this.conn.append(key, str);
+};
+
+/**
+ * @description 删除缓存
+ * @param {String} key 键
+ * @return {Promise|Boolean} 成功返回true,失败返回false
+ */
+Mongodb_cache.prototype.del = function(key) {
+	return this.conn.del(key);
+};
+
+/**
+ * @description 修改缓存
+ * @param {String} key 键
+ * @param {Object} value 值
+ * @param {Number} seconds 秒
+ * @return {Object} 值
+ */
+Mongodb_cache.prototype.set = async function(key, value, seconds) {
+	var ret;
+	ret = await this.conn.set(key, value);
+	if (seconds) {
+		ret = await this.conn.expire(key, seconds);
+	}
+	return ret
+};
+
+/**
+ * @description 添加缓存
+ * @param {String} key 键
+ * @param {Object} value 值
+ * @param {Number} seconds 秒
+ * @return {Promise|Object} 值
+ */
+Mongodb_cache.prototype.add = async function(key, value, seconds) {
+	var ret;
+	var has = await this.conn.exists(key);
+	if (!has) {
+		ret = await this.conn.set(key, value);
+	}
+	if (seconds) {
+		ret = await this.conn.expire(key, seconds);
+	}
+	return ret
+};
+
+/**
+ * @description 查询缓存
+ * @param {String} key 键
+ * @return {Promise|Object} 查询值
+ */
+Mongodb_cache.prototype.get = function(key) {
+	return this.conn.get(key);
+};
+
+/**
+ * @description 判断键是否存在
+ * @param {String} key 键
+ * @return {Promise|Boolean} 有返回true, 没有返回false
+ */
+Mongodb_cache.prototype.has = function(key) {
+	return this.conn.exists(key);
+};
+
+/**
+ * @description 查询缓存的字符串中的一段字符串
+ * @param {String} key 键
+ * @param {Number} start 开始位置
+ * @param {Number} end 结束位置
+ * @return {Promise|String} 查询值
+ */
+Mongodb_cache.prototype.getrange = function(key, start, end) {
+	return this.conn.getrange(key, start, end);
+};
+
+/**
+ * @description 在值的指定位置开始增加一段字符串
+ * @param {String} key 键
+ * @param {Number} index 开始位置
+ * @param {String} value 变更的值
+ * @return {Promise|Number} 字符串长度
+ */
+Mongodb_cache.prototype.setrange = function(key, index, value) {
+	return this.conn.getrange(key, index, value);
+};
+
+/**
+ * @description 清空缓存
+ * @param {String} key 键, 为空则清空所有
+ * @return {Promise|Array} 执行结果
+ */
+Mongodb_cache.prototype.clear = function(key) {
+	if (key) {
+		return this.conn.del(key);
+	} else {
+		return this.conn.flushdb();
+	}
+};
+
+/**
+ * @description 排序
+ * @param {String} key 键
+ * @param {String} way = [asc|desc]排序方式, 可以为空
+ * @param {String} obj_key 排序成员的键
+ * @return {Promise|Array} 排序后的数组
+ */
+Mongodb_cache.prototype.sort = function(key) {
+	return this.conn.sort(key);
+};
+
+/**
+ * @description 获取所有键名
+ * @param {String} key 键 支持*号, 前面加*表示后面名称一致, 前后加*表示包含名称, 后面加*表示前面名称一致
+ * @return {Promise|Array} 键数组
+ */
+Mongodb_cache.prototype.keys = function(key) {
+	if (!key) {
+		key = "*";
+	}
+	return this.conn.keys(key);
+};
+
+/**
+ * @description 修改数组缓存
+ * @param {String} key 键
+ * @param {Object} value 值
+ * @param {Number} seconds 秒
+ * @return {Promise|Array} 执行结果
+ */
+Mongodb_cache.prototype.list_set = function(key, value, seconds) {
+	return this.conn.rpush(key, value);
+};
+
+/**
+ * @description 数组缓存追加对象
+ * @param {String} key 键
+ * @param {Object} value 值
+ * @return {Promise|String} 追加后的数组
+ */
+Mongodb_cache.prototype.list_push = function(key, value) {
+	return this.conn.rpush(key, value);
+};
+
+/**
+ * @description 判断成员是否存在
+ * @param {String} key 键
+ * @param {Object} value 值
+ * @return {Promise|Boolean} 存在返回true, 否则返回false
+ */
+Mongodb_cache.prototype.list_has = async function(key, value) {
+	if (!value) {
+		throw '值不能为空';
+	}
+	var arr = await this.list_get(key);
+	var has = false;
+	var len = arr.length;
+	for (var i = 0; i < len; i++) {
+		if (value == arr[i]) {
+			has = true;
+			break;
+		}
+	}
+	return has;
+};
+
+/**
+ * @description 查询数组缓存
+ * @param {String} key 键
+ * @param {Number} start 起始位置
+ * @param {Number} end 结束位置
+ * @return {Promise|Array} 查询到的数组
+ */
+Mongodb_cache.prototype.list_get = function(key, start, end) {
+	if (start === undefined) {
+		start = 0;
+	}
+	if (end === undefined) {
+		end = -1;
+	}
+	return this.conn.lrange(key, start, end);
+};
+
+/**
+ * @description 清空数组缓存
+ * @param {String} key 键
+ * @param {Array} value 新成员, 没有则删除数组
+ * @return {Promise|Boolean} 成功返回true，是否返回false
+ */
+Mongodb_cache.prototype.list_clear = async function(key, value) {
+	var ret;
+	if (key) {
+		if (value) {
+			// 如果有值则删除其他成员，保留新值
+			try {
+				var len = await this.list_push(key, value);
+				ret = this.conn.ltrim(key, len - value.length, len - 1);
+			} catch (e) {
+
+			}
+		} else {
+			// 直接删除键
+			return await this.conn.del(key);
+		}
+	} else {
+		throw '键不能为空';
+	}
+	return ret
+};
+
+/**
+ * @description 导出Mongodb_cache类
+ */
+exports.Mongodb_cache = Mongodb_cache;
+
+/**
+ * @description mongodb_cache连接池
+ */
+if (!$.pool.mongodb_cache) {
+	$.pool.mongodb_cache = {};
+}
+
+/**
+ * @description 缓存管理器，用于创建缓存
+ * @param {String} scope 作用域
+ * @param {String} dir 当前路径
+ * @return {Object} 返回一个缓存类
+ */
+function mongodb_cache_admin(scope, dir) {
+	if (!scope) {
+		scope = $.val.scope
+	}
+	var obj = $.pool.mongodb_cache[scope];
+	if (!obj) {
+		$.pool.mongodb_cache[scope] = new Mongodb_cache(scope, dir);
+		obj = $.pool.mongodb_cache[scope];
+	}
+	return obj;
+}
+
+/**
+ * @description 导出mongodb_cache管理函数
+ */
+exports.mongodb_cache_admin = mongodb_cache_admin;

package/index.js

@@ -16,7 +16,7 @@ class MongoDB {
 	 * @param {String} dir 当前路径
 	 * @constructor
 	 */
-	constructor(scope, dir) {
+	constructor(scope, dir, config) {
 		// 作用域
 		this.scope;
 		if (scope) {
@@ -29,9 +29,7 @@ class MongoDB {
 		if (dir) {
 			this.dir = dir;
 		}
-
-		// 配置参数
-		MongoDB.prototype.config = {
+		this.config = Object.assign({
 			// 服务器地址
 			host: "localhost",
 			// 端口号
@@ -47,13 +45,14 @@ class MongoDB {
 				// 使用新的url分析器
 				useNewUrlParser: true,
 				// 自动重连接
-				auto_reconnect: true,
+				// auto_reconnect: true,
 				// 使用统一拓扑
 				useUnifiedTopology: true,
+				// useCreateIndex: true,
 				// 连接池数
-				poolSize: 10
+				// poolSize: 10
 			}
-		};
+		}, config);
 		// 连接地址
 		this.url;
 		// 数据库连接器
@@ -61,7 +60,7 @@ class MongoDB {
 		// 操作的数据库
 		this.db;
 		// 表名
-		this.table = "";
+		this.table = "cache";
 		/**
 		 * 页码
 		 */
@@ -107,20 +106,20 @@ MongoDB.prototype.open = async function() {
 	if (!this.url) {
 		this.setUrl();
 	}
-	var _this = this;
-	// 返回一个 Promise
-	var p = new Promise((resolve, reject) => {
-		MongoClient.connect(_this.url, _this.config.options, function(err, conn) {
-			if (err) {
-				reject(err);
-			} else {
-				resolve(conn);
-			}
-		});
-	});
-	_this.conn = await p;
-	_this.db = _this.conn.db(_this.config.database);
-	return _this.conn == true;
+	if (!this.conn) {
+		this.conn = await MongoClient.connect(this.url, this.config.options);
+		this.db = this.conn.db(this.config.database);
+	}
+	return this.conn == true;
+};
+
+/**
+ * 关闭数据库连接
+ */
+MongoDB.prototype.close = function() {
+	if (this.conn) {
+		this.conn.close();
+	}
 };
 
 /**
@@ -129,21 +128,12 @@ MongoDB.prototype.open = async function() {
  * @return {Promise|Object} 执行结果
  */
 MongoDB.prototype.addTable = function(table) {
-	var _this = this;
 	if (table) {
 		this.table = table;
 	} else {
 		table = this.table;
 	}
-	return new Promise((resolve, reject) => {
-		_this.db.createCollection(table, function(err, ret) {
-			if (err) {
-				reject(err);
-			} else {
-				resolve(ret);
-			}
-		});
-	});
+	return this.db.createCollection(table);
 };
 
 /**
@@ -152,16 +142,7 @@ MongoDB.prototype.addTable = function(table) {
  * @return {Promise|Object} 执行结果
  */
 MongoDB.prototype.addObj = function(obj) {
-	var _this = this;
-	return new Promise((resolve, reject) => {
-		_this.db.collection(_this.table).insertOne(obj, function(err, ret) {
-			if (err) {
-				reject(err);
-			} else {
-				resolve(ret);
-			}
-		});
-	});
+	return this.db.collection(this.table).insertOne(obj);
 };
 
 /**
@@ -170,83 +151,40 @@ MongoDB.prototype.addObj = function(obj) {
  * @return {Promise|Object} 执行结果
  */
 MongoDB.prototype.addList = function(list) {
-	var _this = this;
-	return new Promise((resolve, reject) => {
-		_this.db.collection(_this.table).insertMany(list, function(err, ret) {
-			if (err) {
-				reject(err);
-			} else {
-				resolve(ret);
-			}
-		});
-	});
+	return this.db.collection(this.table).insertMany(list);
 };
 
 /**
  * @description 删除缓存
- * @param {Object} query 对象
+ * @param {Object} query 查询条件对象
  * @return {Promise|Object} 执行结果
  */
 MongoDB.prototype.del = function(query) {
-	var _this = this;
-	return new Promise((resolve, reject) => {
-		_this.db.collection(_this.table).remove(query, function(err, ret) {
-			if (err) {
-				reject(err);
-			} else {
-				resolve(ret);
-			}
-		});
-	});
+	return this.db.collection(this.table).remove(query);
 };
 
 /**
  * @description 修改缓存
- * @param {Object} query 对象
- * @param {Object} obj 对象
+ * @param {Object} query 查询条件对象
+ * @param {Object} body 修改对象
  * @return {Promise|Object} 执行结果
  */
-MongoDB.prototype.set = function(query, obj) {
-	var _this = this;
-	return new Promise((resolve, reject) => {
-		_this.db.collection(_this.table).update(query, obj, function(err, ret) {
-			if (err) {
-				reject(err);
-			} else {
-				resolve(ret);
-			}
-		});
-	});
+MongoDB.prototype.set = function(query, body) {
+	return this.db.collection(this.table).update(query, body);
 };
 
 /**
  * @description 查询缓存
- * @param {Object} query 对象
+ * @param {Object} query 查询条件对象
  * @return {Promise|Array} 执行结果
  */
 MongoDB.prototype.get = function(query) {
-	var _this = this;
-	return new Promise((resolve, reject) => {
-		if (_this.page !== 0 && _this.size !== 0) {
-			_this.db.collection(_this.table).find(query).limit(_this.size).skip((_this.page - 1) * _this.size).toArray(
-				function(err,
-					ret) {
-					if (err) {
-						reject(err);
-					} else {
-						resolve(ret);
-					}
-				});
-		} else {
-			_this.db.collection(_this.table).find(query).toArray(function(err, ret) {
-				if (err) {
-					reject(err);
-				} else {
-					resolve(ret);
-				}
-			});
-		}
-	});
+	if (this.page !== 0 && this.size !== 0) {
+		return this.db.collection(this.table).find(query).limit(this.size).skip((this.page - 1) * this
+			.size).toArray();
+	} else {
+		return this.db.collection(this.table).find(query).toArray();
+	}
 };
 
 /**
@@ -283,4 +221,4 @@ function mongoDB_admin(scope, dir) {
 /**
  * @module 导出MongoDB管理器
  */
-exports.mongoDB_admin = mongoDB_admin;
+exports.mongoDB_admin = mongoDB_admin;

package/package.json

@@ -1,36 +1,36 @@
-{
-    "name": "mm_mongodb",
-    "version": "1.3.8",
-    "description": "这是超级美眉mongdb帮助函数模块，用于便捷操作mongdb，使用await方式，可以避免嵌套函数",
-    "main": "index.js",
-    "scripts": {
-        "test": "node test.js"
-    },
-    "repository": {
-        "type": "git",
-        "url": "git+https://github.com/qiuwenwu/mm_mongodb.git"
-    },
-    "keywords": [
-        "mysql",
-        "async",
-        "await",
-        "promise",
-        "add",
-        "del",
-        "set",
-        "get",
-        "query",
-        "run",
-        "exec"
-    ],
-    "author": "邱文武",
-    "license": "ISC",
-    "bugs": {
-        "url": "https://github.com/qiuwenwu/mm_mongodb/issues"
-    },
-    "homepage": "https://github.com/qiuwenwu/mm_mongodb#readme",
-    "dependencies": {
-        "mm_expand": "^1.4.7",
-        "mongodb": "^3.3.2"
-    }
-}
+{
+    "name": "mm_mongodb",
+    "version": "1.4.1",
+    "description": "这是超级美眉mongdb帮助函数模块，用于便捷操作mongdb，使用await方式，可以避免嵌套函数",
+    "main": "index.js",
+    "scripts": {
+        "test": "node test.js"
+    },
+    "repository": {
+        "type": "git",
+        "url": "git+https://github.com/qiuwenwu/mm_mongodb.git"
+    },
+    "keywords": [
+        "mysql",
+        "async",
+        "await",
+        "promise",
+        "add",
+        "del",
+        "set",
+        "get",
+        "query",
+        "run",
+        "exec"
+    ],
+    "author": "邱文武",
+    "license": "ISC",
+    "bugs": {
+        "url": "https://github.com/qiuwenwu/mm_mongodb/issues"
+    },
+    "homepage": "https://github.com/qiuwenwu/mm_mongodb#readme",
+    "dependencies": {
+        "mm_expand": "^1.6.9",
+        "mongodb": "^5.9.1"
+    }
+}

package/test.js

@@ -1,36 +1,54 @@
 const MongoDB = require('./index.js').MongoDB;
+const mongodb_admin = require('./index.js').mongoDB_admin;
+
+async function test2() {
+	var mb = mongodb_admin("sys");
+	mb.setConfig();
+	// mb.open();
+	console.log(mb);
+}
+test2();
 
 /* 调用示例 */
 async function test() {
+	console.log("测试开始");
 	// 实例化构造函数，传入服务器地址和数据库名
-	var mb = new MongoDB("localhost", "mm"); 
+	var mb = new MongoDB("localhost", "mm");
 	mb.setConfig();
-	
+
 	await mb.open();
-	// 创建数据表
-	await mb.addTable('bs_table');
+	try {
+		// 创建数据表
+		await mb.addTable('bs_table');
+		console.log("创建表bs_table");
+	} catch (e) {
+		console.log("创建表错误提示：", e);
+		//TODO handle the exception
+	}
 	mb.table = "mm_table";
-	
+
 	var ret = await mb.get({
 		'name': '李四'
 	});
-	
+	console.log("查询李四",ret);
+
 	ret = await mb.get({
 		'name': '张三'
 	});
-	console.log(ret);
-	
-	 // 选择要操作的表
+	console.log("查询张三",ret);
+
+	// 选择要操作的表
 	mb.table = 'mm_table';
-	
-	 // 添加一条数据
+
+	// 添加一条数据
 	ret = await mb.addObj({
 		name: "张三",
 		age: 21,
 		sex: true
 	});
-	
-	 // 添加多条数据
+	console.log("添加一条数据", ret);
+
+	// 添加多条数据
 	ret = await mb.addList([{
 		name: "李四",
 		age: 18,
@@ -40,6 +58,9 @@ async function test() {
 		age: 15,
 		sex: true
 	}]);
+
+	console.log("添加多条数据", ret);
+	console.log("测试完成");
 }
 
-test();
+test();

package/test2.js

@@ -0,0 +1,57 @@
+const { Mongodb_cache, mongodb_cache_admin } = require('./cache.js');
+
+async function test2() {
+	var mb = mongodb_cache_admin("sys");
+	mb.setConfig();
+	// mb.open();
+	console.log(mb);
+}
+test2();
+
+/* 调用示例 */
+async function test() {
+	console.log("测试开始");
+	// 实例化构造函数，传入服务器地址和数据库名
+	var mb = new Mongodb_cache("localhost", "mm");
+	mb.setConfig();
+
+	await mb.open();
+	try {
+		// 创建数据表
+		await mb.addTable('bs_keyValue');
+		console.log("创建表bs_keyValue");
+	} catch (e) {
+		// console.log("创建表错误提示：", e);
+		//TODO handle the exception
+	}
+	mb.table = "mm_table";
+
+console.log("查询", mb.conn);
+	var ret = await mb.get('李四');
+	console.log("查询李四",ret);
+
+	ret = await mb.get('张三');
+	console.log("查询张三",ret);
+
+	// 选择要操作的表
+	mb.table = 'mm_table';
+
+	// 添加一条数据
+	ret = await mb.add("张三", {
+		age: 21,
+		sex: true
+	});
+	console.log("添加一条数据", ret);
+	
+	
+	// 修改一条数据
+	ret = await mb.add("李四", {
+		age: 24,
+		sex: false
+	});
+	console.log("修改一条数据", ret);
+	
+	console.log("测试完成");
+}
+
+test();