[예제] 1:N P2P 예제 코드

backend/package.json

@@ -23,8 +23,10 @@
     "@nestjs/common": "^10.0.0",
     "@nestjs/core": "^10.0.0",
     "@nestjs/platform-express": "^10.0.0",
+    "@nestjs/platform-socket.io": "^10.4.6",
     "reflect-metadata": "^0.2.0",
-    "rxjs": "^7.8.1"
+    "rxjs": "^7.8.1",
+    "socket.io": "^4.8.1"
   },
   "devDependencies": {
     "@nestjs/cli": "^10.0.0",

backend/src/app.module.ts

@@ -1,10 +1,7 @@
 import { Module } from '@nestjs/common';
-import { AppController } from './app.controller';
-import { AppService } from './app.service';
+import { SocketModule } from './socket.module';
 
 @Module({
-  imports: [],
-  controllers: [AppController],
-  providers: [AppService],
+  imports: [SocketModule],
 })
-export class AppModule {}
+export class AppModule { }

backend/src/socket.gateway.ts

@@ -0,0 +1,94 @@
+import {
+  WebSocketGateway,
+  WebSocketServer,
+  OnGatewayConnection,
+  OnGatewayDisconnect,
+  SubscribeMessage,
+  MessageBody,
+} from '@nestjs/websockets';
+import { Server } from 'socket.io';
+
+interface User {
+  id: string;
+  email: string;
+}
+
+@WebSocketGateway({
+  cors: {
+    origin: '*', // CORS 설정
+  },
+})
+
+export class SocketGateway implements OnGatewayConnection, OnGatewayDisconnect {
+  @WebSocketServer()
+  server: Server;
+
+  private users: { [key: string]: User[] } = {};
+  private socketToRoom: { [key: string]: string } = {};
+  private maximum = 4; // 최대 사용자 수
+
+  handleConnection(socket: any) {
+    console.log(`Client connected: ${socket.id}`);
+  }
+
+  handleDisconnect(socket: any) {
+    console.log(`Client disconnected: ${socket.id}`);
+    const roomID = this.socketToRoom[socket.id];
+    if (roomID) {
+      const room = this.users[roomID];
+      if (room) {
+        this.users[roomID] = room.filter(user => user.id !== socket.id);
+        if (this.users[roomID].length === 0) {
+          delete this.users[roomID];
+        } else {
+          this.server.to(roomID).emit('user_exit', { id: socket.id });
+        }
+      }
+    }
+  }
+
+  @SubscribeMessage('join_room')
+  handleJoinRoom(socket: any, data: { room: string; email: string }) {
+    if (this.users[data.room]) {
+      if (this.users[data.room].length === this.maximum) {
+        socket.emit('room_full');
+        return;
+      }
+      this.users[data.room].push({ id: socket.id, email: data.email });
+    } else {
+      this.users[data.room] = [{ id: socket.id, email: data.email }];
+    }
+
+    this.socketToRoom[socket.id] = data.room;
+    socket.join(data.room);
+    console.log(`[${data.room}]: ${socket.id} enter`);
+
+    const usersInThisRoom = this.users[data.room].filter(user => user.id !== socket.id);
+    socket.emit('all_users', usersInThisRoom);
+  }
+
+  @SubscribeMessage('offer')
+  handleOffer(@MessageBody() data: { offerReceiveID: string; sdp: any; offerSendID: string; offerSendEmail: string }) {
+    this.server.to(data.offerReceiveID).emit('getOffer', {
+      sdp: data.sdp,
+      offerSendID: data.offerSendID,
+      offerSendEmail: data.offerSendEmail,
+    });
+  }
+
+  @SubscribeMessage('answer')
+  handleAnswer(@MessageBody() data: { answerReceiveID: string; sdp: any; answerSendID: string }) {
+    this.server.to(data.answerReceiveID).emit('getAnswer', {
+      sdp: data.sdp,
+      answerSendID: data.answerSendID,
+    });
+  }
+
+  @SubscribeMessage('candidate')
+  handleCandidate(@MessageBody() data: { candidateReceiveID: string; candidate: any; candidateSendID: string }) {
+    this.server.to(data.candidateReceiveID).emit('getCandidate', {
+      candidate: data.candidate,
+      candidateSendID: data.candidateSendID,
+    });
+  }
+}

backend/src/socket.module.ts

@@ -0,0 +1,7 @@
+import { Module } from '@nestjs/common';
+import { SocketGateway } from './socket.gateway';
+
+@Module({
+  providers: [SocketGateway],
+})
+export class SocketModule { }

backend/tsconfig.build.json

@@ -1,4 +1,9 @@
 {
   "extends": "./tsconfig.json",
-  "exclude": ["node_modules", "test", "dist", "**/*spec.ts"]
-}
+  "exclude": [
+    "node_modules",
+    "test",
+    "dist",
+    "**/*spec.ts"
+  ]
+}

backend/tsconfig.json

@@ -18,4 +18,4 @@
     "forceConsistentCasingInFileNames": false,
     "noFallthroughCasesInSwitch": false
   }
-}
+}

frontend/package.json

@@ -18,6 +18,7 @@
     "@eslint/js": "^9.13.0",
     "@types/react": "^18.3.12",
     "@types/react-dom": "^18.3.1",
+    "@types/socket.io-client": "^3.0.0",
     "@vitejs/plugin-react": "^4.3.3",
     "autoprefixer": "^10.4.20",
     "eslint": "^9.13.0",
@@ -30,4 +31,4 @@
     "typescript-eslint": "^8.11.0",
     "vite": "^5.4.10"
   }
-}
+}