079project 8.0.0 → 9.1.0

package/inferenceWorker.cjs

@@ -0,0 +1,94 @@
+'use strict';
+
+// 复用主实现（不会触发 bootstrap，因为主文件有 require.main 守护）
+const core = require('./main.cjs');
+
+const { LmdbStore, NamespacedStore, RuntimeState } = core;
+
+const baseStoresCache = new Map(); // key -> { kvmStore, memeStore, sessionStore, config }
+const runtimeCache = new Map(); // controllerName -> RuntimeState
+
+const parseControllerName = (controllerName) => {
+    const name = String(controllerName || '').trim();
+    // expected: G1_AI3
+    const m = /^(G\d+)_AI(\d+)$/i.exec(name);
+    if (!m) {
+        return { groupId: 'G1', groupIndex: 1 };
+    }
+    return { groupId: String(m[1]).toUpperCase(), groupIndex: Math.max(1, Number(m[2]) || 1) };
+};
+
+const getBaseStores = (config) => {
+    const lmdbRoot = config?.lmdbRoot;
+    const mapSizeBytes = config?.lmdbMapSizeBytes;
+    const key = `${lmdbRoot}|${mapSizeBytes}`;
+    if (baseStoresCache.has(key)) {
+        return baseStoresCache.get(key);
+    }
+    if (!lmdbRoot) {
+        throw new Error('missing-lmdbRoot');
+    }
+    const kvmStore = new LmdbStore({ name: 'kvm', rootDir: lmdbRoot, mapSizeBytes });
+    const memeStore = new LmdbStore({ name: 'meme_graph', rootDir: lmdbRoot, mapSizeBytes });
+    const sessionStore = new LmdbStore({ name: 'session', rootDir: lmdbRoot, mapSizeBytes });
+    const out = { kvmStore, memeStore, sessionStore, config };
+    baseStoresCache.set(key, out);
+    return out;
+};
+
+const getRuntime = ({ controllerName, config }) => {
+    const name = String(controllerName || '').trim();
+    if (!name) throw new Error('missing-controllerName');
+    if (runtimeCache.has(name)) {
+        return runtimeCache.get(name);
+    }
+
+    const base = getBaseStores(config);
+    const { groupId, groupIndex } = parseControllerName(name);
+
+    // 与 ControllerPool 的命名空间保持一致：${groupId}:kvm / ${groupId}:graph / ${groupId}:session
+    const sharedStores = {
+        kvmStore: new NamespacedStore(base.kvmStore, `${groupId}:kvm`),
+        memeStore: new NamespacedStore(base.memeStore, `${groupId}:graph`),
+        sessionStore: new NamespacedStore(base.sessionStore, `${groupId}:session`)
+    };
+
+    const stores = {
+        ...base,
+        ...sharedStores,
+        config: { ...(config || {}), controllerName: name, groupId, groupIndex }
+    };
+
+    const runtime = new RuntimeState(stores);
+    runtimeCache.set(name, runtime);
+    return runtime;
+};
+
+const respond = ({ controllerName, payload, config } = {}) => {
+    const runtime = getRuntime({ controllerName, config });
+    return runtime.processInput(payload || {});
+};
+
+process.on('message', (msg) => {
+    if (!msg || typeof msg !== 'object') return;
+    const { id, cmd } = msg;
+    if (!id) return;
+    if (cmd !== 'respond') {
+        try {
+            process.send({ id, ok: false, error: 'unknown-cmd' });
+        } catch (_e) {
+            // ignore
+        }
+        return;
+    }
+    try {
+        const result = respond(msg);
+        process.send({ id, ok: true, result });
+    } catch (e) {
+        try {
+            process.send({ id, ok: false, error: e?.message || String(e) });
+        } catch (_e) {
+            // ignore
+        }
+    }
+});