Add size judgment to gop cache

illuspas · Dec 3, 2024 · fb08b3e · fb08b3e
1 parent e7be993
commit fb08b3e
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 21 deletions.
diff --git a/src/protocol/rtmp.js b/src/protocol/rtmp.js
@@ -664,7 +664,6 @@ export default class Rtmp {
     let payload = this.parserPacket.payload.subarray(offset, this.parserPacket.header.length);
 
     let invokeMessage = AMF.decodeAmf0Cmd(payload);
-    logger.debug(`handle invoke message ${invokeMessage.cmd}`);
     switch (invokeMessage.cmd) {
     case "connect":
       this.onConnect(invokeMessage);
@@ -682,7 +681,7 @@ export default class Rtmp {
       this.onDeleteStream(invokeMessage);
       break;
     default:
-      logger.debug(`unhandle invoke message ${invokeMessage.cmd}`);
+      logger.trace(`unhandle invoke message ${invokeMessage.cmd}`);
       break;
     }
   }

diff --git a/src/server/broadcast_server.js b/src/server/broadcast_server.js
@@ -117,6 +117,11 @@ export default class BroadcastServer {
       this.flvMetaData = null;
       this.flvAudioHeader = null;
       this.flvVideoHeader = null;
+      this.rtmpMetaData = null;
+      this.rtmpAudioHeader = null;
+      this.rtmpVideoHeader = null;
+      this.flvGopCache?.clear();
+      this.rtmpGopCache?.clear();
     }
   };
 
@@ -132,43 +137,36 @@ export default class BroadcastServer {
       this.rtmpAudioHeader = Buffer.from(rtmpMessage);
       break;
     case 1:
-      if (this.flvGopCache !== null) {
-        this.flvGopCache.add(flvMessage);
-      }
-      if (this.rtmpGopCache !== null) {
-        this.rtmpGopCache.add(rtmpMessage);
-      }
+      this.flvGopCache?.add(flvMessage);
+      this.rtmpGopCache?.add(rtmpMessage);
       break;
     case 2:
       this.flvVideoHeader = Buffer.from(flvMessage);
       this.rtmpVideoHeader = Buffer.from(rtmpMessage);
       break;
     case 3:
-      if (this.flvGopCache !== null) {
-        this.flvGopCache.clear();
-      }
-      if (this.rtmpGopCache !== null) {
-        this.rtmpGopCache.clear();
-      }
+      this.flvGopCache?.clear();
+      this.rtmpGopCache?.clear();
       this.flvGopCache = new Set();
       this.rtmpGopCache = new Set();
       this.flvGopCache.add(flvMessage);
       this.rtmpGopCache.add(rtmpMessage);
       break;
     case 4:
-      if (this.flvGopCache !== null) {
-        this.flvGopCache.add(flvMessage);
-      }
-      if (this.rtmpGopCache !== null) {
-        this.rtmpGopCache.add(rtmpMessage);
-      }
+      this.flvGopCache?.add(flvMessage);
+      this.rtmpGopCache?.add(rtmpMessage);
       break;
     case 5:
       this.flvMetaData = Buffer.from(flvMessage);
       this.rtmpMetaData = Buffer.from(rtmpMessage);
       break;
     }
-
+    if (this.flvGopCache && this.flvGopCache.size > 4096) {
+      this.flvGopCache.clear();
+    }
+    if (this.rtmpGopCache && this.rtmpGopCache.size > 4096) {
+      this.rtmpGopCache.clear();
+    }
     this.subscribers.forEach((v, k) => {
       switch (v.protocol) {
       case "flv":

diff --git a/src/server/http_server.js b/src/server/http_server.js
@@ -27,6 +27,11 @@ export default class NodeHttpServer {
       allowHTTP1: true
     };
 
+    app.all("*", (req, res, next) => {
+      res.header("Access-Control-Allow-Origin", "*");
+      req.method === "OPTIONS" ? res.sendStatus(200) : next();
+    });
+
     app.all("/:app/:name.flv", this.handleFlv);
 
     this.server1 = http.createServer(app);