feat: auto remove room if no peer

woody146 · Jun 21, 2023 · a07e34b · a07e34b
1 parent 14801c1
commit a07e34b
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/services/user.ts b/src/services/user.ts
@@ -3,6 +3,7 @@ import { MediaPeer } from '../entities/media.peer.js';
 import { MediaWorker } from '../entities/media.worker.js';
 import { fetchApi } from '../utils/api.js';
 import { BaseService } from './base.js';
+import { RoomService } from './room.js';
 
 export class UserService extends BaseService {
   /**
@@ -32,8 +33,21 @@ export class UserService extends BaseService {
         await this.dataSource
           .getRepository(MediaWorker)
           .decrement({ id: peer.workerId }, 'peerCount', 1);
+
+        this.removeEmptyRoom(peer);
       })
     );
     return {};
   }
+
+  async removeEmptyRoom(data: { roomId: string }) {
+    const exist = await this.dataSource.getRepository(MediaPeer).findOne({
+      select: { id: true },
+      where: { roomId: data.roomId },
+    });
+    // if no one in room
+    if (!exist) {
+      this.createService(RoomService).close(data);
+    }
+  }
 }