@lenne.tech/ng-base 0.3.10 → 0.3.11

package/bundles/lenne.tech-ng-base-src-lib-base.umd.js

@@ -541,8 +541,11 @@
                 return this._token.value;
             },
             set: function (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();
             },
             enumerable: false,
             configurable: true
@@ -588,27 +591,17 @@
             return forward(operation);
         });
         if (baseModuleConfig.wsUrl) {
-            var wsClient = void 0;
-            if (localStorage) {
-                var token = authService.token || null;
-                wsClient = new ws.WebSocketLink({
-                    uri: baseModuleConfig.wsUrl,
-                    options: {
-                        reconnect: true,
-                        connectionParams: {
-                            Authorization: 'Bearer ' + token,
-                        },
-                    },
-                });
-            }
-            else {
-                wsClient = new ws.WebSocketLink({
-                    uri: baseModuleConfig.wsUrl,
-                    options: {
-                        reconnect: true,
-                    },
-                });
-            }
+            var wsLink = new ws.WebSocketLink({
+                uri: baseModuleConfig.wsUrl,
+                options: {
+                    reconnect: true,
+                    connectionParams: function () { return ({
+                        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(core.split(
@@ -618,7 +611,7 @@
                 // @ts-ignore
                 var _b = utilities.getMainDefinition(query), kind = _b.kind, operation = _b.operation;
                 return kind === 'OperationDefinition' && operation === 'subscription';
-            }, wsClient, core.concat(authMiddleware, http)));
+            }, wsLink, core.concat(authMiddleware, http)));
         }
         else {
             links.push(http);