@lenne.tech/ng-base 0.3.10 → 0.3.11

package/fesm2015/lenne.tech-ng-base.js

@@ -190,8 +190,11 @@ class AuthService {
         return this._token.value;
     }
     set token(token) {
+        var _a, _b;
         this._token.next(token);
         this.storageService.save('token', token);
+        (_a = this.subscriptionClient) === null || _a === void 0 ? void 0 : _a.close();
+        (_b = this.subscriptionClient) === null || _b === void 0 ? void 0 : _b.connect();
     }
     get tokenObservable() {
         return this._token.asObservable();
@@ -229,27 +232,17 @@ function apolloOptionsFactory(baseModuleConfig, httpLink, authService) {
         return forward(operation);
     });
     if (baseModuleConfig.wsUrl) {
-        let wsClient;
-        if (localStorage) {
-            const token = authService.token || null;
-            wsClient = new WebSocketLink({
-                uri: baseModuleConfig.wsUrl,
-                options: {
-                    reconnect: true,
-                    connectionParams: {
-                        Authorization: 'Bearer ' + token,
-                    },
-                },
-            });
-        }
-        else {
-            wsClient = new WebSocketLink({
-                uri: baseModuleConfig.wsUrl,
-                options: {
-                    reconnect: true,
-                },
-            });
-        }
+        const wsLink = new WebSocketLink({
+            uri: baseModuleConfig.wsUrl,
+            options: {
+                reconnect: true,
+                connectionParams: () => ({
+                    Authorization: authService.token ? 'Bearer ' + authService.token : undefined,
+                }),
+            },
+        });
+        // Set client for reconnection on logout/login
+        authService.subscriptionClient = wsLink.subscriptionClient;
         // using the ability to split links, you can send data to each link
         // depending on what kind of operation is being sent
         links.push(split(
@@ -258,7 +251,7 @@ function apolloOptionsFactory(baseModuleConfig, httpLink, authService) {
             // @ts-ignore
             const { kind, operation } = getMainDefinition(query);
             return kind === 'OperationDefinition' && operation === 'subscription';
-        }, wsClient, concat(authMiddleware, http)));
+        }, wsLink, concat(authMiddleware, http)));
     }
     else {
         links.push(http);