079project 1.0.0 → 2.0.0

package/main_Serve.cjs

@@ -31,11 +31,28 @@ protobuf.load(runtimeProtoPath, (err, root) => {
     if (err) throw err;
     RuntimeMessage = root.lookupType('Runtime');
 });
-// 全局配置
+function parseArgs(argv) {
+  const out = {};
+  for (let i = 0; i < argv.length; i++) {
+    const a = argv[i];
+    if (a && a.startsWith('--')) {
+      const k = a.slice(2);
+      const v = argv[i + 1] && !argv[i + 1].startsWith('--') ? argv[++i] : true;
+      out[k] = v;
+    }
+  }
+  return out;
+}
+const __args = parseArgs(process.argv.slice(2));
 global.config = {
-    masterPortOfSql: 3125,
-    masterPortOfMain: process.argv[2],
-    emitExitport: process.argv[3] || 8641
+  masterPortOfSql: 3125,
+  masterPortOfMain: process.argv[2],
+  emitExitport: process.argv[3] || 8641,
+  groupId: Number(__args['group-id'] || -1),
+  forwarderPort: Number(__args['forwarder-port'] || 0),
+  studyPort: Number(__args['study-port'] || 0),
+  peerServePorts: String(__args['peers'] || '').split(',').map(s => Number(s)).filter(n => Number.isFinite(n) && n > 0),
+  isStudy: !!__args['study']
 };
 const modelDefaults = {
     decayFactor: 0.5,
@@ -2280,6 +2297,78 @@ async function boot() {
     return model;
 }
 
+
+// 激活副本：从快照加载 ctrlB/ctrlC 并加入轮询
+let activeControllers = []; // 轮询数组
+let rrIdx = 0;
+function getNextController() {
+  if (!activeControllers.length) return global.ctrlA;
+  const c = activeControllers[rrIdx % activeControllers.length];
+  rrIdx = (rrIdx + 1) % activeControllers.length;
+  return c;
+}
+async function activateReplicasIfNeeded() {
+  if (global.config.isStudy) return; // study 进程不启用副本扩容
+  if (global.ctrlB && global.ctrlC) return; // 已激活
+  try {
+    const loadReplicaByName = async (name) => {
+      const ctrl = new controller();
+      // 为副本单独挂载快照管理器
+      ctrl.snapshotManager = new SnapshotManager(ctrl.runtime);
+      // 在 A 的快照目录中寻找对应 id
+      const smA = global.ctrlA.snapshotManager;
+      smA.loadSnapshotList();
+      const snap = smA.snapshotList.find(s => s.name === name || s.id.endsWith(`_${name}`));
+      if (!snap) {
+        console.warn(`[REPLICA] 未找到快照: ${name}，尝试现生成...`);
+        await smA.createSnapshot(name);
+      }
+      const snap2 = smA.snapshotList.find(s => s.name === name || s.id.endsWith(`_${name}`));
+      if (snap2) {
+        // 直接读取该文件内容并恢复到 ctrl.runtime
+        const dataStr = fs.readFileSync(snap2.path, 'utf-8');
+        const data = JSON.parse(dataStr);
+        // 简单恢复（与 SnapshotManager.restoreSnapshot 类似）
+        ctrl.runtime.graph = new GraphDB();
+        ctrl.runtime.wordGraph = new GraphDB();
+        ctrl.runtime.kvm = new KVM();
+        ctrl.runtime.wordAccessLog = new Map(data.wordAccessLog || []);
+        if (data.memes) for (const p of data.memes) ctrl.runtime.graph.addPoint(p.pointID, p.connect);
+        if (data.wordGraph) for (const p of data.wordGraph) ctrl.runtime.wordGraph.addPoint(p.pointID, p.connect);
+        if (data.kvm) for (const [k, v] of data.kvm) ctrl.runtime.kvm.set(k, v);
+        if (data.vocab) { ctrl.runtime.vocabManager.vocab = data.vocab; ctrl.runtime.vocabManager.updateMappings(); }
+        return ctrl;
+      }
+      return null;
+    };
+
+    if (!global.ctrlB) global.ctrlB = await loadReplicaByName('replica_B');
+    if (!global.ctrlC) global.ctrlC = await loadReplicaByName('replica_C');
+    activeControllers = [global.ctrlA, ...(global.ctrlB ? [global.ctrlB] : []), ...(global.ctrlC ? [global.ctrlC] : [])];
+    console.log(`[REPLICA] 已激活副本: ${activeControllers.length} 个控制器`);
+  } catch (e) {
+    console.warn('[REPLICA] 激活失败:', e.message);
+  }
+}
+// 对端探测/反触发：同组任一对端端口死亡则激活副本
+function installPeerFailoverMonitor() {
+  const peers = global.config.peerServePorts || [];
+  if (!Array.isArray(peers) || peers.length === 0) return;
+  const axiosInst = axios.create({ timeout: 1500 });
+  setInterval(async () => {
+    try {
+      const checks = await Promise.all(peers.map(p =>
+        axiosInst.get(`http://127.0.0.1:${p}/health`).then(() => true).catch(() => false)
+      ));
+      const dead = checks.some(ok => !ok);
+      if (dead) {
+        console.warn('[ANTI-TRIGGER] 检测到对端 serve 进程离线，启动副本扩容...');
+        await activateReplicasIfNeeded();
+      }
+    } catch (_) { /* 忽略一次错误 */ }
+  }, 3000);
+}
+
 // 在main函数末尾添加
 function setupExitHandler() {
     let isExiting = false;
@@ -2324,6 +2413,27 @@ function setupExitHandler() {
         exitHandler();
     });
 }
+
+// 预缓存副本快照（不在内存常驻）
+async function preCacheReplicas() {
+  try {
+    if (!global.ctrlA?.snapshotManager) return;
+    const sm = global.ctrlA.snapshotManager;
+    sm.loadSnapshotList();
+    const hasB = sm.getSnapshotList().some(s => s.name.includes('replica_B'));
+    const hasC = sm.getSnapshotList().some(s => s.name.includes('replica_C'));
+    if (!hasB) {
+      console.log('[REPLICA] 预生成 ctrlB 快照(replica_B)...');
+      await sm.createSnapshot('replica_B');
+    }
+    if (!hasC) {
+      console.log('[REPLICA] 预生成 ctrlC 快照(replica_C)...');
+      await sm.createSnapshot('replica_C');
+    }
+  } catch (e) {
+    console.warn('[REPLICA] 预缓存失败:', e.message);
+  }
+}
 // 添加定期垃圾回收帮助函数
 function optimizeMemory() {
     const memUsage = process.memoryUsage();
@@ -2340,24 +2450,19 @@ async function main() {
 
     let spiderPrivate = new Spider();
     // 创建三个全局控制器副本
-    const ctrlA = new controller();
-    const ctrlB = new controller();
-    const ctrlC = new controller();
-    global.ctrlA = ctrlA;
-    global.ctrlB = ctrlB;
-    global.ctrlC = ctrlC;
-    global.ctrl = ctrlA;
-    // 启动快照自动保存（每30分钟）
-    if (ctrlA.snapshotManager) {
-        ctrlA.snapshotManager.setupAutoSnapshot(30 * 60 * 1000);
-        console.log('[MAIN] 已设置自动快照，每30分钟检查一次');
-    }
-    setupExitHandler();
-    console.log('Starting AI system...');
-    // 恢复各自持久化数据
-    loadAll(ctrlA.runtime);
-    loadAll(ctrlB.runtime);
-    loadAll(ctrlC.runtime);
+  const ctrlA = new controller();
+  global.ctrlA = ctrlA;
+  global.ctrl = ctrlA;
+  // 启动快照自动保存（每30分钟）
+  if (ctrlA.snapshotManager) {
+    ctrlA.snapshotManager.setupAutoSnapshot(30 * 60 * 1000);
+    console.log('[MAIN] 已设置自动快照，每30分钟检查一次');
+  }
+  setupExitHandler();
+  console.log('Starting AI system...');
+  // 恢复 ctrlA
+  loadAll(ctrlA.runtime);
+    await preCacheReplicas();
     const redisClient = redis.createClient();
     redisClient.connect();
     redisClient.on("error", function (err) {
@@ -2451,36 +2556,33 @@ async function main() {
     await ctrlA.handleInput('I like apple');
     await ctrlA.handleInput('I love orange');
     await ctrlA.handleInput('you are good');
-
-    // 启动主循环（只用A副本）
-    ctrlA.startMainLoop();
-
-    app.post('/api/chat', async (req, res) => {
-        try {
-            const { message } = req.body;
-            // 先分词，再归一化
-            const words = typeof message === 'string'
-                ? message.toLowerCase().split(/\s+/).filter(w => w.length > 0)
-                : [];
-            const normWords = spiderPrivate.lemmatizeWords(words);
-            const normMessage = normWords.join(' ');
-
-      ///      console.log(`[DEBUG] 收到消息: ${message}`);
-      ///      console.log(`[DEBUG] 归一化后: ${normMessage}`);
-
-            const response = await ctrlA.handleInput(normMessage);
-
-            if (!response || response.trim() === '') {
-                console.warn('[WARN] AI响应为空，可能KVM损坏或无记忆');
-            }
-
-     //       console.log(`[DEBUG] 响应: ${response}`);
-            res.json({ response });
-        } catch (error) {
-            console.error('[ERROR] 处理请求失败:', error);
-            res.status(500).json({ error: error.message });
-        }
-    });
+  // 启动主循环（仅 A）
+  ctrlA.startMainLoop();
+
+  // 安装对端监控触发器
+  installPeerFailoverMonitor();
+
+// 替换 /api/chat 为轮询分发（A|B|C）
+app.post('/api/chat', async (req, res) => {
+  try {
+    const { message } = req.body || {};
+    const ctrl = getNextController();
+    // 分词与归一化沿用原逻辑
+    const words = typeof message === 'string'
+      ? message.toLowerCase().split(/\s+/).filter(w => w.length > 0)
+      : [];
+    const normWords = (new Spider()).lemmatizeWords(words);
+    const normMessage = normWords.join(' ');
+    const response = await ctrl.handleInput(normMessage);
+    if (!response || response.trim() === '') {
+      console.warn('[WARN] AI响应为空');
+    }
+    res.json({ response });
+  } catch (error) {
+    console.error('[ERROR] 处理请求失败:', error);
+    res.status(500).json({ error: error.message });
+  }
+});
     // 添加健康检查路由
     app.get('/health', (req, res) => {
         res.status(200).send('OK');
@@ -2580,8 +2682,7 @@ async function main() {
             // 参数校验（可根据需要扩展）
             Object.assign(currentModelParams, updates);
             applyModelParams(global.ctrlA?.runtime);
-            applyModelParams(global.ctrlB?.runtime);
-            applyModelParams(global.ctrlC?.runtime);
+
             res.json({ success: true, params: currentModelParams });
         } catch (err) {
             res.status(500).json({ error: err.message });
@@ -2592,8 +2693,7 @@ async function main() {
         try {
             Object.assign(currentModelParams, modelDefaults);
             applyModelParams(global.ctrlA?.runtime);
-            applyModelParams(global.ctrlB?.runtime);
-            applyModelParams(global.ctrlC?.runtime);
+
             res.json({ success: true, params: currentModelParams });
         } catch (err) {
             res.status(500).json({ error: err.message });
@@ -2722,16 +2822,7 @@ async function plainObjToRuntime(runtime, obj) {
 // 如果直接运行此文件，启动主函数
 // 如果直接运行此文件，启动主函数
 if (require.main === module) {
-    /*antiTrigger(
-        () => main().catch(console.error), // onContinue
-        () => {
-            // onExit: 保存数据
-            try {
-                if (global.ctrl && global.ctrl.runtime) saveAll(global.ctrl.runtime);
-            } catch (e) { }
-        }
-    );
-    */
+
     main().catch(console.error)
 }