wip

winglang · Jan 23, 2024 · c35e1a2 · c35e1a2
1 parent b1136b0
commit c35e1a2
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/websockets/platform/tf-aws.w b/websockets/platform/tf-aws.w
@@ -60,7 +60,7 @@ pub class WebSocket_tfaws impl awsapi.IAwsWebSocket {
       unsafeCast(inflight (event: awsapi.WebSocketAwsRequest): awsapi.WebSocketAwsResponse => {
         if event.requestContext.routeKey == routeKey {
           handler(event.requestContext.connectionId, {
-            headers: event.headers,
+            headers: event.requestContext.headers,
           });
         }
         return {
@@ -79,12 +79,9 @@ pub class WebSocket_tfaws impl awsapi.IAwsWebSocket {
     let onDisconnectFunction = new cloud.Function(
       unsafeCast(inflight (event: awsapi.WebSocketAwsRequest): awsapi.WebSocketAwsResponse => {
         if event.requestContext.routeKey == routeKey {
-          handler(
-            event.requestContext.connectionId,
-            {
-              headers: event.headers,
-            }
-          );
+          handler(event.requestContext.connectionId, {
+            headers: event.requestContext.headers,
+          });
         }
 
         return {