Jayden/AFFiNE-Mirror
1
0
Fork 0
mirror of https://github.com/toeverything/AFFiNE.git synced 2026-02-13 04:48:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(server): graceful shutdown for AI streams (#10025)

Browse Source 
https://github.com/user-attachments/assets/8dd3c4f5-4059-4f03-9f51-68078d7ab5c4
This commit is contained in:
Brooooooklyn
2025-02-07 11:25:02 +00:00
parent 0df94b8e35
commit 4e00ddd5f1
3 changed files with 42 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
packages/backend/server/src/base/prisma/service.ts
View File

@@ -1,11 +1,11 @@
import type { OnModuleDestroy, OnModuleInit } from '@nestjs/common';
import type { OnApplicationShutdown, OnModuleInit } from '@nestjs/common';
import { Injectable } from '@nestjs/common';
import { Prisma, PrismaClient } from '@prisma/client';
@Injectable()
export class PrismaService
extends PrismaClient
implements OnModuleInit, OnModuleDestroy
implements OnModuleInit, OnApplicationShutdown
{
static INSTANCE: PrismaService | null = null;
@@ -18,7 +18,7 @@ export class PrismaService
await this.$connect();
}
async onModuleDestroy(): Promise<void> {
async onApplicationShutdown(): Promise<void> {
if (!AFFiNE.node.test) {
await this.$disconnect();
PrismaService.INSTANCE = null;