@antseed/cli 0.1.193 → 0.1.194

package/dist/proxy/buyer-proxy.js

@@ -41,6 +41,10 @@ import { buyerConfigReason, buyerHardPolicy, buyerPreferencePolicy, orderByPolic
 // Re-export for backward compatibility (used by tests and other consumers)
 export { selectCandidatePeersForRouting } from './routing.js';
 export { parsePeerPinnedService, rewritePeerPinnedServiceInBody, substituteRoutedModelAlias, ROUTED_MODEL_ALIAS } from './request-utils.js';
+/** On shutdown, requests in flight get this long to finish before their connections are closed. */
+export const BUYER_SHUTDOWN_DRAIN_MS = 60_000;
+/** How long a graceful buyer shutdown may take before the process exits anyway: the proxy's drain, then the node. */
+export const BUYER_SHUTDOWN_DEADLINE_MS = BUYER_SHUTDOWN_DRAIN_MS + 15_000;
 const WATCHER_ABSENCE_ERRORS = {
     'payments-disabled': 'Deposit watcher unavailable — payments are disabled on this buyer.',
     'no-deposit-relay': 'Deposit watcher unavailable — this chain has no deposit relay.',
@@ -1618,8 +1622,13 @@ export class BuyerProxy {
         }
         await this._writeStateFile('stopped');
         await this._conversations.flush();
+        // Requests in flight finish (a supervisor restarting the buyer holds new
+        // ones meanwhile), up to the drain limit.
         return new Promise((resolve) => {
+            const drain = setTimeout(() => this._server.closeAllConnections(), BUYER_SHUTDOWN_DRAIN_MS);
+            drain.unref?.();
             this._server.close(() => {
+                clearTimeout(drain);
                 this._spendAttribution.close();
                 resolve();
             });