-
Notifications
You must be signed in to change notification settings - Fork 11
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat(repeater): replace rabbitmq with repeater server
- Loading branch information
Showing
12 changed files
with
429 additions
and
337 deletions.
There are no files selected for viewing
Large diffs are not rendered by default.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,147 @@ | ||
| import { logger } from '../Utils'; | ||
| import { | ||
| Repeater, | ||
| RepeaterEventHandler, | ||
| RepeaterDeployedEvent, | ||
| RepeaterEvents | ||
| } from './Repeater'; | ||
| import { inject, injectable } from 'tsyringe'; | ||
| import io, { Socket } from 'socket.io-client'; | ||
| import parser from 'socket.io-msgpack-parser'; | ||
| import { hostname } from 'os'; | ||
| import { once } from 'events'; | ||
|
|
||
| export interface DefaultRepeaterOptions { | ||
| readonly uri: string; | ||
| readonly token: string; | ||
| } | ||
|
|
||
| export const DefaultRepeaterOptions: unique symbol = Symbol( | ||
| 'DefaultRepeaterOptions' | ||
| ); | ||
|
|
||
| @injectable() | ||
| export class DefaultRepeater implements Repeater { | ||
| private latestReconnectionError?: Error; | ||
| private readonly DEFAULT_RECONNECT_TIMES = 3; | ||
| private socket?: Socket; | ||
|
|
||
| constructor( | ||
| @inject(DefaultRepeaterOptions) | ||
| private readonly options: DefaultRepeaterOptions | ||
| ) {} | ||
|
|
||
| public disconnect() { | ||
| if (!this.socket) { | ||
| throw new Error('Not connected'); | ||
| } | ||
|
|
||
| this.socket?.disconnect(); | ||
| this.socket?.removeAllListeners(); | ||
| this.socket = undefined; | ||
| } | ||
|
|
||
| public async deploy( | ||
| repeaterId?: string | ||
| ): Promise<RepeaterDeployedEvent['response']> { | ||
| if (!this.socket) { | ||
| throw new Error('Repeater is not connected yet'); | ||
| } | ||
|
|
||
| this.socket.emit('deploy', { | ||
| repeaterId | ||
| }); | ||
|
|
||
| const [result]: RepeaterDeployedEvent['response'][] = await once( | ||
| this.socket, | ||
| 'deployed' | ||
| ); | ||
|
|
||
| return result; | ||
| } | ||
|
|
||
| public connect() { | ||
| this.socket = io(this.options.uri, { | ||
| parser, | ||
| path: '/api/ws/v1', | ||
| transports: ['websocket'], | ||
| reconnectionAttempts: this.DEFAULT_RECONNECT_TIMES, | ||
| auth: { | ||
| token: this.options.token, | ||
| domain: hostname() | ||
| } | ||
| }); | ||
|
|
||
| this.socket.on('connect_error', (error: Error) => { | ||
| logger.debug(`Unexpected error: %s`, error); | ||
| }); | ||
|
|
||
| logger.debug('Event bus connected to %s', this.options.uri); | ||
| } | ||
|
|
||
| public on<E extends keyof RepeaterEvents, H extends RepeaterEventHandler<E>>( | ||
| event: E, | ||
| handler: H | ||
| ): void { | ||
| const eventName: string = event; | ||
|
|
||
| if (event === 'reconnection_failed') { | ||
| // TODO: Figure out why type is not narrowing | ||
| this.onReconnectionFailure(handler as any); | ||
| } else { | ||
| this.socket.on(eventName, (payload, callback) => { | ||
| this.processEventHandler(event, payload, handler, callback); | ||
| }); | ||
| } | ||
| } | ||
|
|
||
| private onReconnectionFailure< | ||
| H extends RepeaterEventHandler<'reconnection_failed'> | ||
| >(handler: H) { | ||
| this.socket.io.on('reconnect', () => { | ||
| this.latestReconnectionError = undefined; | ||
| }); | ||
|
|
||
| this.socket.io.on('reconnect_error', (error) => { | ||
| this.latestReconnectionError = error; | ||
| }); | ||
|
|
||
| this.socket.io.on('reconnect_failed', () => { | ||
| this.processEventHandler( | ||
| 'reconnection_failed', | ||
| { | ||
| error: this.latestReconnectionError | ||
| }, | ||
| handler | ||
| ); | ||
| }); | ||
| } | ||
|
|
||
| private processEventHandler< | ||
| E extends keyof RepeaterEvents, | ||
| H extends RepeaterEventHandler<E> | ||
| >( | ||
| event: E, | ||
| payload: RepeaterEvents[E]['request'], | ||
| handler: H, | ||
| callback?: unknown | ||
| ) { | ||
| Promise.resolve(handler(payload)) | ||
| .then((response) => { | ||
| if (typeof callback !== 'function' || typeof response === 'undefined') { | ||
| return; | ||
| } | ||
|
|
||
| callback(response); | ||
| }) | ||
| .catch((error) => { | ||
| logger.debug( | ||
| 'Error processing event "%s" with the following payload: %s. Details: %s', | ||
| event, | ||
| payload, | ||
| error | ||
| ); | ||
| logger.error('Error: %s', error.message); | ||
| }); | ||
| } | ||
| } |
Oops, something went wrong.