fix: add user-agent to websocket request (#14)

aws-amplify · Sep 4, 2024 · 288d164 · 288d164
1 parent cbac364
commit 288d164
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Sources/AWSAppSyncApolloExtensions/Websocket/AppSyncWebSocketClient.swift b/Sources/AWSAppSyncApolloExtensions/Websocket/AppSyncWebSocketClient.swift
@@ -129,8 +129,8 @@ public class AppSyncWebSocketClient: NSObject, ApolloWebSocket.WebSocketClient,
 
             var request = self.request
             request.httpBody = Data(startRequest.data.utf8)
-            let headers = try await authorizer.getWebSocketSubscriptionPayload(request: request)
-
+            var headers = try await authorizer.getWebSocketSubscriptionPayload(request: request)
+            headers.updateValue(await PackageInfo.userAgent, forKey: "User-Agent")
             let interceptedEvent = AppSyncRealTimeStartRequest(
                 id: startRequest.id,
                 data: startRequest.data,