feat: add PeerService.resume()

woody146 · Jun 13, 2023 · e3f6175 · e3f6175
1 parent 430c6a5
commit e3f6175
Show file tree

Hide file tree

Showing 5 changed files with 51 additions and 3 deletions.
diff --git a/examples/rooms/src/components/Consumer.tsx b/examples/rooms/src/components/Consumer.tsx
@@ -66,6 +66,10 @@ export function Consumer({
           case 'connected':
             if (ref.current) {
               ref.current.srcObject = await stream;
+              fetchApi({
+                path: `/api/peers/${data.id}/resume`,
+                method: 'POST',
+              });
             }
             setLog('subscribed');
             break;

diff --git a/src/apis/master.ts b/src/apis/master.ts
@@ -44,6 +44,13 @@ export default [
       return new PeerService(getDataSource()).connect(data);
     },
   },
+  {
+    method: 'POST',
+    url: '/peers/:peerId/resume',
+    handler: (data) => {
+      return new PeerService(getDataSource()).resume(data);
+    },
+  },
   {
     method: 'POST',
     url: '/peers/:peerId/produce',

diff --git a/src/apis/slave.ts b/src/apis/slave.ts
@@ -64,4 +64,11 @@ export default [
       return mediasoupConsumerManager.create(data);
     },
   },
+  {
+    method: 'POST',
+    url: '/consumers/:consumerId/resume',
+    handler: (data) => {
+      return mediasoupConsumerManager.resume(data);
+    },
+  },
 ] as RouteConfig[];
diff --git a/src/services/mediasoup/consumer.ts b/src/services/mediasoup/consumer.ts
@@ -35,9 +35,7 @@ class MediasoupConsumerManager {
       rtpCapabilities: data.rtpCapabilities,
       paused: true,
     });
-    setTimeout(() => {
-      consumer.resume();
-    }, 2000);
+
     MediasoupConsumerManager.consumers.push(consumer);
 
     return {
@@ -48,6 +46,21 @@ class MediasoupConsumerManager {
       producerPaused: consumer.producerPaused,
     };
   }
+
+  resume(data: { consumerId: string }) {
+    const consumer = this.get(data);
+    if (consumer) {
+      consumer.resume();
+      return {};
+    }
+    throw new ServiceError(404, 'Consumer not found');
+  }
+
+  get(data: { consumerId: string }) {
+    return MediasoupConsumerManager.consumers.find(
+      (item) => item.id === data.consumerId
+    );
+  }
 }
 
 export const mediasoupConsumerManager = new MediasoupConsumerManager();
diff --git a/src/services/peer.ts b/src/services/peer.ts
@@ -129,6 +129,23 @@ export class PeerService extends BaseService {
     throw new ServiceError(404, 'Peer not found');
   }
 
+  async resume(data: { peerId: string }) {
+    const peer = await this.get({ peerId: data.peerId });
+    if (peer && peer.consumerId && peer.type === constants.CONSUMER) {
+      const result = await fetchApi({
+        host: peer.slave.externalHost,
+        port: peer.slave.apiPort,
+        path: '/consumers/:consumerId/resume',
+        method: 'POST',
+        data: {
+          consumerId: peer.consumerId,
+        },
+      });
+      return result;
+    }
+    throw new ServiceError(404, 'Peer not found');
+  }
+
   async get(data: { peerId: string }) {
     return this.dataSource.getRepository(MediaPeer).findOne({
       relations: { slave: true },