@agentunion/fastaun 0.5.34 → 0.5.35

package/dist/transport.js

@@ -601,6 +601,9 @@ export class RPCTransport {
     _budgetWaiters = new Set();
     _requestAdmission = null;
     _requestCheck = null;
+    _standbyGate = null;
+    _businessActivity = null;
+    _standbyAuthParams = null;
     _riskControlHandler = null;
     _classMessageHandler = null;
     _protocolMetaObserver = false;
@@ -876,6 +879,8 @@ export class RPCTransport {
         }
         if (this._workerGeneration === null || message.generation !== this._workerGeneration)
             return;
+        if (message.event && !['_transport.disconnect', '_transport.observer', '_transport.sync_control', '_transport.rate_limit'].includes(message.event))
+            this._businessActivity?.();
         if (message.event === '_transport.sync_control') {
             const value = message.payload;
             this._syncControlObserver?.(value.meta, value.method, value.params ?? {});
@@ -1057,6 +1062,11 @@ export class RPCTransport {
             return this._workerConnected;
         return !this._closed && this._ws !== null && this._ws.readyState === WebSocket.OPEN;
     }
+    setStandbyGate(gate, inbound = null, authentication = null) {
+        this._standbyGate = gate;
+        this._businessActivity = inbound;
+        this._standbyAuthParams = authentication;
+    }
     /**
      * 注册 RPC envelope _meta 字段观察者；observer(meta) 在每次成功 RPC 时调用。
      *
@@ -1140,7 +1150,7 @@ export class RPCTransport {
         const value = guard;
         const policy = this._syncAdmission;
         if (!policy || policy.parent_connection_id !== value.parent_connection_id || policy.version !== value.version
-            || Number(policy.max_history_pull_inflight) <= 0 || Number(policy.max_history_inflight_bytes) <= 0 || Number(policy.history_bytes_per_second) <= 0) {
+            || Number(policy.max_history_pull_inflight) <= 0 || Number(policy.max_history_inflight_bytes) <= 0) {
             throw new StateError('history sync policy changed before send', { data: { sync_policy_changed: true }, retryable: true });
         }
     }
@@ -1709,7 +1719,29 @@ export class RPCTransport {
         const result = Promise.all([request, sent]).then(([response]) => response);
         return Object.assign(result, { cancel: request.cancel });
     }
-    _call(method, params, timeout, trace, background = false, onSent, sentCheckpoint) {
+    _call(method, params, timeout, trace, background = false, onSent, sentCheckpoint, bypassStandby = false) {
+        if (method === 'auth.connect' && this._standbyAuthParams)
+            params = this._standbyAuthParams(params ?? {});
+        if (this._standbyGate && !bypassStandby && !method.startsWith('gateway.standby.')
+            && !['auth.aid_login1', 'auth.aid_login2', 'auth.connect'].includes(method)) {
+            const deadline = Math.min(typeof params?._rpc_deadline === 'number' ? params._rpc_deadline : Infinity, Date.now() + (timeout ?? this._timeout));
+            const abort = new AbortController();
+            let active;
+            const gate = this._standbyGate;
+            const waiting = (async () => {
+                const release = await gate(method, deadline, abort.signal);
+                try {
+                    if (abort.signal.aborted || Date.now() >= deadline)
+                        throw new TimeoutError('rpc timeout during standby recovery', { retryable: true });
+                    active = this._call(method, params, deadline - Date.now(), trace, background, onSent, sentCheckpoint, true);
+                    return await active;
+                }
+                finally {
+                    release();
+                }
+            })();
+            return Object.assign(waiting, { cancel: () => { abort.abort(); active?.cancel?.(); } });
+        }
         const selected = normalizeRpcBackground(background !== false ? background : params?._rpc_background);
         const kind = selected === true && ['message.peers', 'message.pull', 'message.v2.pull', 'message.ack', 'message.v2.ack', 'group.pull', 'group.v2.pull', 'group.pull_events', 'group.ack_messages', 'group.v2.ack'].includes(method) ? 'pull' : selected;
         const deadline = Math.min(typeof params?._rpc_deadline === 'number' ? params._rpc_deadline : Infinity, Date.now() + (timeout ?? this._timeout));
@@ -1916,10 +1948,13 @@ export class RPCTransport {
     /** 发送 JSON-RPC 2.0 Notification，不分配 id，也不等待响应。 */
     async notify(method, params) {
         const release = this._requestAdmission?.(`rpc:${String(method).replace(/^(notification|event)\//, '')}`);
+        let standbyRelease;
         try {
+            standbyRelease = await this._standbyGate?.(method, Date.now() + this._timeout);
             await this._notifyAdmitted(method, params);
         }
         finally {
+            standbyRelease?.();
             release?.();
         }
     }
@@ -2434,6 +2469,8 @@ export class RPCTransport {
     }
     /** 路由消息：RPC 响应 / challenge / 事件 / 通知 */
     _routeMessage(message) {
+        if (typeof message.method === 'string' && !['meta.ping', 'system.ping', 'ping'].includes(message.method))
+            this._businessActivity?.();
         const syncMethod = String(message.method ?? '');
         const syncMeta = isJsonObject(message._meta) ? message._meta : {};
         if (syncMeta.sync_policy !== undefined || syncMethod.startsWith('event/gateway.sync_')) {