Add ping/pong message support required by graphql-ws

apollographql · Feb 22, 2022 · 348dcf0 · 348dcf0
1 parent 37336c5
commit 348dcf0
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/Sources/ApolloWebSocket/OperationMessage.swift b/Sources/ApolloWebSocket/OperationMessage.swift
@@ -19,6 +19,9 @@ final class OperationMessage {
     case error = "error"                               // Server -> Client
     case complete = "complete"                         // Server -> Client
     case next = "next"                                 // Server -> Client
+
+    case ping = "ping"                                 // Bidirectional
+    case pong = "pong"                                 // Bidirectional
   }
 
   let serializationFormat = JSONSerializationFormat.self

diff --git a/Sources/ApolloWebSocket/WebSocketTransport.swift b/Sources/ApolloWebSocket/WebSocketTransport.swift
@@ -181,9 +181,16 @@ public class WebSocketTransport {
         writeQueue()
 
       case .connectionKeepAlive,
-           .startAck:
+           .startAck,
+           .pong:
         writeQueue()
 
+      case .ping:
+        if let str = OperationMessage(type: .pong).rawMessage {
+          write(str)
+          writeQueue()
+        }
+
       case .connectionInit,
            .connectionTerminate,
            .subscribe,