fix: keep alive

izatop · Jan 20, 2021 · 4ab9cda · 4ab9cda
1 parent 3439a72
commit 4ab9cda
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/packages/ws/src/Protocol/GQL/GQLProtoLayer.ts b/packages/ws/src/Protocol/GQL/GQLProtoLayer.ts
@@ -25,7 +25,10 @@ export class GQLProtoLayer {
     constructor(client: GQLClientConnection, factory: GQLSubscribeFunction) {
         this.#client = client;
         this.#subscribe = factory;
+
+        const interval = setInterval(() => this.keepAliveUpdate(), 30000);
         this.#client.on("close", () => this.unsubscribeAll());
+        this.#client.on("close", () => clearInterval(interval));
     }
 
     public async handle(operation: GQLOperationMessage): Promise<void> {
@@ -34,6 +37,7 @@ export class GQLProtoLayer {
             case GQLClientOperationType.CONNECTION_INIT:
                 Object.assign(this.#params, operation.payload);
                 await this.#client.send({type: GQLServerOperationType.CONNECTION_ACK});
+                await this.keepAliveUpdate();
                 break;
             case GQLClientOperationType.CONNECTION_TERMINATE:
                 this.terminate();
@@ -51,6 +55,10 @@ export class GQLProtoLayer {
         }
     }
 
+    private keepAliveUpdate() {
+        return this.#client.send({type: GQLServerOperationType.CONNECTION_KEEP_ALIVE});
+    }
+
     private unsubscribeAll() {
         for (const subscription of this.#subscriptions.values()) {
             subscription.return?.();