AAE-20808 Implementing different approach to create http link handler

Alfresco · Dec 4, 2024 · a914029 · a914029
1 parent 60c0803
commit a914029
Showing 1 changed file with 19 additions and 14 deletions.
diff --git a/lib/process-services-cloud/src/lib/services/web-socket.service.ts b/lib/process-services-cloud/src/lib/services/web-socket.service.ts
@@ -23,23 +23,23 @@ import {
     DefaultContext,
     FetchResult,
     from,
-    HttpLink,
     InMemoryCache,
     InMemoryCacheConfig,
     NextLink,
     Operation,
     split,
     SubscriptionOptions
 } from '@apollo/client/core';
-import { getMainDefinition } from '@apollo/client/utilities';
-import { Kind, OperationTypeNode } from 'graphql';
+import { Observable } from 'rxjs';
 import { Apollo } from 'apollo-angular';
+import { HttpLink, HttpLinkHandler } from 'apollo-angular/http';
+import { Kind, OperationTypeNode } from 'graphql';
 import { onError } from '@apollo/client/link/error';
 import { RetryLink } from '@apollo/client/link/retry';
-import { Observable } from 'rxjs';
+import { getMainDefinition } from '@apollo/client/utilities';
 import { filter, map, switchMap, take, tap } from 'rxjs/operators';
-import { FeaturesServiceToken, IFeaturesService } from '@alfresco/adf-core/feature-flags';
 import { AuthenticationService, AppConfigService } from '@alfresco/adf-core';
+import { FeaturesServiceToken, IFeaturesService } from '@alfresco/adf-core/feature-flags';
 
 interface serviceOptions {
     apolloClientName: string;
@@ -55,10 +55,11 @@ export class WebSocketService {
     private host = '';
     private subscriptionProtocol: 'graphql-ws' | 'transport-ws' = 'transport-ws';
     private wsLink: GraphQLWsLink | WebSocketLink;
-    private httpLink: HttpLink;
+    private httpLinkHandler: HttpLinkHandler;
 
     constructor(
         private readonly apollo: Apollo,
+        private readonly httpLink: HttpLink,
         private readonly appConfigService: AppConfigService,
         private readonly authService: AuthenticationService,
         @Optional() @Inject(FeaturesServiceToken) private featuresService: IFeaturesService
@@ -117,19 +118,15 @@ export class WebSocketService {
                 throw new Error('Unknown subscription protocol');
         }
 
-        this.httpLink = options.httpUrl
-            ? new HttpLink({
-                  uri: this.createHttpUrl(options.httpUrl)
-              })
-            : undefined;
+        this.createHttpLinkHandler(options);
 
         const link = split(
             ({ query }) => {
                 const definition = getMainDefinition(query);
                 return definition.kind === Kind.OPERATION_DEFINITION && definition.operation === OperationTypeNode.SUBSCRIPTION;
             },
             this.wsLink,
-            this.httpLink
+            this.httpLinkHandler
         );
 
         const authLink = (operation: Operation, forward: NextLink) => {
@@ -179,7 +176,7 @@ export class WebSocketService {
         });
     }
 
-    private createTransportWsLink(options: serviceOptions) {
+    private createTransportWsLink(options: serviceOptions): void {
         this.wsLink = new WebSocketLink({
             uri: this.createWsUrl(options.wsUrl) + '/ws/graphql',
             options: {
@@ -193,7 +190,7 @@ export class WebSocketService {
         });
     }
 
-    private createGraphQLWsLink(options: serviceOptions) {
+    private createGraphQLWsLink(options: serviceOptions): void {
         this.wsLink = new GraphQLWsLink(
             createClient({
                 url: this.createWsUrl(options.wsUrl) + '/v2/ws/graphql',
@@ -210,4 +207,12 @@ export class WebSocketService {
             })
         );
     }
+
+    private createHttpLinkHandler(options: serviceOptions): void {
+        this.httpLinkHandler = options.httpUrl
+            ? this.httpLink.create({
+                  uri: this.createHttpUrl(options.httpUrl)
+              })
+            : undefined;
+    }
 }