move creation of header object into try catch block

awslabs · Feb 14, 2022 · 7d15cbd · 7d15cbd
1 parent 92b50c4
commit 7d15cbd
Showing 1 changed file with 29 additions and 27 deletions.
diff --git a/packages/aws-appsync-subscription-link/src/realtime-subscription-handshake-link.ts b/packages/aws-appsync-subscription-link/src/realtime-subscription-handshake-link.ts
@@ -259,37 +259,39 @@ export class AppSyncRealTimeSubscriptionHandshakeLink extends ApolloLink {
     // Preparing payload for subscription message
 
     const dataString = JSON.stringify(data);
-    const headerObj = {
-      ...(await this._awsRealTimeHeaderBasedAuth({
-        apiKey,
-        appSyncGraphqlEndpoint,
-        authenticationType,
-        payload: dataString,
-        canonicalUri: "",
-        region,
-        credentials,
-        token,
-        graphql_headers
-      })),
-      [USER_AGENT_HEADER]: USER_AGENT
-    };
+    let stringToAWSRealTime: string;
 
-    const subscriptionMessage = {
-      id: subscriptionId,
-      payload: {
-        data: dataString,
-        extensions: {
-          authorization: {
-            ...headerObj
+    try {
+      const headerObj = {
+        ...(await this._awsRealTimeHeaderBasedAuth({
+          apiKey,
+          appSyncGraphqlEndpoint,
+          authenticationType,
+          payload: dataString,
+          canonicalUri: "",
+          region,
+          credentials,
+          token,
+          graphql_headers
+        })),
+        [USER_AGENT_HEADER]: USER_AGENT
+      };
+
+      const subscriptionMessage = {
+        id: subscriptionId,
+        payload: {
+          data: dataString,
+          extensions: {
+            authorization: {
+              ...headerObj
+            }
           }
-        }
-      },
-      type: MESSAGE_TYPES.GQL_START
-    };
+        },
+        type: MESSAGE_TYPES.GQL_START
+      };
 
-    const stringToAWSRealTime = JSON.stringify(subscriptionMessage);
+      stringToAWSRealTime = JSON.stringify(subscriptionMessage);
 
-    try {
       await this._initializeWebSocketConnection({
         apiKey,
         appSyncGraphqlEndpoint,