feat: 初始化版本！

src/plugins/mqttclient.ts

@@ -0,0 +1,165 @@
+import { generateUUID } from "@/utils/geek";
+import * as mqtt from "mqtt/dist/mqtt";
+const enableJSONDecoding = true // 开启JSON解析消息，需要开启JSON解析消息才能开启uuid和event
+const enableJSONEncoding = true // 开启JSON消息编码
+const enableUUID = true // 需要接收信息中包含uuid字段，uuid优先级高于event
+
+type MqttConnectOptions = {
+    clean: boolean,
+    connectTimeout: number,
+    clientId: string, // 认证信息
+    keepalive: number,
+    protocolId?: string, // 默认 'MQTT'
+    protocolVersion?: number, // 默认4
+    username?: string,
+    password?: string,
+}
+type MqttConnectConfig = {
+    url?: string,
+    options: MqttConnectOptions
+}
+interface MqttClient {
+    connected:boolean,
+    on: {
+        (event: "connect", fun: () => void): void;
+        (event: "reconnect", fun: () => void): void;
+        (event: "disconnect", fun: () => void): void;
+        (event: "offline", fun: () => void): void;
+        (event: "message", fun: (topic: string, message: any, packet?: any) => void): void;
+        (event: "packetsend", fun: (packet: any) => void): void;
+        (event: "packetreceive", fun: (packet: any) => void): void;
+        (event: "error", fun: (err: any) => void): void;
+    };
+    publish: {
+        (topic: string, message: any, options?: any, callback?: (err: any) => void): void;
+    }
+    subscribe: {
+        (topic: string, options?: any, callback?: (err: any, granted: { topic: string, qos: any }) => void): void
+    }
+    unsubscribe: {
+        (topic: string, callback?: (err: any) => void): void
+    }
+    end(force?:boolean, options?:any, callback?: (err: any) => void): void
+}
+
+let _client: MqttClient;
+let _callback: { [key: string]: (data: any) => void } = {}
+let onmassage = (topic:string,message:string)=>{}
+export default {
+    /**
+     * 连接websocket
+     * 最简单的用法就是传入{url:"ws://demo"}
+     * 当连接成功后触发回调函数
+     */
+    connect(config: MqttConnectConfig) {
+        return new Promise<void>((resolve, reject) => {
+            if(!_client || !_client.connected){
+                _client = mqtt.connect(config.url, config.options);
+                _client.on('connect', resolve);
+                _client.on('error', reject);
+                _client.on('message', (topic,message) => {
+                    if (enableJSONDecoding) {
+                        let data = JSON.parse(message)
+                        if (enableUUID && (data || {}).uuid !== undefined) {
+                            _callback[data.uuid](data)
+                            delete _callback[data.uuid]
+                        }
+                    }
+                    onmassage(topic,message)
+                })
+            }
+        })
+    },
+    /**
+     * 发送信息
+     * @param msg 消息，会被处理成json字符串
+     * @param uuid 唯一标识,可以传入uuid，也可以传入true自动生成uuid，flase表示该消息不需要单独处理
+     * @returns
+     */
+    send(topic: string, msg: any, uuid: string | boolean = false, options: any = undefined) {
+        return new Promise((resolve, reject) => {
+            const useUUID = enableUUID && uuid != undefined && uuid != "" && uuid != false
+            if (useUUID) {
+                if (uuid === true) {
+                    msg.uuid = generateUUID()
+                    _callback[msg.uuid] = resolve
+                } else {
+                    _callback[uuid] = resolve
+                }
+            }
+            if(enableJSONEncoding){
+                msg = JSON.stringify(msg)
+            }
+            _client.publish(topic, msg, options, err => {
+                if (err) {
+                    reject(err)
+                } else if (uuid === false) {
+                    resolve(err)
+                }
+            })
+        })
+    },
+    /**
+     * 关闭连接
+     * @returns 关闭连接的Promise，回调函数只会运行一次
+     */
+    close(force?: boolean, options?: any, callback?: (err: any) => void) {
+        return new Promise<void>((resolve, reject) => {
+            _client.end(force,options,()=>{
+                resolve()
+            })
+        })
+    },
+    /**
+     * 订阅事件
+     * @param event 要监听的事件
+     * @returns 在回调函数中处理事件
+     */
+    subscribe(topic: string, options: string | undefined = undefined) {
+        return new Promise<void>((resolve, reject) => {
+            _client.subscribe(topic, options, err => {
+                if (err) {
+                    reject(err)
+                } else {
+                    resolve()
+                }
+            })
+        })
+    },
+    /**
+     * 取消订阅事件
+     * @param event 要取消监听的事件
+     */
+    unsubscribe(topic: string) {
+        return new Promise<void>((resolve, reject) => {
+            _client.unsubscribe(topic, err => {
+                if (err) {
+                    reject(err)
+                } else {
+                    resolve()
+                }
+            })
+        })
+    },
+    /**
+     * 定义默认监听事件
+     * @param callback 默认监听事件的处理函数
+     */
+    onMessage(callback: (topic:string,message:string) => void) {
+        onmassage = callback
+    },
+    /**
+     * 定义异常事件
+     * @param callback 默认异常事件的处理函数
+     */
+    onError(callback: (data: any) => void) {
+        _client.on('error', callback)
+    },
+    /**
+     * 定义关闭事件
+     * @param callback 默认关闭事件的处理函数
+     */
+    onClose(callback: () => void) {
+        _client.on('disconnect', callback)
+    }
+}