mm_os 2.1.9 → 2.2.1

package/core/com/mqtt/index.js

@@ -1,3 +1,8 @@
+const MQTT_client = require('mqtt');
+const {
+	v4
+} = require('uuid');
+
 const Index = require('mm_machine').Index;
 const Drive = require('./drive');
 
@@ -17,19 +22,64 @@ class MQTT extends Index {
 		super(scope, __dirname);
 		this.Drive = Drive;
 		this.type = "mqtt";
-		this.dict = {};
 		this.title = title;
+		this.dict = {};
 
 		this.config = {
-			host: "127.0.0.1",
+			hostname: "127.0.0.1",
 			port: "1883",
 			protocol: "mqtt",
-			clientId: "test123",
-			username: "admin",
+			clientId: "server",
+			subscribe_qos: 0,
+			publish_qos: 1,
+			username: "server",
 			password: "asd123",
-			clean: false
+			clean: false,
+			longtime: 15000,
+			// maxInflight: 20,
+			// 重连间隔
+			interval: 6000
 		}
+
+		// mqtt客户端服务器
 		this.client = null;
+		/**
+		 * message queue
+		 */
+		this.list_msg = [
+			/*
+			{
+				// information ID
+				id: "",
+				// request method
+				method: "",
+				// message parameters
+				params: {},
+				// Callback
+				func: function(res){}
+			}
+			*/
+		];
+
+		/**
+		 * 订阅集合 (主题 => 函数列表)
+		 */
+		this.dict_subscribe = {};
+
+		/**
+		 * method collection
+		 */
+		this.methods = {
+			/**
+			 * Provide to the server to see how many open functions there are
+			 */
+			get_method: function() {
+				return Object.keys(_this.methods)
+			}
+		};
+
+		this.retryTimes = 0;
+		this.connecting = false;
 	}
 }
 
@@ -56,57 +106,290 @@ MQTT.prototype.match = function(topic, top) {
 /**
  * 初始化
  */
-MQTT.prototype.init_after = function() {
-	this.client.on("message", (topic, msg) => {
-		var db = $.sql.db();
-		var list = this.list;
-		for (var i = 0, o; o = list[i++];) {
-			if (this.match(topic, o.config.topic)) {
-				o.run(topic, msg);
+MQTT.prototype.update_after = function() {
+	var _this = this;
+	var list = this.list;
+	for (var i = 0, o; o = list[i++];) {
+		o.send = function(topic, msg, qos = 0, retain = false) {
+			if (topic) {
+				_this.send(topic, msg, qos, retain);
+			}
+		};
+
+		o.run('init');
+		var topics = o.config.topic;
+		if (topics) {
+			for (var i = 0; i < topics.length; i++) {
+				var tc = topics[i];
+				var retain = o.config.retain || false;
+				var qos = o.config.qos || 0;
+				if (tc.indexOf("heartbeat") !== -1) {
+					retain = false;
+					qos = 0;
+				}
+				this.subscribe(tc, null, qos, retain);
 			}
 		}
+	}
+}
+
+
+/**
+ * 重新连接
+ */
+MQTT.prototype.reconnect = function() {
+	this.retryTimes += 1;
+	if (this.retryTimes > 5) {
+		try {
+			this.client.end();
+			$.sleep(this.config.interval);
+			this.init();
+		} catch (error) {
+			this.$message.error(error.toString());
+		}
+	}
+}
+
+/**
+ * 初始化
+ * @param {Object} config 配置参数
+ */
+MQTT.prototype.init = function(config) {
+	this.config = Object.assign(this.config, config);
+	this.client = {
+		connected: false
+	};
+	this.retryTimes = 0;
+	this.connecting = false;
+};
+
+/**
+ * 运行MQTT
+ * @param {Object} config 配置参数
+ */
+MQTT.prototype.start = function() {
+	this.connecting = true;
+	var err;
+	try {
+		var cg = this.config;
+		if (cg.host) {
+			delete cg.hostname;
+			delete cg.port;
+			this.client = MQTT_client.connect(cg.host, cg);
+		} else {
+			this.client = MQTT_client.connect(cg);
+		}
+		if (this.client.on) {
+			return new Promise((resolve, reject) => {
+				this.client.on('connect', (packet, err) => {
+					this.connecting = false;
+					if (err) {
+						resolve(null);
+						reject(err);
+					} else {
+						this.client.on('message', async (topic, message) => {
+							await this.receive(topic, message.toString());
+						});
+						resolve(packet);
+					}
+				});
+				this.client.on("reconnect", this.reconnect);
+				this.client.on("error", (error) => {
+					console.error("Connection failed", error);
+				});
+			});
+		}
+	} catch (error) {
+		this.connecting = false;
+		err = error;
+	}
+	return new Promise((resolve, reject) => {
+		resolve(null);
+		reject(err);
 	});
+};
+
+/**
+ * 监听事件
+ * @param {String} topic 事件名称
+ * @param {Function} func 回调函数
+ */
+MQTT.prototype.on = function(topic, func) {
+	this.client.on(topic, func);
+}
+
+/**
+ * 接收消息
+ * @param {String} topic 订阅板块
+ * @param {Object} msg 消息主体
+ */
+MQTT.prototype.message = async function(topic, msg) {
+	if (this.dict_subscribe[topic]) {
+		var list = this.dict_subscribe[topic];
+		try {
+			for (var key in list) {
+				list[key](msg);
+			}
+		} catch (err) {
+			console.error(err);
+		}
+	}
 }
 
 /**
- * 处理mqtt请求
- * @param {Object} conn 通讯连接器
+ * 接收消息
+ * @param {String} push_topic 推送过来的主题
+ * @param {Object} msg 消息主体
  */
-MQTT.prototype.run = async function(conn) {
+MQTT.prototype.receive = async function(push_topic, msg) {
+	if (msg && (msg.indexOf('[') === 0) || msg.indexOf('{') === 0) {
+		try {
+			msg = JSON.parse(msg);
+		} catch {}
+	}
+	var ret;
 	var list = this.list;
 	for (var i = 0, o; o = list[i++];) {
-		client.subscribe(o.config.topic);
+		var topics = o.config.topic;
+		if (topics) {
+			for (var i = 0; i < topics.length; i++) {
+				var topic = topics[i];
+				if (this.match(push_topic, topic)) {
+					try {
+						ret = await o.run("main", push_topic, msg, topic, i);
+					} catch (error) {
+						$.log.error("mqtt处理程序错误", error);
+						return
+					}
+				}
+			}
+		}
+	}
+	try {
+		await this.message(topic, msg);
+	} catch (error) {
+		$.log.error("mqtt处理程序错误", error);
+		return
+	}
+};
+
+/**
+ * 订阅
+ * @param {String} topic 主题
+ * @param {Function} func 回调函数
+ * @param {Number} qos 推送方式 0
+ * @param {Boolean} retain 保持
+ */
+MQTT.prototype.subscribe = function(topic, func, qos = null, retain = false) {
+	if (qos === null || qos === undefined) {
+		qos = this.config.subscribe_qos || 0;
+	}
+	this.client.subscribe(topic, {
+		// 订阅消息方式，0为保留 ， 1为确认收到1次
+		qos,
+		retain
+	});
+	if (func) {
+		if (!this.dict_subscribe[topic]) {
+			this.dict_subscribe[topic] = {};
+		}
+		var key = this.key_num + 1;
+		this.dict_subscribe[topic][key] = func;
+		return key;
+	}
+}
+
+/**
+ * 取消订阅
+ * @param {String} topic 主题
+ * @param {String} key 主键
+ */
+MQTT.prototype.unsubscribe = function(topic, key) {
+	this.client.unsubscribe(topic);
+	if (this.dict_subscribe[topic]) {
+		delete this.dict_subscribe[topic][key];
+	}
+}
+
+/**
+ * 结束客户端
+ */
+MQTT.prototype.end = function() {
+	this.client.end();
+};
+
+/**
+ * 发布主题
+ * @param {String} topic 主题
+ * @param {Object} message 消息对象
+ * @param {Number} qos 推送方式 0
+ * @param {Boolean} retain 保持
+ */
+MQTT.prototype.publish = function(topic, message, qos = null, retain = false) {
+	if (qos === null || qos === undefined) {
+		qos = this.config.publish_qos || 0;
 	}
+	// console.log("发布", qos, topic);
+	this.client.publish(topic, message, {
+		qos,
+		retain
+	});
+};
+
+/**
+ * 主题推送
+ * @param {String} topic 主题
+ * @param {Object} msgObj 消息对象
+ * @param {Number} qos 推送方式 0
+ * @param {Boolean} retain 保持
+ */
+MQTT.prototype.send = function(topic, msgObj, qos = null, retain = false) {
+	this.publish(topic, JSON.stringify(msgObj), qos, retain);
 };
 
 /**
- * 加载插件
- * @param {String} path 检索路径
- * @param {Boolean} isApp 是否APP
+ * 请求
+ * @param {String} topic 主题
+ * @param {String} method 请求方法
+ * @param {Object} params 请求参数
+ * @param {Function} func 回调函数
  */
-MQTT.prototype.load = function(path) {
-	if (!path) {
-		path = "./mqtt/";
+MQTT.prototype.req = function(topic, method, params, func) {
+	var data = {
+		id: this.client.options.clientId + "_" + Date.parse(new Date()),
+		method,
+		params
+	};
+
+	if (func) {
+		data.func = func;
+		this.list_msg.push(data);
 	}
-	// 获取所有应用路径
-	var list_scope = $.dir.getAll(path, "mqtt");
+	this.send(topic, data);
+};
 
-	// 遍历目录路径
+/**
+ * 同步请求 - 可及时取回消息
+ * @param {String} method 请求方法
+ * @param {Object} params 传递参数
+ * @returns {Object} 返回响应结果
+ */
+MQTT.prototype.reqASync = function(topic, method, params) {
 	var _this = this;
-	list_scope.map(async function(f) {
-		var list_file = $.file.getAll(f, "*" + _this.type + ".json");
-		list_file.map(async (file) => {
-			var dir = file.dirname();
-			if (file.hasFile()) {
-				var obj = new Drive(dir);
-				obj.load(file);
-				if (obj.config.name) {
-					_this.list.push(obj);
-				}
-			}
+	return new Promise((resolve, reject) => {
+		var hasMsg;
+		_this.req(topic, method, params, (res) => {
+			hasMsg = true;
+			resolve(res);
 		});
+		setTimeout(function() {
+			if (!hasMsg) {
+				resolve(null);
+				reject("request timeout!");
+			}
+		}, 3000);
 	});
-}
+};
 
 module.exports = MQTT;
 
@@ -136,5 +419,4 @@ function mqtt_admin(scope, title) {
  * @param {string} title 标题
  * @return {Object} 返回一个缓存类
  */
-$.mqtt_admin = mqtt_admin;
-
+$.mqtt_admin = mqtt_admin;

package/core/com/mqtt/mm_mqtt.js

@@ -0,0 +1,327 @@
+const MQTT = require('mqtt');
+const {
+	v4
+} = require('uuid');
+
+
+class MM_mqtt {
+	/**
+	 * 构造函数
+	 * @param {Object} config 配置参数
+	 */
+	constructor(config) {
+		var clientId = v4();
+		this.config = Object.assign({
+			hostname: "127.0.0.1",
+			port: "",
+			protocol: "ws",
+			clientId,
+			subscribe_qos: 0,
+			publish_qos: 1,
+			username: "demo",
+			password: "asd123",
+			clean: false,
+			topic_receive: "client/message/",
+			longtime: 15000,
+			// maxInflight: 20,
+			// 重连间隔
+			interval: 6000
+		}, config);
+
+		// mqtt客户端服务器
+		this.client = null;
+
+		/**
+		 * message queue
+		 */
+		this.list_msg = [
+			/*
+			{
+				// information ID
+				id: "",
+				// request method
+				method: "",
+				// message parameters
+				params: {},
+				// Callback
+				func: function(res){}
+			}
+			*/
+		];
+
+		/**
+		 * 订阅集合 (主题 => 函数列表)
+		 */
+		this.dict_subscribe = {};
+
+		/**
+		 * method collection
+		 */
+		this.methods = {
+			/**
+			 * Provide to the server to see how many open functions there are
+			 */
+			get_method: function() {
+				return Object.keys(_this.methods)
+			}
+		};
+
+		this.retryTimes = 0;
+		this.connecting = false;
+	}
+}
+
+/**
+ * 重新连接
+ */
+MM_mqtt.prototype.reconnect = function() {
+	this.retryTimes += 1;
+	if (this.retryTimes > 5) {
+		try {
+			this.client.end();
+			$.sleep(this.config.interval);
+			this.init();
+		} catch (error) {
+			this.$message.error(error.toString());
+		}
+	}
+}
+
+/**
+ * 初始化
+ * @param {Object} config 配置参数
+ */
+MM_mqtt.prototype.init = function(config) {
+	this.config = Object.assign(this.config, config);
+	this.client = {
+		connected: false,
+	};
+	this.retryTimes = 0;
+	this.connecting = false;
+};
+
+/**
+ * 运行MQTT
+ */
+MM_mqtt.prototype.run = function(config) {
+	this.connecting = true;
+	var err;
+	try {
+		var cg = Object.assign(this.config, config);
+		if (cg.host) {
+			delete cg.hostname;
+			delete cg.port;
+			this.client = MQTT.connect(cg.host, cg);
+		} else {
+			this.client = MQTT.connect(cg);
+		}
+		if (this.client.on) {
+			return new Promise((resolve, reject) => {
+				this.client.on('connect', (packet, err) => {
+					this.connecting = false;
+					if (err) {
+						resolve(null);
+						reject(err);
+					} else {
+						this.client.on('message', async(topic, message) => {
+							await this.receive(topic, message.toString());
+						});
+						resolve(packet);
+					}
+				});
+				this.client.on("reconnect", this.reconnect);
+				this.client.on("error", (error) => {
+					console.error("Connection failed", error);
+				});
+			});
+		}
+	} catch (error) {
+		this.connecting = false;
+		err = error;
+	}
+	return new Promise((resolve, reject) => {
+		resolve(null);
+		reject(err);
+	});
+};
+
+/**
+ * 监听事件
+ * @param {String} eventName
+ * @param {Function} func
+ */
+MM_mqtt.prototype.on = function(eventName, func) {
+	this.client.on(eventName, func);
+};
+
+/**
+ * 接收消息
+ * @param {String} topic 订阅板块
+ * @param {Object} msg 消息主体
+ */
+MM_mqtt.prototype.message = async function(topic, msg) {
+	if (this.dict_subscribe[topic]) {
+		var list = this.dict_subscribe[topic];
+		try {
+			for (var key in list) {
+				list[key](msg);
+			}
+		} catch (err) {
+			console.error(err);
+		}
+	}
+}
+
+
+/**
+ * 接收消息
+ * @param {String} topic 订阅
+ * @param {Object} bodyStr 消息主体
+ */
+MM_mqtt.prototype.receive = async function(topic, bodyStr) {
+	var json;
+	if (bodyStr && (bodyStr.indexOf('[') === 0) || bodyStr.indexOf('{') === 0) {
+		try {
+			json = JSON.parse(bodyStr);
+		} catch {}
+	}
+	if (json) {
+		var id = json.id;
+		if (json.result && id) {
+			var lt = this.list_msg;
+			var len = lt.length;
+			var has = false;
+			for (var i = 0; i < len; i++) {
+				var o = lt[i];
+				if (id === o.id) {
+					o.func(json.result);
+					lt.splice(i, 1);
+					has = true;
+					break;
+				}
+			}
+			if (has) {
+				return;
+			}
+		} else if (json.method) {
+			var func;
+			var methods = this.methods;
+			var arr = json.method.split('.');
+			for (var i = 0; i < arr.length; i++) {
+				var m = methods[arr[i]];
+				if (m) {
+					methods = m;
+				}
+			}
+			if (methods && typeof(methods) == 'function') {
+				func = methods;
+			}
+			if (func) {
+				var params = json.params;
+				var from = params.from;
+				var ret = func(params);
+				if (ret) {
+					var obj = {};
+					if (id) {
+						obj.id = id
+					}
+					obj.result = ret;
+					if (from) {
+						var topic_receive = this.config.topic_receive + from;
+						this.send(topic_receive, obj);
+					}
+				}
+				return;
+			}
+		}
+		await this.message(topic, json);
+		return
+	}
+	await this.message(topic, bodyStr);
+};
+
+
+MM_mqtt.prototype.subscribe = function(topic, func, qos = null, retain = false) {
+	if (qos === null || qos === undefined) {
+		qos = this.config.subscribe_qos || 0;
+	}
+	// console.log("订阅", qos);
+	this.client.subscribe(topic, {
+		// 订阅消息方式，0为保留 ， 1为确认收到1次
+		qos,
+		retain
+	});
+	if (func) {
+		if (!this.dict_subscribe[topic]) {
+			this.dict_subscribe[topic] = {};
+		}
+		var key = this.key_num + 1;
+		this.dict_subscribe[topic][key] = func;
+		return key;
+	}
+}
+
+MM_mqtt.prototype.unsubscribe = function(topic, key) {
+	this.client.unsubscribe(topic);
+	if (this.dict_subscribe[topic]) {
+		delete this.dict_subscribe[topic][key];
+	}
+}
+
+MM_mqtt.prototype.end = function() {
+	this.client.end();
+};
+
+MM_mqtt.prototype.publish = function(topic, message, qos = null, retain = false) {
+	if (qos === null || qos === undefined) {
+		qos = this.config.publish_qos || 0;
+	}
+	// console.log("发布", qos, topic);
+	this.client.publish(topic, message, {
+		qos,
+		retain
+	});
+};
+
+MM_mqtt.prototype.send = function(topic, msgObj) {
+	this.publish(topic, JSON.stringify(msgObj));
+};
+
+MM_mqtt.prototype.req = function(topic, method, params, func) {
+	var data = {
+		id: this.client.options.clientId + "_" + Date.parse(new Date()),
+		method,
+		params
+	};
+
+	if (func) {
+		data.func = func;
+		this.list_msg.push(data);
+	}
+	this.send(topic, data);
+};
+
+/**
+ * 同步请求 - 可及时取回消息
+ * @param {String} method 请求方法
+ * @param {Object} params 传递参数
+ * @returns {Object} 返回响应结果
+ */
+MM_mqtt.prototype.reqASync = function(topic, method, params) {
+	var _this = this;
+	return new Promise((resolve, reject) => {
+		var hasMsg;
+		_this.req(topic, method, params, (res) => {
+			hasMsg = true;
+			resolve(res);
+		});
+		setTimeout(function() {
+			if (!hasMsg) {
+				resolve(null);
+				reject("request timeout!");
+			}
+		}, 3000);
+	});
+};
+
+module.exports = MM_mqtt;