triflux 3.2.0-dev.1 → 3.2.0-dev.10

package/hub/router.mjs

@@ -1,73 +1,306 @@
-// hub/router.mjs — Actor mailbox 라우터 + QoS 스케줄러
-// 메시지 라우팅, ask/publish/handoff 처리, TTL 정리
+// hub/router.mjs — 실시간 라우팅/수신함 상태 관리자
+// SQLite는 감사 로그만 담당하고, 실제 배달 상태는 메모리에서 관리한다.
 import { EventEmitter, once } from 'node:events';
 import { uuidv7 } from './store.mjs';
-
-/**
- * 라우터 생성
- * @param {object} store — createStore() 반환 객체
- */
+
+function uniqueStrings(values = []) {
+  return Array.from(new Set((values || []).map((value) => String(value || '').trim()).filter(Boolean)));
+}
+
+function normalizeAgentTopics(store, agentId, runtimeTopics) {
+  const topics = new Set(runtimeTopics || []);
+  const persisted = store.getAgent(agentId)?.topics || [];
+  for (const topic of persisted) topics.add(topic);
+  return Array.from(topics);
+}
+
+/**
+ * 라우터 생성
+ * @param {object} store
+ */
 export function createRouter(store) {
   let sweepTimer = null;
   let staleTimer = null;
   const responseEmitter = new EventEmitter();
+  const deliveryEmitter = new EventEmitter();
   responseEmitter.setMaxListeners(200);
-
-  const router = {
-    /**
-     * 메시지를 대상에게 라우팅
-     * "topic:XXX" → 토픽 구독자 전체 fanout
-     * 직접 agent_id → 1:1 배달
-     * @returns {number} 배달된 에이전트 수
-     */
-    route(msg) {
-      const to = msg.to_agent ?? msg.to;
-      if (to.startsWith('topic:')) {
-        return store.deliverToTopic(msg.id, to.slice(6));
-      }
-      store.deliverToAgent(msg.id, to);
-      return 1;
-    },
-
-    /**
-     * ask — 질문 요청 (request/reply 패턴)
-     * await_response_ms > 0 이면 짧은 폴링으로 응답 대기
-     * 0 이면 티켓(correlation_id) 즉시 반환
-     */
-    async handleAsk({
-      from, to, topic, question, context_refs,
-      payload = {}, priority = 5, ttl_ms = 300000,
-      await_response_ms = 0, trace_id, correlation_id,
-    }) {
-      const cid = correlation_id || uuidv7();
-      const tid = trace_id || uuidv7();
-
-      const msg = store.enqueueMessage({
-        type: 'request', from, to, topic, priority, ttl_ms,
-        payload: { question, context_refs, ...payload },
-        correlation_id: cid, trace_id: tid,
-      });
-      router.route(msg);
-
-      // 티켓 모드: 즉시 반환
-      if (await_response_ms <= 0) {
-        return {
-          ok: true,
-          data: { request_message_id: msg.id, correlation_id: cid, trace_id: tid, state: 'queued' },
-        };
-      }
-
-      // 이벤트 기반 대기 (최대 30초 제한)
+  deliveryEmitter.setMaxListeners(200);
+
+  const runtimeTopics = new Map();
+  const queuesByAgent = new Map();
+  const liveMessages = new Map();
+  const deliveryLatencies = [];
+
+  function ensureAgentQueue(agentId) {
+    let queue = queuesByAgent.get(agentId);
+    if (!queue) {
+      queue = new Map();
+      queuesByAgent.set(agentId, queue);
+    }
+    return queue;
+  }
+
+  function pruneDeliveryStats(now = Date.now()) {
+    while (deliveryLatencies.length && deliveryLatencies[0].at < now - 300000) {
+      deliveryLatencies.shift();
+    }
+  }
+
+  function upsertRuntimeTopics(agentId, topics, { replace = true } = {}) {
+    const normalized = uniqueStrings(topics);
+    const current = replace ? new Set() : new Set(runtimeTopics.get(agentId) || []);
+    for (const topic of normalized) current.add(topic);
+    runtimeTopics.set(agentId, current);
+    store.updateAgentTopics(agentId, Array.from(current));
+    return Array.from(current);
+  }
+
+  function listRuntimeTopics(agentId) {
+    return normalizeAgentTopics(store, agentId, runtimeTopics.get(agentId));
+  }
+
+  function trackMessage(message, recipients) {
+    liveMessages.set(message.id, {
+      message,
+      recipients: new Set(recipients),
+      ackedBy: new Set(),
+    });
+  }
+
+  function getMessageRecord(messageId) {
+    return liveMessages.get(messageId) || null;
+  }
+
+  function removeMessage(messageId) {
+    const record = liveMessages.get(messageId);
+    if (!record) return;
+    for (const agentId of record.recipients) {
+      queuesByAgent.get(agentId)?.delete(messageId);
+    }
+    liveMessages.delete(messageId);
+  }
+
+  function queueMessage(agentId, message) {
+    const queue = ensureAgentQueue(agentId);
+    queue.set(message.id, {
+      message,
+      attempts: 0,
+      delivered_at_ms: null,
+      acked_at_ms: null,
+    });
+    deliveryEmitter.emit('message', agentId, message);
+  }
+
+  function resolveRecipients(msg) {
+    const to = msg.to_agent ?? msg.to;
+    if (!to?.startsWith('topic:')) {
+      return [to];
+    }
+
+    const topic = to.slice(6);
+    const recipients = new Set();
+    for (const [agentId, topics] of runtimeTopics) {
+      if (topics.has(topic)) recipients.add(agentId);
+    }
+    for (const agent of store.getAgentsByTopic(topic)) {
+      recipients.add(agent.agent_id);
+    }
+    return Array.from(recipients);
+  }
+
+  function sortedPending(agentId, { max_messages = 20, include_topics = null } = {}) {
+    const queue = ensureAgentQueue(agentId);
+    const topicFilter = include_topics?.length ? new Set(include_topics) : null;
+    const now = Date.now();
+    const pending = [];
+
+    for (const delivery of queue.values()) {
+      const { message } = delivery;
+      if (delivery.acked_at_ms) continue;
+      if (message.expires_at_ms <= now) continue;
+      if (topicFilter && !topicFilter.has(message.topic)) continue;
+      pending.push(message);
+    }
+
+    pending.sort((a, b) => {
+      if (b.priority !== a.priority) return b.priority - a.priority;
+      return a.created_at_ms - b.created_at_ms;
+    });
+    return pending.slice(0, max_messages);
+  }
+
+  function markDelivered(agentId, messageId) {
+    const delivery = queuesByAgent.get(agentId)?.get(messageId);
+    const record = getMessageRecord(messageId);
+    if (!delivery || !record) return false;
+
+    delivery.attempts += 1;
+    if (!delivery.delivered_at_ms) {
+      delivery.delivered_at_ms = Date.now();
+      record.message.status = 'delivered';
+      store.updateMessageStatus(messageId, 'delivered');
+      deliveryLatencies.push({
+        at: delivery.delivered_at_ms,
+        ms: delivery.delivered_at_ms - record.message.created_at_ms,
+      });
+      pruneDeliveryStats(delivery.delivered_at_ms);
+      return true;
+    }
+    return false;
+  }
+
+  function ackMessages(ids, agentId) {
+    const now = Date.now();
+    let count = 0;
+
+    for (const id of ids || []) {
+      const delivery = queuesByAgent.get(agentId)?.get(id);
+      const record = getMessageRecord(id);
+      if (!delivery || !record || delivery.acked_at_ms) continue;
+
+      delivery.acked_at_ms = now;
+      record.ackedBy.add(agentId);
+      count += 1;
+
+      if (record.ackedBy.size >= record.recipients.size) {
+        record.message.status = 'acked';
+        store.updateMessageStatus(id, 'acked');
+        removeMessage(id);
+      }
+    }
+
+    return count;
+  }
+
+  function dispatchMessage({ type, from, to, topic, priority = 5, ttl_ms = 300000, payload = {}, trace_id, correlation_id }) {
+    const msg = store.auditLog({
+      type,
+      from,
+      to,
+      topic,
+      priority,
+      ttl_ms,
+      payload,
+      trace_id,
+      correlation_id,
+    });
+    const recipients = uniqueStrings(resolveRecipients(msg));
+    if (recipients.length) {
+      trackMessage(msg, recipients);
+      for (const agentId of recipients) {
+        queueMessage(agentId, msg);
+      }
+      msg.status = 'delivered';
+      store.updateMessageStatus(msg.id, 'delivered');
+    }
+    if (msg.type === 'response') {
+      responseEmitter.emit(msg.correlation_id, msg.payload);
+    }
+    return { msg, recipients };
+  }
+
+  const router = {
+    responseEmitter,
+    deliveryEmitter,
+
+    registerAgent(args) {
+      const result = store.registerAgent(args);
+      upsertRuntimeTopics(args.agent_id, args.topics || [], { replace: true });
+      return result;
+    },
+
+    refreshAgentLease(agentId, ttlMs = 30000) {
+      return store.refreshLease(agentId, ttlMs);
+    },
+
+    subscribeAgent(agentId, topics, { replace = false } = {}) {
+      const nextTopics = upsertRuntimeTopics(agentId, topics, { replace });
+      return { agent_id: agentId, topics: nextTopics };
+    },
+
+    getSubscribedTopics(agentId) {
+      return listRuntimeTopics(agentId);
+    },
+
+    updateAgentStatus(agentId, status) {
+      if (status === 'offline') {
+        runtimeTopics.delete(agentId);
+      }
+      return store.updateAgentStatus(agentId, status);
+    },
+
+    route(msg) {
+      const recipients = uniqueStrings(resolveRecipients(msg));
+      if (!recipients.length) return 0;
+      if (!getMessageRecord(msg.id)) {
+        trackMessage(msg, recipients);
+      }
+      for (const agentId of recipients) {
+        queueMessage(agentId, msg);
+      }
+      store.updateMessageStatus(msg.id, 'delivered');
+      return recipients.length;
+    },
+
+    getPendingMessages(agentId, options = {}) {
+      return sortedPending(agentId, options);
+    },
+
+    markMessagePushed(agentId, messageId) {
+      return markDelivered(agentId, messageId);
+    },
+
+    drainAgent(agentId, { max_messages = 20, include_topics = null, auto_ack = false } = {}) {
+      const messages = sortedPending(agentId, { max_messages, include_topics });
+      for (const message of messages) {
+        markDelivered(agentId, message.id);
+      }
+      if (auto_ack && messages.length) {
+        ackMessages(messages.map((message) => message.id), agentId);
+      }
+      return messages;
+    },
+
+    ackMessages(ids, agentId) {
+      return ackMessages(ids, agentId);
+    },
+
+    async handleAsk({
+      from, to, topic, question, context_refs,
+      payload = {}, priority = 5, ttl_ms = 300000,
+      await_response_ms = 0, trace_id, correlation_id,
+    }) {
+      const cid = correlation_id || uuidv7();
+      const tid = trace_id || uuidv7();
+
+      const { msg } = dispatchMessage({
+        type: 'request',
+        from,
+        to,
+        topic,
+        priority,
+        ttl_ms,
+        payload: { question, context_refs, ...payload },
+        correlation_id: cid,
+        trace_id: tid,
+      });
+
+      if (await_response_ms <= 0) {
+        return {
+          ok: true,
+          data: { request_message_id: msg.id, correlation_id: cid, trace_id: tid, state: 'queued' },
+        };
+      }
+
       try {
-        const [payload] = await once(responseEmitter, cid, {
+        const [response] = await once(responseEmitter, cid, {
           signal: AbortSignal.timeout(Math.min(await_response_ms, 30000)),
         });
         return {
           ok: true,
-          data: { request_message_id: msg.id, correlation_id: cid, trace_id: tid, state: 'answered', response: payload },
+          data: { request_message_id: msg.id, correlation_id: cid, trace_id: tid, state: 'answered', response },
         };
       } catch {
-        // 타임아웃 — DB에서 최종 확인
         const resp = store.getResponseByCorrelation(cid);
         if (resp) {
           return {
@@ -80,121 +313,149 @@ export function createRouter(store) {
           data: { request_message_id: msg.id, correlation_id: cid, trace_id: tid, state: 'delivered' },
         };
       }
-    },
-
-    /**
-     * publish — 이벤트/응답 발행
-     * correlation_id 존재 시 response 타입, 없으면 event 타입
-     */
-    handlePublish({
-      from, to, topic, priority = 5, ttl_ms = 300000,
-      payload = {}, trace_id, correlation_id,
-    }) {
-      const type = correlation_id ? 'response' : 'event';
-      const msg = store.enqueueMessage({
-        type, from, to, topic, priority, ttl_ms, payload,
+    },
+
+    handlePublish({
+      from, to, topic, priority = 5, ttl_ms = 300000,
+      payload = {}, trace_id, correlation_id, message_type,
+    }) {
+      const type = message_type || (correlation_id ? 'response' : 'event');
+      const { msg, recipients } = dispatchMessage({
+        type,
+        from,
+        to,
+        topic,
+        priority,
+        ttl_ms,
+        payload,
+        trace_id: trace_id || uuidv7(),
         correlation_id: correlation_id || uuidv7(),
+      });
+      return {
+        ok: true,
+        data: {
+          message_id: msg.id,
+          fanout_count: recipients.length,
+          expires_at_ms: msg.expires_at_ms,
+        },
+      };
+    },
+
+    handleHandoff({
+      from, to, topic, task, acceptance_criteria, context_refs,
+      priority = 5, ttl_ms = 600000, trace_id, correlation_id,
+    }) {
+      const { msg } = dispatchMessage({
+        type: 'handoff',
+        from,
+        to,
+        topic,
+        priority,
+        ttl_ms,
+        payload: { task, acceptance_criteria, context_refs },
         trace_id: trace_id || uuidv7(),
+        correlation_id: correlation_id || uuidv7(),
       });
-      const fanout = router.route(msg);
-      if (correlation_id) {
-        responseEmitter.emit(correlation_id, msg.payload);
-      }
       return {
         ok: true,
-        data: { message_id: msg.id, fanout_count: fanout, expires_at_ms: msg.expires_at_ms },
+        data: { handoff_message_id: msg.id, state: 'queued', assigned_to: to },
       };
-    },
-
-    /**
-     * handoff — 작업 인계
-     * acceptance_criteria, context_refs 포함 가능
-     */
-    handleHandoff({
-      from, to, topic, task, acceptance_criteria, context_refs,
-      priority = 5, ttl_ms = 600000, trace_id, correlation_id,
-    }) {
-      const msg = store.enqueueMessage({
-        type: 'handoff', from, to, topic, priority, ttl_ms,
-        payload: { task, acceptance_criteria, context_refs },
-        correlation_id: correlation_id || uuidv7(),
-        trace_id: trace_id || uuidv7(),
-      });
-      router.route(msg);
-      return {
-        ok: true,
-        data: { handoff_message_id: msg.id, state: 'queued', assigned_to: to },
-      };
-    },
-
-    // ── 스위퍼 ──
-
-    /** 주기적 만료 정리 시작 (10초: 메시지, 60초: 비활성 에이전트) */
+    },
+
+    sweepExpired() {
+      const now = Date.now();
+      let expired = 0;
+      for (const [messageId, record] of Array.from(liveMessages.entries())) {
+        if (record.message.expires_at_ms > now) continue;
+        store.moveToDeadLetter(messageId, 'ttl_expired', null);
+        removeMessage(messageId);
+        expired += 1;
+      }
+      return { messages: expired };
+    },
+
     startSweeper() {
       if (sweepTimer) return;
       sweepTimer = setInterval(() => {
-        try { store.sweepExpired(); } catch { /* 무시 */ }
+        try { router.sweepExpired(); } catch {}
       }, 10000);
       staleTimer = setInterval(() => {
-        try { store.sweepStaleAgents(); } catch { /* 무시 */ }
+        try { store.sweepStaleAgents(); } catch {}
       }, 120000);
-      sweepTimer.unref();
-      staleTimer.unref();
-    },
-
-    /** 정리 중지 */
-    stopSweeper() {
-      if (sweepTimer) { clearInterval(sweepTimer); sweepTimer = null; }
-      if (staleTimer) { clearInterval(staleTimer); staleTimer = null; }
-    },
-
-    // ── 상태 조회 ──
-
-    /**
-     * 허브/에이전트/큐/트레이스 상태 조회
-     * @param {'hub'|'agent'|'queue'|'trace'} scope
-     */
-    getStatus(scope = 'hub', { agent_id, trace_id, include_metrics = true } = {}) {
-      const data = {};
-
-      if (scope === 'hub' || scope === 'queue') {
-        data.hub = {
-          state: 'healthy',
-          uptime_ms: process.uptime() * 1000 | 0,
-          db_wal_mode: true,
-        };
-        if (include_metrics) {
-          const depths = store.getQueueDepths();
-          const stats = store.getDeliveryStats();
-          data.queues = {
-            urgent_depth: depths.urgent,
-            normal_depth: depths.normal,
-            dlq_depth: depths.dlq,
-            p95_delivery_ms: stats.avg_delivery_ms,
-            timeout_rate: 0,
-          };
-        }
-      }
-
-      if (scope === 'agent' && agent_id) {
-        const agent = store.getAgent(agent_id);
-        if (agent) {
-          data.agent = {
-            agent_id: agent.agent_id,
-            status: agent.status,
-            pending: 0,
-            last_seen_ms: agent.last_seen_ms,
-          };
-        }
-      }
-
-      if (scope === 'trace' && trace_id) {
-        data.trace = store.getMessagesByTrace(trace_id);
-      }
-
-      return { ok: true, data };
-    },
-  };
-
-  return { ...router, responseEmitter };
+      sweepTimer.unref();
+      staleTimer.unref();
+    },
+
+    stopSweeper() {
+      if (sweepTimer) { clearInterval(sweepTimer); sweepTimer = null; }
+      if (staleTimer) { clearInterval(staleTimer); staleTimer = null; }
+    },
+
+    getQueueDepths() {
+      const counts = { urgent: 0, normal: 0, dlq: store.getAuditStats().dlq };
+      for (const record of liveMessages.values()) {
+        const pending = record.recipients.size > record.ackedBy.size;
+        if (!pending) continue;
+        if (record.message.priority >= 7) counts.urgent += 1;
+        else counts.normal += 1;
+      }
+      return counts;
+    },
+
+    getDeliveryStats() {
+      pruneDeliveryStats();
+      if (!deliveryLatencies.length) {
+        return { total_deliveries: 0, avg_delivery_ms: 0 };
+      }
+      const total = deliveryLatencies.reduce((sum, item) => sum + item.ms, 0);
+      return {
+        total_deliveries: deliveryLatencies.length,
+        avg_delivery_ms: Math.round(total / deliveryLatencies.length),
+      };
+    },
+
+    getStatus(scope = 'hub', { agent_id, trace_id, include_metrics = true } = {}) {
+      const data = {};
+
+      if (scope === 'hub' || scope === 'queue') {
+        data.hub = {
+          state: 'healthy',
+          uptime_ms: process.uptime() * 1000 | 0,
+          realtime_transport: 'named-pipe',
+          audit_store: 'sqlite',
+        };
+        if (include_metrics) {
+          const depths = router.getQueueDepths();
+          const stats = router.getDeliveryStats();
+          data.queues = {
+            urgent_depth: depths.urgent,
+            normal_depth: depths.normal,
+            dlq_depth: depths.dlq,
+            avg_delivery_ms: stats.avg_delivery_ms,
+          };
+        }
+      }
+
+      if (scope === 'agent' && agent_id) {
+        const agent = store.getAgent(agent_id);
+        if (agent) {
+          data.agent = {
+            agent_id: agent.agent_id,
+            status: agent.status,
+            pending: sortedPending(agent_id, { max_messages: 1000 }).length,
+            last_seen_ms: agent.last_seen_ms,
+            topics: listRuntimeTopics(agent_id),
+          };
+        }
+      }
+
+      if (scope === 'trace' && trace_id) {
+        data.trace = store.getMessagesByTrace(trace_id);
+      }
+
+      return { ok: true, data };
+    },
+  };
+
+  return router;
 }