@cocorograph/hub-agent 0.6.38 → 0.6.39

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cocorograph/hub-agent",
-  "version": "0.6.38",
+  "version": "0.6.39",
   "description": "Hub Hosted Cockpit のローカル常駐 agent。Hub と outbound WSS で接続し、ローカルの tmux/pty を中継する。",
   "type": "module",
   "license": "UNLICENSED",

package/src/ws-client.mjs

@@ -23,6 +23,12 @@ const MIN_BACKOFF_MS = 1_000
 const MAX_BACKOFF_MS = 30_000
 const BUNDLE_WATCH_DEBOUNCE_MS = 500
 
+// 接続が安定したとみなすまでの猶予。open 直後に即 backoff をリセットすると、
+// open→即 close のフラッピング時に毎回 backoff が最小 (1s) へ戻り、1〜2 秒間隔の
+// 再接続ストロボに陥る (CF/backend 側の瞬断時に hub-agent↔backend WS で観測)。
+// この時間だけ接続を維持できて初めて backoff をリセットする。
+const STABLE_CONNECTION_MS = 10_000
+
 // CF / origin が 5xx を返した直後の再接続は短い backoff だと 5xx キャッシュに
 // 当たって連発失敗するため、最低 5s 待つ。30s リトライまで段階的に伸びる。
 const MIN_BACKOFF_AFTER_5XX_MS = 5_000
@@ -76,6 +82,9 @@ export class WsClient extends EventEmitter {
     this.ptyOutboundBuffer = []
     // 直前の close 原因が CF/origin の 5xx か。次回 backoff の下限を引き上げる。
     this.lastCloseWas5xx = false
+    // 接続が STABLE_CONNECTION_MS 維持できたら backoff/5xx フラグをリセットする
+    // タイマー。close (_clearStableReset) でキャンセルされる。
+    this.stableResetTimer = null
   }
 
   /** WSS 接続を開始する。`stop()` まで自動で reconnect 続行。 */
@@ -104,6 +113,7 @@ export class WsClient extends EventEmitter {
     })
 
     ws.on("close", (code, reason) => {
+      this._clearStableReset()
       this._stopHeartbeat()
       this._stopBundleWatcher()
       this.logger?.info({ code, reason: reason?.toString() }, "ws close")
@@ -136,8 +146,11 @@ export class WsClient extends EventEmitter {
    * のクロージャだと spy しづらい)。
    */
   _onOpen() {
-    this.backoff = MIN_BACKOFF_MS
-    this.lastCloseWas5xx = false
+    // backoff の即時リセットはしない。接続が STABLE_CONNECTION_MS 維持できてから
+    // _armStableReset() でリセットする (open→即 close フラッピング時の再接続
+    // ストロボ対策)。すぐ切れた接続では backoff がリセットされず指数バックオフが
+    // 効き続けるため、1〜2 秒間隔の再接続ループに陥らない。
+    this._armStableReset()
     this.logger?.info("ws open")
     // 切断中に積んだ pty.data を flush。hello より先に送ると backend で stream
     // 未登録のため unknown_stream error になる可能性があるが、hello の応答待ち
@@ -232,6 +245,7 @@ export class WsClient extends EventEmitter {
   /** Reconnect を止めて切断する。 */
   stop() {
     this.stopped = true
+    this._clearStableReset()
     this._stopHeartbeat()
     this._stopBundleWatcher()
     if (this.reconnectTimer) {
@@ -428,4 +442,29 @@ export class WsClient extends EventEmitter {
     }, delay)
     this.reconnectTimer.unref?.()
   }
+
+  /**
+   * 接続が STABLE_CONNECTION_MS 維持できたら backoff と 5xx フラグをリセットする
+   * タイマーを張る (open 直後に呼ぶ)。close で _clearStableReset によりキャンセル
+   * されるため、安定しなかった (= すぐ切れた) 接続では backoff がリセットされず
+   * 指数バックオフが効き続け、再接続ストロボを防ぐ。
+   */
+  _armStableReset() {
+    this._clearStableReset()
+    this.stableResetTimer = setTimeout(() => {
+      this.stableResetTimer = null
+      this.backoff = MIN_BACKOFF_MS
+      this.lastCloseWas5xx = false
+      this.logger?.debug("ws connection stable, backoff reset")
+    }, STABLE_CONNECTION_MS)
+    this.stableResetTimer.unref?.()
+  }
+
+  /** 安定リセットタイマーをキャンセルする (close / stop 時)。 */
+  _clearStableReset() {
+    if (this.stableResetTimer) {
+      clearTimeout(this.stableResetTimer)
+      this.stableResetTimer = null
+    }
+  }
 }