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

feat(server): add request id on cluster event (#9998)

Browse Source
This commit is contained in:
fengmk2
2025-02-07 02:06:53 +00:00
parent 891d9df0b1
commit 1e83a056fc
3 changed files with 34 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
packages/backend/server/src/base/event/eventbus.ts
View File

@@ -1,3 +1,5 @@
import { randomUUID } from 'node:crypto';
import {
applyDecorators,
Injectable,
@@ -15,6 +17,7 @@ import {
WebSocketGateway,
WebSocketServer,
} from '@nestjs/websockets';
import { CLS_ID, ClsService } from 'nestjs-cls';
import type { Server, Socket } from 'socket.io';
import { CallMetric } from '../metrics';
@@ -69,7 +72,8 @@ export class EventBus implements OnGatewayConnection, OnApplicationBootstrap {
constructor(
private readonly emitter: EventEmitter2,
private readonly watcher: EventEmitterReadinessWatcher
private readonly watcher: EventEmitterReadinessWatcher,
private readonly cls: ClsService
) {}
handleConnection(client: Socket) {
@@ -88,9 +92,13 @@ export class EventBus implements OnGatewayConnection, OnApplicationBootstrap {
events.forEach(event => {
// Proxy all events received from server(trigger by `server.serverSideEmit`)
// to internal event system
this.server?.on(event, payload => {
this.logger.log(`Server Event: ${event} (Received)`);
this.emit(event, payload);
this.server?.on(event, (payload, requestId?: string) => {
this.cls.run(() => {
requestId = requestId ?? `server_event-${randomUUID()}`;
this.cls.set(CLS_ID, requestId);
this.logger.log(`Server Event: ${event} (Received)`);
this.emit(event, payload);
});
});
});
})
@@ -120,7 +128,7 @@ export class EventBus implements OnGatewayConnection, OnApplicationBootstrap {
*/
broadcast<T extends EventName>(event: T, payload: Events[T]) {
this.logger.log(`Server Event: ${event} (Send)`);
this.server?.serverSideEmit(event, payload);
this.server?.serverSideEmit(event, payload, this.cls.getId());
}
on<T extends EventName>(