@agentunion/fastaun 0.5.36 → 0.5.37

package/dist/client.js

@@ -231,9 +231,32 @@ function clampReconnectDelayMs(value, fallback, upper = RECONNECT_MAX_BASE_DELAY
     return Math.min(Math.max(ms, RECONNECT_MIN_BASE_DELAY_MS), upper);
 }
 function reconnectSleepDelayMs(baseDelay, maxBaseDelay) {
-    // 抖动只随当前退避基数增长，并限制在 maxBaseDelay 内。
-    const jitterLimit = Math.min(Math.max(baseDelay, 0), Math.max(maxBaseDelay - baseDelay, 0));
-    return baseDelay + Math.random() * jitterLimit;
+    return Math.max(1, Math.random() * Math.min(baseDelay, maxBaseDelay));
+}
+const reconnectSlots = new Map();
+async function acquireReconnectSlot(aid, wait, pause) {
+    const issuer = aid.slice(aid.indexOf('.') + 1).toLowerCase();
+    let slots = reconnectSlots.get(issuer);
+    if (!slots) {
+        slots = { active: new Set(), next: 0 };
+        reconnectSlots.set(issuer, slots);
+    }
+    while (slots.active.size >= 2)
+        await wait(Promise.race(slots.active));
+    let done;
+    const ticket = new Promise(resolve => { done = resolve; });
+    slots.active.add(ticket);
+    const release = () => { slots.active.delete(ticket); done(); };
+    const delay = Math.min(500, Math.max(0, slots.next - Date.now()));
+    slots.next = Date.now() + delay + 250 + Math.random() * 250;
+    try {
+        await pause(delay);
+        return release;
+    }
+    catch (error) {
+        release();
+        throw error;
+    }
 }
 /** peer 证书缓存 TTL（1 小时） */
 const PEER_CERT_CACHE_TTL = 3600;
@@ -743,7 +766,7 @@ export class AUNClient {
             if (current())
                 this._credentialRenewalTimer = setTimeout(() => { void scan(); }, 60_000);
         };
-        void scan();
+        this._credentialRenewalTimer = setTimeout(() => { void scan(); }, 30_000 + Math.random() * 30_000);
     }
     _stopCredentialRenewal() {
         this._credentialRenewalAbort?.abort();
@@ -2475,28 +2498,43 @@ export class AUNClient {
         tokens.saveMetadata(orgAid, { org_identity_creation: null });
         return checkpoint.result;
     }
-    _schedulePostConnectRecovery() {
+    _schedulePostConnectRecovery(backgroundSync = this._sessionOptions.background_sync !== false) {
         this._postConnectRecoveryAbort?.abort();
         if (this._postConnectRecoveryTimer !== null)
             clearTimeout(this._postConnectRecoveryTimer);
         this._postConnectRecoveryTimer = null;
         const store = this._aidStore;
-        if (this._sessionOptions.background_sync === false)
-            return;
         const abort = new AbortController();
         this._postConnectRecoveryAbort = abort;
         const aid = this._aid, version = this._serviceEventIdentityVersion, connection = this._entityOwnershipConnection;
         const current = () => !abort.signal.aborted && aid === this._aid && version === this._serviceEventIdentityVersion && connection === this._entityOwnershipConnection && ['ready', 'connected'].includes(this._state) && !this._closing && !this._disconnecting;
+        const tasks = [];
+        if (!this._ownershipOperation)
+            tasks.push(() => this._checkIdentityAdmissionAfterConnect());
+        tasks.push(() => this._reconcileReadStates());
+        if (backgroundSync) {
+            tasks.push(() => this._recoverGroupInvites());
+            if (this._v2Session && this._sessionOptions.connection_kind !== 'short')
+                tasks.push(() => this._v2AutoConfirmPendingProposals());
+            tasks.push(() => this._delivery.restoreMessageClassPulls(), () => this._fillP2pGap());
+            if (store && !this._ownershipOperation)
+                tasks.push(() => this._resumeOrgIdentityAfterConnect(store, current));
+        }
+        const run = async () => {
+            for (let index = 0; index < tasks.length; index++) {
+                if (!current())
+                    return;
+                const pending = tasks[index]();
+                this._safeAsync(pending);
+                await pending.catch(() => { });
+                if (index + 1 < tasks.length)
+                    await this._sleep(1000 + Math.random() * 4000, abort.signal);
+            }
+        };
         this._postConnectRecoveryTimer = setTimeout(() => {
             this._postConnectRecoveryTimer = null;
-            if (!current())
-                return;
-            this._safeAsync(this._delivery.restoreMessageClassPulls());
-            void this._fillP2pGap().catch((exc) => {
-                this._clientLog.warn(`schedule post-connect P2P gap fill failed: ${formatCaughtError(exc)}`);
-            });
-            if (store && !this._ownershipOperation)
-                this._safeAsync(this._resumeOrgIdentityAfterConnect(store, current));
+            if (current())
+                this._safeAsync(run());
         }, 30_000 + Math.random() * 30_000);
         this._postConnectRecoveryTimer.unref?.();
     }
@@ -4990,18 +5028,11 @@ export class AUNClient {
             }
             this._lifecycle.assertConnectionAttemptOwner(connectionOwner);
             this._startBackgroundTasks();
-            this._safeAsync(this._reconcileReadStates());
             const connectionKind = String(params.connection_kind ?? 'long');
             const isShortConnection = connectionKind === 'short';
             const hasExplicitBackgroundSync = Object.prototype.hasOwnProperty.call(params, 'background_sync');
             const backgroundSyncEnabled = this._sessionOptions.background_sync !== false
                 && (!isShortConnection || hasExplicitBackgroundSync);
-            if (backgroundSyncEnabled) {
-                // 连接建立/恢复后先从事实列表恢复定向邀请提示；失败不影响连接本身。
-                this._safeAsync(this._recoverGroupInvites().catch((exc) => {
-                    this._clientLog.debug(`group invite recovery failed: ${formatCaughtError(exc)}`);
-                }));
-            }
             if (!isShortConnection) {
                 await this._v2E2EE.onConnected({ backgroundSync: backgroundSyncEnabled });
                 this._assertReconnectOwner(reconnectOwner);
@@ -5016,13 +5047,7 @@ export class AUNClient {
             // 群消息按惰性触发，不在此处主动 pull
             this._entityOwnershipConnection++;
             this._entityOwnershipChecks.clear();
-            if (!this._ownershipOperation)
-                this._safeAsync(this._checkIdentityAdmissionAfterConnect());
-            if (backgroundSyncEnabled)
-                this._schedulePostConnectRecovery();
-            if (!isShortConnection && backgroundSyncEnabled && this._v2Session) {
-                this._safeAsync(this._v2AutoConfirmPendingProposals());
-            }
+            this._schedulePostConnectRecovery(backgroundSyncEnabled);
             this._activateUdpStandby();
             this._clientLog.debug(`_connectOnce exit: elapsed=${Date.now() - tStart}ms gateway=${gatewayUrl}, aid=${this._aid ?? ''}`);
         }
@@ -6463,8 +6488,17 @@ export class AUNClient {
             reconnect: async () => {
                 if (this._closing || this._disconnecting)
                     throw new ConnectionError('client closed');
-                this._state = 'standby';
-                await this._lifecycle.connect(this._udpStandbyConnectOptions);
+                await this._sleep(reconnectSleepDelayMs(1000, RECONNECT_MAX_BASE_DELAY_MS));
+                const release = await acquireReconnectSlot(this._aid ?? '', task => task, ms => this._sleep(ms));
+                try {
+                    if (this._closing || this._disconnecting || this._udpStandby?.state === 'offline')
+                        throw new ConnectionError('client closed');
+                    this._state = 'standby';
+                    await this._lifecycle.connect(this._udpStandbyConnectOptions);
+                }
+                finally {
+                    release();
+                }
             },
         });
         this._udpStandby.activate(this._udpStandbyPolicy);
@@ -6592,9 +6626,10 @@ export class AUNClient {
                 next_retry_at: this._nextRetryAt,
             }))
                 return;
+            let releaseReconnectSlot;
             try {
-                // 抖动随当前 base 增长且不超过 max_base，避免首次重连被放大到几十秒。
                 await this._sleep(Math.max(0, this._nextRetryAt - Date.now()), reconnectAbort?.signal);
+                releaseReconnectSlot = await acquireReconnectSlot(this._aid ?? '', task => this._lifecycle.waitForAttempt(task, undefined, reconnectAbort ?? undefined), ms => this._sleep(ms, reconnectAbort?.signal));
                 if (!this._ownsReconnect(reconnectAbort))
                     return;
                 this._state = 'reconnecting';
@@ -6603,17 +6638,6 @@ export class AUNClient {
                     attempt,
                 }))
                     return;
-                // /health 仅作旁路诊断，不能阻止实际 WebSocket 重连。
-                if (this._gatewayUrl) {
-                    void this._discovery.checkHealth(this._gatewayUrl, 5_000)
-                        .then((healthy) => {
-                        if (!healthy)
-                            this._clientLog.warn(`reconnect health advisory failed: gateway=${this._gatewayUrl}`);
-                    })
-                        .catch((exc) => {
-                        this._clientLog.debug(`reconnect health advisory error: ${formatCaughtError(exc)}`);
-                    });
-                }
                 await this._transport.close();
                 if (!this._ownsReconnect(reconnectAbort))
                     return;
@@ -6650,7 +6674,7 @@ export class AUNClient {
                         this._sessionParams.access_token = '';
                     }
                 }
-                const candidates = await this._lifecycle.waitForAttempt(this._resolveGatewayCandidatesForAid(this._aid, true, reconnectAbort ?? undefined), undefined, reconnectAbort ?? undefined);
+                const candidates = await this._lifecycle.waitForAttempt(this._resolveGatewayCandidatesForAid(this._aid, this._migrationPending || this._migrationRevision.size > 0 || attempt % 3 === 2, reconnectAbort ?? undefined), undefined, reconnectAbort ?? undefined);
                 if (!this._ownsReconnect(reconnectAbort))
                     return;
                 this._sessionParams.gateways = [...candidates];
@@ -6710,6 +6734,9 @@ export class AUNClient {
                 }
                 delay = Math.min(delay * 2, maxBaseDelay);
             }
+            finally {
+                releaseReconnectSlot?.();
+            }
         }
     }
     /** 每轮重连从当前地址之后开始轮转，避免永远黏在失效的首个 Gateway。 */