diff --git a/server/src/maintenance/maintenance-websocket.repository.ts b/server/src/maintenance/maintenance-websocket.repository.ts index 8d52e516c5..cf04c0ad12 100644 --- a/server/src/maintenance/maintenance-websocket.repository.ts +++ b/server/src/maintenance/maintenance-websocket.repository.ts @@ -37,7 +37,10 @@ export class MaintenanceWebsocketRepository implements OnGatewayConnection, OnGa afterInit(websocketServer: Server) { this.logger.log('Initialized websocket server'); - websocketServer.on('AppRestart', (_, ack?: (ok: 'ok') => void) => { + + websocketServer.on('AppRestart', (event: ArgsOf<'AppRestart'>, ack?: (ok: 'ok') => void) => { + this.logger.log(`Restarting due to event... ${JSON.stringify(event)}`); + ack?.('ok'); this.appRepository.exitApp(); }); diff --git a/server/src/services/maintenance.service.ts b/server/src/services/maintenance.service.ts index 312f86cd44..0f5fa06957 100644 --- a/server/src/services/maintenance.service.ts +++ b/server/src/services/maintenance.service.ts @@ -2,6 +2,7 @@ import { Injectable } from '@nestjs/common'; import { OnEvent } from 'src/decorators'; import { MaintenanceAuthDto } from 'src/dtos/maintenance.dto'; import { SystemMetadataKey } from 'src/enum'; +import { ArgOf } from 'src/repositories/event.repository'; import { BaseService } from 'src/services/base.service'; import { MaintenanceModeState } from 'src/types'; import { createMaintenanceLoginUrl, generateMaintenanceSecret, signMaintenanceJwt } from 'src/utils/maintenance'; @@ -31,7 +32,9 @@ export class MaintenanceService extends BaseService { } @OnEvent({ name: 'AppRestart', server: true }) - onRestart(_: undefined, ack?: (ok: 'ok') => void): void { + onRestart(event: ArgOf<'AppRestart'>, ack?: (ok: 'ok') => void): void { + this.logger.log(`Restarting due to event... ${JSON.stringify(event)}`); + ack?.('ok'); this.appRepository.exitApp(); }