diff --git a/server/libs/domain/src/domain.module.ts b/server/libs/domain/src/domain.module.ts index 712b5eab84..c79490a460 100644 --- a/server/libs/domain/src/domain.module.ts +++ b/server/libs/domain/src/domain.module.ts @@ -1,4 +1,4 @@ -import { DynamicModule, Global, Module, ModuleMetadata, Provider } from '@nestjs/common'; +import { DynamicModule, Global, Module, ModuleMetadata, OnApplicationShutdown, Provider } from '@nestjs/common'; import { AlbumService } from './album'; import { APIKeyService } from './api-key'; import { AssetService } from './asset'; @@ -44,7 +44,9 @@ const providers: Provider[] = [ @Global() @Module({}) -export class DomainModule { +export class DomainModule implements OnApplicationShutdown { + constructor(private searchService: SearchService) {} + static register(options: Pick): DynamicModule { return { module: DomainModule, @@ -53,4 +55,8 @@ export class DomainModule { exports: [...providers], }; } + + onApplicationShutdown() { + this.searchService.teardown(); + } }