Vladyslav_Prykhodko
4 years ago
12 changed files with 456 additions and 592 deletions
@ -0,0 +1,245 @@ |
|||||
|
import { CmdWrapper, WsService, WsSubscriber } from '@shared/models/websocket/websocket.models'; |
||||
|
import { select, Store } from '@ngrx/store'; |
||||
|
import { AppState } from '@core/core.state'; |
||||
|
import { AuthService } from '@core/auth/auth.service'; |
||||
|
import { NgZone } from '@angular/core'; |
||||
|
import { selectIsAuthenticated } from '@core/auth/auth.selectors'; |
||||
|
import { webSocket, WebSocketSubject } from 'rxjs/webSocket'; |
||||
|
import { WebsocketNotificationMsg } from '@shared/models/websocket/notification-ws.models'; |
||||
|
import { CmdUpdateMsg } from '@shared/models/telemetry/telemetry.models'; |
||||
|
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; |
||||
|
|
||||
|
export abstract class WebsocketService<T extends WsSubscriber> implements WsService<T> { |
||||
|
|
||||
|
isActive = false; |
||||
|
isOpening = false; |
||||
|
isOpened = false; |
||||
|
isReconnect = false; |
||||
|
|
||||
|
socketCloseTimer: Timeout; |
||||
|
reconnectTimer: Timeout; |
||||
|
|
||||
|
lastCmdId = 0; |
||||
|
subscribersCount = 0; |
||||
|
subscribersMap = new Map<number, T>(); |
||||
|
|
||||
|
reconnectSubscribers = new Set<T>(); |
||||
|
|
||||
|
notificationUri: string; |
||||
|
|
||||
|
dataStream: WebSocketSubject<CmdWrapper | CmdUpdateMsg>; |
||||
|
|
||||
|
errorName = 'WebSocket Error'; |
||||
|
|
||||
|
protected constructor(protected store: Store<AppState>, |
||||
|
protected authService: AuthService, |
||||
|
protected ngZone: NgZone, |
||||
|
protected apiEndpoint: string, |
||||
|
protected cmdWrapper: CmdWrapper, |
||||
|
protected window: Window) { |
||||
|
this.store.pipe(select(selectIsAuthenticated)).subscribe( |
||||
|
() => { |
||||
|
this.reset(true); |
||||
|
} |
||||
|
); |
||||
|
|
||||
|
let port = this.window.location.port; |
||||
|
if (this.window.location.protocol === 'https:') { |
||||
|
if (!port) { |
||||
|
port = '443'; |
||||
|
} |
||||
|
this.notificationUri = 'wss:'; |
||||
|
} else { |
||||
|
if (!port) { |
||||
|
port = '80'; |
||||
|
} |
||||
|
this.notificationUri = 'ws:'; |
||||
|
} |
||||
|
this.notificationUri += `//${this.window.location.hostname}:${port}/${apiEndpoint}`; |
||||
|
} |
||||
|
|
||||
|
abstract subscribe(subscriber: T); |
||||
|
|
||||
|
abstract update(subscriber: T); |
||||
|
|
||||
|
abstract unsubscribe(subscriber: T); |
||||
|
|
||||
|
abstract processOnMessage(message: any); |
||||
|
|
||||
|
protected nextCmdId(): number { |
||||
|
this.lastCmdId++; |
||||
|
return this.lastCmdId; |
||||
|
} |
||||
|
|
||||
|
protected publishCommands() { |
||||
|
while (this.isOpened && this.cmdWrapper.hasCommands()) { |
||||
|
this.dataStream.next(this.cmdWrapper.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.cmdWrapper.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({ |
||||
|
next: () => { |
||||
|
this.openSocket(AuthService.getJwtToken()); |
||||
|
}, |
||||
|
error: () => { |
||||
|
this.isOpening = false; |
||||
|
this.authService.logout(true, true); |
||||
|
} |
||||
|
}); |
||||
|
} |
||||
|
} |
||||
|
if (this.socketCloseTimer) { |
||||
|
clearTimeout(this.socketCloseTimer); |
||||
|
this.socketCloseTimer = null; |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private openSocket(token: string) { |
||||
|
const uri = `${this.notificationUri}?token=${token}`; |
||||
|
this.dataStream = webSocket( |
||||
|
{ |
||||
|
url: uri, |
||||
|
openObserver: { |
||||
|
next: () => { |
||||
|
this.onOpen(); |
||||
|
} |
||||
|
}, |
||||
|
closeObserver: { |
||||
|
next: (e: CloseEvent) => { |
||||
|
this.onClose(e); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
); |
||||
|
|
||||
|
this.dataStream.subscribe({ |
||||
|
next: (message) => { |
||||
|
this.ngZone.runOutsideAngular(() => { |
||||
|
this.onMessage(message as WebsocketNotificationMsg); |
||||
|
}); |
||||
|
}, |
||||
|
error: (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: WebsocketNotificationMsg) { |
||||
|
if (message.errorCode) { |
||||
|
this.showWsError(message.errorCode, message.errorMsg); |
||||
|
} else { |
||||
|
this.processOnMessage(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 > 1001 && closeEvent.code !== 1006 |
||||
|
&& closeEvent.code !== 1011 && closeEvent.code !== 1012 && closeEvent.code !== 4500) { |
||||
|
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 = errorMsg; |
||||
|
if (!message) { |
||||
|
message += `${this.errorName}: error code - ${errorCode}.`; |
||||
|
} |
||||
|
this.store.dispatch(new ActionNotificationShow( |
||||
|
{ |
||||
|
message, type: 'error' |
||||
|
})); |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,51 @@ |
|||||
|
import { NgZone } from '@angular/core'; |
||||
|
import { WebsocketCmd } from '@shared/models/telemetry/telemetry.models'; |
||||
|
import { Subject } from 'rxjs'; |
||||
|
|
||||
|
export interface WsService<T extends WsSubscriber> { |
||||
|
subscribe(subscriber: T); |
||||
|
update(subscriber: T); |
||||
|
unsubscribe(subscriber: T); |
||||
|
} |
||||
|
|
||||
|
export abstract class CmdWrapper { |
||||
|
abstract hasCommands(): boolean; |
||||
|
abstract clear(): void; |
||||
|
abstract preparePublishCommands(maxCommands: number): CmdWrapper; |
||||
|
|
||||
|
[key: string]: WebsocketCmd | any; |
||||
|
} |
||||
|
|
||||
|
export abstract class WsSubscriber { |
||||
|
|
||||
|
protected reconnectSubject = new Subject<void>(); |
||||
|
|
||||
|
subscriptionCommands: Array<WebsocketCmd>; |
||||
|
|
||||
|
reconnect$ = this.reconnectSubject.asObservable(); |
||||
|
|
||||
|
protected constructor(protected wsService: WsService<WsSubscriber>, protected zone?: NgZone) { |
||||
|
this.subscriptionCommands = []; |
||||
|
} |
||||
|
|
||||
|
public subscribe() { |
||||
|
this.wsService.subscribe(this); |
||||
|
} |
||||
|
|
||||
|
public update() { |
||||
|
this.wsService.update(this); |
||||
|
} |
||||
|
|
||||
|
public unsubscribe() { |
||||
|
this.wsService.unsubscribe(this); |
||||
|
this.complete(); |
||||
|
} |
||||
|
|
||||
|
public complete() { |
||||
|
this.reconnectSubject.complete(); |
||||
|
} |
||||
|
|
||||
|
public onReconnected() { |
||||
|
this.reconnectSubject.next(); |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue