fix(agent): crush on real mode account publishing (#202)

No-Trade-No-Life · Oct 30, 2023 · 4dbaa50 · 4dbaa50
1 parent 5fda80a
commit 4dbaa50
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 4 deletions.
diff --git a/common/changes/@yuants/agent/2023-10-30-15-20.json b/common/changes/@yuants/agent/2023-10-30-15-20.json
@@ -0,0 +1,10 @@
+{
+  "changes": [
+    {
+      "packageName": "@yuants/agent",
+      "comment": "crush on real mode account publishing",
+      "type": "patch"
+    }
+  ],
+  "packageName": "@yuants/agent"
+}
diff --git a/libraries/agent/src/AgentScene.ts b/libraries/agent/src/AgentScene.ts
@@ -194,12 +194,12 @@ export const AgentScene = async (terminal: Terminal, agentConf: IAgentConf) => {
   if (agentConf.publish_account) {
     const unit = new BasicUnit(kernel);
     const mapAccountIdToAccountInfo$: Record<string, Subject<IAccountInfo>> = {};
-    accountInfoUnit.mapAccountIdToAccountInfo.forEach((accountInfo) => {
-      const accountInfo$ = (mapAccountIdToAccountInfo$[accountInfo.account_id] = new Subject());
-      terminal.provideAccountInfo(accountInfo$);
-    });
     unit.onIdle = () => {
       for (const accountInfo of accountInfoUnit.mapAccountIdToAccountInfo.values()) {
+        if (!mapAccountIdToAccountInfo$[accountInfo.account_id]) {
+          mapAccountIdToAccountInfo$[accountInfo.account_id] = new Subject();
+          terminal.provideAccountInfo(mapAccountIdToAccountInfo$[accountInfo.account_id]);
+        }
         const accountInfo$ = mapAccountIdToAccountInfo$[accountInfo.account_id];
         accountInfo$.next(accountInfo);
       }