fix(judge): socket.io message body limit exceeded caused connection c…

…lose
lyrio-dev · Mar 30, 2022 · a7f4c40 · a7f4c40
1 parent 2cf6a10
commit a7f4c40
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/src/judge/judge.gateway.ts b/src/judge/judge.gateway.ts
@@ -44,7 +44,13 @@ const REDIS_LOCK_JUDGE_CLIENT_CONNECT_DISCONNECT = "judge-client-connect-disconn
 
 const REDIS_CHANNEL_CANCEL_TASK = "cancel-task";
 
-@WebSocketGateway({ namespace: "judge", path: "/api/socket", transports: ["websocket"], parser: SocketIOParser })
+@WebSocketGateway({
+ maxHttpBufferSize: 1e9,
+ namespace: "judge",
+ path: "/api/socket",
+ transports: ["websocket"],
+ parser: SocketIOParser
+})
 export class JudgeGateway implements OnGatewayConnection, OnGatewayDisconnect {
  @WebSocketServer()
  private server: Server;

diff --git a/src/submission/submission-progress.gateway.ts b/src/submission/submission-progress.gateway.ts
@@ -41,6 +41,7 @@ interface SubmissionProgressMessage {
 // TODO: This should be refactored if we add hack, custom judge, etc
 // Maybe refactor to a general "task progress"
 @WebSocketGateway({
+ maxHttpBufferSize: 1e9,
  namespace: "submission-progress",
  path: "/api/socket",
  transports: ["websocket"],