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

fix(server): use ClsInterceptor on websocket (#9859)

Browse Source 
https://papooch.github.io/nestjs-cls/considerations/compatibility#websockets
This commit is contained in:
fengmk2
2025-01-23 06:54:23 +00:00
parent d52d03e1cd
commit 8021b89944
2 changed files with 17 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
packages/backend/server/src/core/sync/gateway.ts
View File

@@ -1,4 +1,4 @@
import { applyDecorators, Logger } from '@nestjs/common';
import { applyDecorators, Logger, UseInterceptors } from '@nestjs/common';
import {
ConnectedSocket,
MessageBody,
@@ -7,6 +7,7 @@ import {
SubscribeMessage as RawSubscribeMessage,
WebSocketGateway,
} from '@nestjs/websockets';
import { ClsInterceptor } from 'nestjs-cls';
import { Socket } from 'socket.io';
import {
@@ -131,6 +132,7 @@ interface UpdateAwarenessMessage {
}
@WebSocketGateway()
@UseInterceptors(ClsInterceptor)
export class SpaceSyncGateway
implements OnGatewayConnection, OnGatewayDisconnect
{
@@ -147,11 +149,13 @@ export class SpaceSyncGateway
handleConnection() {
this.connectionCount++;
this.logger.log(`New connection, total: ${this.connectionCount}`);
metrics.socketio.gauge('realtime_connections').record(this.connectionCount);
}
handleDisconnect() {
this.connectionCount--;
this.logger.log(`Connection disconnected, total: ${this.connectionCount}`);
metrics.socketio.gauge('realtime_connections').record(this.connectionCount);
}