Function main

client-sdks/client-js/examples/event-subscription.ts:8–56

()

import { WebSocketClient, SubscriptionManager, isProgressEvent, isControlEvent, isMonitorEvent, isEventType } from "@aster/client-js";

async function main() {
  // 1. 创建 WebSocket 客户端
  const ws = new WebSocketClient({
    maxReconnectAttempts: 5,
    reconnectDelay: 1000,
    heartbeatInterval: 30000,
    heartbeatTimeout: 10000,
  });

  // 2. 连接到 aster 服务器
  try {
    await ws.connect("ws://localhost:8080/ws");
    console.log("✅ Connected to aster");
  } catch (error) {
    console.error("❌ Connection failed:", error);
    return;
  }

  // 3. 创建订阅管理器
  const subscriptionManager = new SubscriptionManager(ws);

  // 4. 订阅所有三个通道
  const subscription = subscriptionManager.subscribe(["progress", "control", "monitor"], {
    agentId: "agent-123",
    eventTypes: ["thinking", "text_chunk", "tool_start", "token_usage"],
  });

  // 5. 处理事件
  try {
    for await (const envelope of subscription) {
      const event = envelope.event;

      // 按通道分类处理
      if (isProgressEvent(event)) {
        handleProgressEvent(event);
      } else if (isControlEvent(event)) {
        handleControlEvent(event);
      } else if (isMonitorEvent(event)) {
        handleMonitorEvent(event);
      }
    }
  } catch (error) {
    console.error("❌ Event subscription error:", error);
  }

  // 6. 清理
  subscription.unsubscribe();
  ws.disconnect();
}

/**
 * 处理 Progress Channel 事件