Igor Kulikov
7 years ago
11 changed files with 641 additions and 10 deletions
@ -0,0 +1,309 @@ |
|||
///
|
|||
/// Copyright © 2016-2019 The Thingsboard Authors
|
|||
///
|
|||
/// Licensed under the Apache License, Version 2.0 (the "License");
|
|||
/// you may not use this file except in compliance with the License.
|
|||
/// You may obtain a copy of the License at
|
|||
///
|
|||
/// http://www.apache.org/licenses/LICENSE-2.0
|
|||
///
|
|||
/// Unless required by applicable law or agreed to in writing, software
|
|||
/// distributed under the License is distributed on an "AS IS" BASIS,
|
|||
/// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|||
/// See the License for the specific language governing permissions and
|
|||
/// limitations under the License.
|
|||
///
|
|||
|
|||
import { Inject, Injectable } from '@angular/core'; |
|||
import { |
|||
AttributesSubscriptionCmd, |
|||
GetHistoryCmd, |
|||
SubscriptionCmd, |
|||
SubscriptionUpdate, |
|||
SubscriptionUpdateMsg, |
|||
TelemetryFeature, |
|||
TelemetryPluginCmdsWrapper, |
|||
TelemetryService, |
|||
TelemetrySubscriber, |
|||
TimeseriesSubscriptionCmd |
|||
} from '@app/shared/models/telemetry/telemetry.models'; |
|||
import { select, Store } from '@ngrx/store'; |
|||
import { AppState } from '@core/core.state'; |
|||
import { AuthService } from '@core/auth/auth.service'; |
|||
import { selectIsAuthenticated } from '@core/auth/auth.selectors'; |
|||
import { WINDOW } from '@core/services/window.service'; |
|||
import { webSocket, WebSocketSubject } from 'rxjs/webSocket'; |
|||
import { ActionNotificationShow } from '@core/notification/notification.actions'; |
|||
import Timeout = NodeJS.Timeout; |
|||
|
|||
const RECONNECT_INTERVAL = 2000; |
|||
const WS_IDLE_TIMEOUT = 90000; |
|||
const MAX_PUBLISH_COMMANDS = 10; |
|||
|
|||
@Injectable({ |
|||
providedIn: 'root' |
|||
}) |
|||
export class TelemetryWebsocketService implements TelemetryService { |
|||
|
|||
isActive = false; |
|||
isOpening = false; |
|||
isOpened = false; |
|||
isReconnect = false; |
|||
|
|||
socketCloseTimer: Timeout; |
|||
reconnectTimer: Timeout; |
|||
|
|||
lastCmdId = 0; |
|||
subscribersCount = 0; |
|||
subscribersMap = new Map<number, TelemetrySubscriber>(); |
|||
|
|||
reconnectSubscribers = new Set<TelemetrySubscriber>(); |
|||
|
|||
cmdsWrapper = new TelemetryPluginCmdsWrapper(); |
|||
telemetryUri: string; |
|||
|
|||
dataStream: WebSocketSubject<TelemetryPluginCmdsWrapper | SubscriptionUpdateMsg>; |
|||
|
|||
constructor(private store: Store<AppState>, |
|||
private authService: AuthService, |
|||
@Inject(WINDOW) private window: Window) { |
|||
this.store.pipe(select(selectIsAuthenticated)).subscribe( |
|||
(authenticated: boolean) => { |
|||
if (!authenticated) { |
|||
this.reset(true); |
|||
} |
|||
} |
|||
); |
|||
|
|||
let port = this.window.location.port; |
|||
if (this.window.location.protocol === 'https:') { |
|||
if (!port) { |
|||
port = '443'; |
|||
} |
|||
this.telemetryUri = 'wss:'; |
|||
} else { |
|||
if (!port) { |
|||
port = '80'; |
|||
} |
|||
this.telemetryUri = 'ws:'; |
|||
} |
|||
this.telemetryUri += `//${this.window.location.hostname}:${port}/api/ws/plugins/telemetry`; |
|||
} |
|||
|
|||
public subscribe(subscriber: TelemetrySubscriber) { |
|||
this.isActive = true; |
|||
subscriber.subscriptionCommands.forEach( |
|||
(subscriptionCommand) => { |
|||
const cmdId = this.nextCmdId(); |
|||
this.subscribersMap.set(cmdId, subscriber); |
|||
subscriptionCommand.cmdId = cmdId; |
|||
if (subscriptionCommand instanceof SubscriptionCmd) { |
|||
if (subscriptionCommand.getType() === TelemetryFeature.TIMESERIES) { |
|||
this.cmdsWrapper.tsSubCmds.push(subscriptionCommand as TimeseriesSubscriptionCmd); |
|||
} else { |
|||
this.cmdsWrapper.attrSubCmds.push(subscriptionCommand as AttributesSubscriptionCmd); |
|||
} |
|||
} else if (subscriptionCommand instanceof GetHistoryCmd) { |
|||
this.cmdsWrapper.historyCmds.push(subscriptionCommand); |
|||
} |
|||
} |
|||
); |
|||
this.subscribersCount++; |
|||
this.publishCommands(); |
|||
} |
|||
|
|||
public unsubscribe(subscriber: TelemetrySubscriber) { |
|||
if (this.isActive) { |
|||
subscriber.subscriptionCommands.forEach( |
|||
(subscriptionCommand) => { |
|||
if (subscriptionCommand instanceof SubscriptionCmd) { |
|||
subscriptionCommand.unsubscribe = true; |
|||
if (subscriptionCommand.getType() === TelemetryFeature.TIMESERIES) { |
|||
this.cmdsWrapper.tsSubCmds.push(subscriptionCommand as TimeseriesSubscriptionCmd); |
|||
} else { |
|||
this.cmdsWrapper.attrSubCmds.push(subscriptionCommand as AttributesSubscriptionCmd); |
|||
} |
|||
} |
|||
const cmdId = subscriptionCommand.cmdId; |
|||
if (cmdId) { |
|||
this.subscribersMap.delete(cmdId); |
|||
} |
|||
} |
|||
); |
|||
this.reconnectSubscribers.delete(subscriber); |
|||
this.subscribersCount--; |
|||
this.publishCommands(); |
|||
} |
|||
} |
|||
|
|||
private nextCmdId(): number { |
|||
this.lastCmdId++; |
|||
return this.lastCmdId; |
|||
} |
|||
|
|||
private publishCommands() { |
|||
while (this.isOpened && this.cmdsWrapper.hasCommands()) { |
|||
this.dataStream.next(this.cmdsWrapper.preparePublishCommands(MAX_PUBLISH_COMMANDS)); |
|||
this.checkToClose(); |
|||
} |
|||
this.tryOpenSocket(); |
|||
} |
|||
|
|||
private checkToClose() { |
|||
if (this.subscribersCount === 0 && this.isOpened) { |
|||
if (!this.socketCloseTimer) { |
|||
this.socketCloseTimer = setTimeout( |
|||
() => this.closeSocket(), WS_IDLE_TIMEOUT); |
|||
} |
|||
} |
|||
} |
|||
|
|||
private reset(close: boolean) { |
|||
if (this.socketCloseTimer) { |
|||
clearTimeout(this.socketCloseTimer); |
|||
this.socketCloseTimer = null; |
|||
} |
|||
this.lastCmdId = 0; |
|||
this.subscribersMap.clear(); |
|||
this.subscribersCount = 0; |
|||
this.cmdsWrapper.clear(); |
|||
if (close) { |
|||
this.closeSocket(); |
|||
} |
|||
} |
|||
|
|||
private closeSocket() { |
|||
this.isActive = false; |
|||
if (this.isOpened) { |
|||
this.dataStream.unsubscribe(); |
|||
} |
|||
} |
|||
|
|||
private tryOpenSocket() { |
|||
if (this.isActive) { |
|||
if (!this.isOpened && !this.isOpening) { |
|||
this.isOpening = true; |
|||
if (AuthService.isJwtTokenValid()) { |
|||
this.openSocket(AuthService.getJwtToken()); |
|||
} else { |
|||
this.authService.refreshJwtToken().subscribe(() => { |
|||
this.openSocket(AuthService.getJwtToken()); |
|||
}, |
|||
() => { |
|||
this.isOpening = false; |
|||
this.authService.logout(true); |
|||
} |
|||
); |
|||
} |
|||
} |
|||
if (this.socketCloseTimer) { |
|||
clearTimeout(this.socketCloseTimer); |
|||
this.socketCloseTimer = null; |
|||
} |
|||
} |
|||
} |
|||
|
|||
private openSocket(token: string) { |
|||
const uri = `${this.telemetryUri}?token=${token}`; |
|||
this.dataStream = webSocket( |
|||
{ |
|||
url: uri, |
|||
openObserver: { |
|||
next: (e: Event) => { |
|||
this.onOpen(); |
|||
} |
|||
}, |
|||
closeObserver: { |
|||
next: (e: CloseEvent) => { |
|||
this.onClose(e); |
|||
} |
|||
} |
|||
} |
|||
); |
|||
|
|||
this.dataStream.subscribe((message) => { |
|||
this.onMessage(message as SubscriptionUpdateMsg); |
|||
}, |
|||
(error) => { |
|||
this.onError(error); |
|||
}); |
|||
} |
|||
|
|||
private onOpen() { |
|||
this.isOpening = false; |
|||
this.isOpened = true; |
|||
if (this.reconnectTimer) { |
|||
clearTimeout(this.reconnectTimer); |
|||
this.reconnectTimer = null; |
|||
} |
|||
if (this.isReconnect) { |
|||
this.isReconnect = false; |
|||
this.reconnectSubscribers.forEach( |
|||
(reconnectSubscriber) => { |
|||
reconnectSubscriber.onReconnected(); |
|||
this.subscribe(reconnectSubscriber); |
|||
} |
|||
); |
|||
this.reconnectSubscribers.clear(); |
|||
} else { |
|||
this.publishCommands(); |
|||
} |
|||
} |
|||
|
|||
private onMessage(message: SubscriptionUpdateMsg) { |
|||
if (message.errorCode) { |
|||
this.showWsError(message.errorCode, message.errorMsg); |
|||
} else if (message.subscriptionId) { |
|||
const subscriber = this.subscribersMap.get(message.subscriptionId); |
|||
if (subscriber) { |
|||
subscriber.onData(new SubscriptionUpdate(message)); |
|||
} |
|||
} |
|||
this.checkToClose(); |
|||
} |
|||
|
|||
private onError(errorEvent) { |
|||
if (errorEvent) { |
|||
console.warn('WebSocket error event', errorEvent); |
|||
} |
|||
this.isOpening = false; |
|||
} |
|||
|
|||
private onClose(closeEvent: CloseEvent) { |
|||
if (closeEvent && closeEvent.code > 1000 && closeEvent.code !== 1006) { |
|||
this.showWsError(closeEvent.code, closeEvent.reason); |
|||
} |
|||
this.isOpening = false; |
|||
this.isOpened = false; |
|||
if (this.isActive) { |
|||
if (!this.isReconnect) { |
|||
this.reconnectSubscribers.clear(); |
|||
this.subscribersMap.forEach( |
|||
(subscriber) => { |
|||
this.reconnectSubscribers.add(subscriber); |
|||
} |
|||
); |
|||
this.reset(false); |
|||
this.isReconnect = true; |
|||
} |
|||
if (this.reconnectTimer) { |
|||
clearTimeout(this.reconnectTimer); |
|||
} |
|||
this.reconnectTimer = setTimeout(() => this.tryOpenSocket(), RECONNECT_INTERVAL); |
|||
} |
|||
} |
|||
|
|||
private showWsError(errorCode: number, errorMsg: string) { |
|||
let message = 'WebSocket Error: '; |
|||
if (errorMsg) { |
|||
message += errorMsg; |
|||
} else { |
|||
message += `error code - ${errorCode}.`; |
|||
} |
|||
this.store.dispatch(new ActionNotificationShow( |
|||
{ |
|||
message, type: 'error' |
|||
})); |
|||
} |
|||
|
|||
} |
|||
Loading…
Reference in new issue