rol-websocket-channel 1.0.2 → 1.0.6

package/src/mqtt/mqtt-client.ts

@@ -1,119 +1,170 @@
-// MQTT 客户端类
-// 封装 MQTT 连接和消息处理逻辑（使用 MQTT.js 自带的重连功能）
-
-import type { MqttConfig } from "./types.js";
-import * as ConnectionManager from "./connection-manager.js";
-
-export interface MqttClientOptions {
-  config: MqttConfig;
-  abortSignal: AbortSignal;
-  onMessage: (topic: string, payload: Buffer) => Promise<void>;
-  onConnect?: () => void;
-  onDisconnect?: () => void;
-  onError?: (err: Error) => void;
-}
-
-export class MqttClient {
-  private config: MqttConfig;
-  private abortSignal: AbortSignal;
-  private onMessage: (topic: string, payload: Buffer) => Promise<void>;
-  private onConnect?: () => void;
-  private onDisconnect?: () => void;
-  private onError?: (err: Error) => void;
-
-  constructor(options: MqttClientOptions) {
-    this.config = options.config;
-    this.abortSignal = options.abortSignal;
-    this.onMessage = options.onMessage;
-    this.onConnect = options.onConnect;
-    this.onDisconnect = options.onDisconnect;
-    this.onError = options.onError;
-  }
-
-  /**
-   * 连接到 MQTT Broker
-   * 使用 MQTT.js 自带的重连功能
-   */
-  async connect(): Promise<void> {
-    const { accountId, mqttUrl, mqttTopic } = this.config;
-
-    console.log(`[MQTT] MqttClient.connect(${accountId}): starting connection to ${mqttUrl}...`);
-
-    // 检查是否已中止
-    if (this.abortSignal.aborted) {
-      console.log(`[MQTT] MqttClient.connect(${accountId}): aborted before starting`);
-      return;
-    }
-
-    // 创建连接
-    await ConnectionManager.createMqttConnection(
-      this.config,
-      {
-        onConnect: () => {
-          console.log(`[MQTT] MqttClient(${accountId}): onConnect callback`);
-          this.onConnect?.();
-        },
-        onDisconnect: () => {
-          console.log(`[MQTT] MqttClient(${accountId}): onDisconnect callback`);
-          this.onDisconnect?.();
-        },
-        onError: (err) => {
-          console.error(`[MQTT] MqttClient(${accountId}): onError callback - ${err.message}`);
-          this.onError?.(err);
-        },
-        onClose: () => {
-          console.log(`[MQTT] MqttClient(${accountId}): onClose callback`);
-          // MQTT.js 会自动重连，不需要我们干预
-          // 只有在 abortSignal 触发时才真正关闭
-          if (this.abortSignal.aborted) {
-            console.log(`[MQTT] MqttClient(${accountId}): abort signaled, cleaning up`);
-            ConnectionManager.removeConnection(accountId);
-          }
-        },
-        onMessage: this.onMessage,
-      }
-    );
-
-    console.log(`[MQTT] MqttClient.connect(${accountId}): connection created, waiting for abort signal...`);
-
-    // 等待中止信号
-    await new Promise<void>((resolve) => {
-      const handleAbort = () => {
-        console.log(`[MQTT] MqttClient(${accountId}): abort signal received`);
-        this.disconnect();
-        resolve();
-      };
-
-      this.abortSignal.addEventListener("abort", handleAbort, { once: true });
-    });
-
-    console.log(`[MQTT] MqttClient.connect(${accountId}): connection loop ended`);
-  }
-
-  /**
-   * 断开连接（手动停止时使用）
-   */
-  disconnect(): void {
-    const { accountId } = this.config;
-    console.log(`[MQTT] MqttClient.disconnect(${accountId}): disconnecting...`);
-    ConnectionManager.closeConnection(accountId);
-    console.log(`[MQTT] MqttClient.disconnect(${accountId}): disconnected`);
-  }
-
-  /**
-   * 发布消息
-   */
-  publish(topic: string, message: string): boolean {
-    const { accountId } = this.config;
-    console.log(`[MQTT] MqttClient.publish(${accountId}): publishing to ${topic}`);
-    return ConnectionManager.publishMessage(accountId, topic, message);
-  }
-
-  /**
-   * 检查是否已连接
-   */
-  isConnected(): boolean {
-    const { accountId } = this.config;
-    return ConnectionManager.isConnected(accountId);
-  }
-}
+// MQTT 全局客户端
+// 封装全局 MQTT 连接逻辑（使用 MQTT.js 自带的重连功能）
+// 全进程共享单一连接，随机生成 clientId，从 topic 解析用户名
+
+import * as ConnectionManager from "./connection-manager.js";
+
+// ─────────────────────────────────────────────
+// 选项接口
+// ─────────────────────────────────────────────
+
+export interface GlobalMqttClientOptions {
+  /** Broker 地址，如 ws://192.168.1.152:8083/mqtt */
+  mqttUrl: string;
+  /**
+   * 原始 topic，格式为 announcement/{user_name}/{agent_id}/...
+   * 用于解析用户名（作为 MQTT username）和作为消息发布目标
+   * 例：announcement/0b9a784d-e044-4c6f-9024-ef8799c131d1/ce627d3d-9e22-47f7-b1c6-a6ab69570fef/bot
+   */
+  mqttTopic: string;
+  /** AbortSignal，触发时断开连接并结束 connect() */
+  abortSignal: AbortSignal;
+  /** 收到消息时回调 */
+  onMessage: (topic: string, payload: Buffer) => Promise<void>;
+  /** 连接成功回调 */
+  onConnect?: () => void;
+  /** 连接断开回调 */
+  onDisconnect?: () => void;
+  /** 连接错误回调 */
+  onError?: (err: Error) => void;
+}
+
+// ─────────────────────────────────────────────
+// GlobalMqttClient
+// ─────────────────────────────────────────────
+
+/**
+ * 全局共享 MQTT 客户端
+ *
+ * 全进程只维护一个 MQTT 连接：
+ *  - clientId 每次启动随机生成
+ *  - MQTT username 从 mqttTopic 中解析（announcement/{user_name}/...），默认 "default_name"
+ *  - 订阅通配符 topic（announcement/{user_name}/#），不区分 agent_id
+ *  - 多处调用 connect() 时复用同一连接
+ */
+export class GlobalMqttClient {
+  private mqttUrl: string;
+  private mqttTopic: string;
+  private abortSignal: AbortSignal;
+  private onMessageCb: (topic: string, payload: Buffer) => Promise<void>;
+  private onConnectCb?: () => void;
+  private onDisconnectCb?: () => void;
+  private onErrorCb?: (err: Error) => void;
+
+  constructor(options: GlobalMqttClientOptions) {
+    this.mqttUrl = options.mqttUrl;
+    this.mqttTopic = options.mqttTopic;
+    this.abortSignal = options.abortSignal;
+    this.onMessageCb = options.onMessage;
+    this.onConnectCb = options.onConnect;
+    this.onDisconnectCb = options.onDisconnect;
+    this.onErrorCb = options.onError;
+  }
+
+  /**
+   * 建立全局 MQTT 连接并等待 AbortSignal
+   *
+   * 若全局连接已存在且已连接则直接复用。
+   * 方法持续挂起直到 abortSignal 触发，触发后自动断开并 resolve。
+   */
+  async connect(): Promise<void> {
+    const username = ConnectionManager.parseUsernameFromTopic(this.mqttTopic);
+    const subscribeTopic = ConnectionManager.getSubscribeTopic(this.mqttTopic);
+
+    console.log(
+      `[MQTT] GlobalMqttClient.connect(): url=${this.mqttUrl}, username=${username}, subscribeTopic=${subscribeTopic}`,
+    );
+
+    if (this.abortSignal.aborted) {
+      console.log("[MQTT] GlobalMqttClient.connect(): aborted before starting");
+      return;
+    }
+
+    await ConnectionManager.createGlobalMqttConnection(
+      this.mqttUrl,
+      this.mqttTopic,
+      {
+        onConnect: () => {
+          console.log("[MQTT] GlobalMqttClient: onConnect");
+          this.onConnectCb?.();
+        },
+        onDisconnect: () => {
+          console.log("[MQTT] GlobalMqttClient: onDisconnect");
+          this.onDisconnectCb?.();
+        },
+        onError: (err) => {
+          console.error(`[MQTT] GlobalMqttClient: onError - ${err.message}`);
+          this.onErrorCb?.(err);
+        },
+        onClose: () => {
+          console.log("[MQTT] GlobalMqttClient: onClose");
+          // MQTT.js 会自动重连；仅在 abort 触发时才真正清理全局连接
+          if (this.abortSignal.aborted) {
+            ConnectionManager.closeGlobalConnection();
+          }
+        },
+        onMessage: this.onMessageCb,
+      },
+    );
+
+    console.log(
+      "[MQTT] GlobalMqttClient.connect(): connection created, waiting for abort signal...",
+    );
+
+    // 挂起，直到 AbortSignal 触发
+    await new Promise<void>((resolve) => {
+      const handleAbort = () => {
+        console.log(
+          "[MQTT] GlobalMqttClient: abort signal received, disconnecting",
+        );
+        this.disconnect();
+        resolve();
+      };
+      this.abortSignal.addEventListener("abort", handleAbort, { once: true });
+    });
+
+    console.log("[MQTT] GlobalMqttClient.connect(): connection loop ended");
+  }
+
+  /**
+   * 手动断开全局连接
+   */
+  disconnect(): void {
+    console.log(
+      "[MQTT] GlobalMqttClient.disconnect(): closing global connection",
+    );
+    ConnectionManager.closeGlobalConnection();
+  }
+
+  /**
+   * 向指定 topic 发布消息
+   * @returns 发布成功返回 true，未连接返回 false
+   */
+  publish(topic: string, message: string): boolean {
+    console.log(`[MQTT] GlobalMqttClient.publish(): topic=${topic}`);
+    return ConnectionManager.publishGlobalMessage(topic, message);
+  }
+
+  /**
+   * 检查全局连接是否处于已连接状态
+   */
+  isConnected(): boolean {
+    return ConnectionManager.isGlobalConnected();
+  }
+
+  /**
+   * 获取当前解析的订阅 topic（通配符形式）
+   * 例：announcement/0b9a784d-e044-4c6f-9024-ef8799c131d1/#
+   */
+  getSubscribeTopic(): string {
+    return ConnectionManager.getSubscribeTopic(this.mqttTopic);
+  }
+
+  /**
+   * 获取从 topic 中解析的用户名
+   * 例：announcement/myuser/agent/bot → "myuser"
+   */
+  getUsername(): string {
+    return ConnectionManager.parseUsernameFromTopic(this.mqttTopic);
+  }
+}