[fix] H264 화면 송출 에러 수정 (#30)

* fix: H264 화면 송출 에러 fix * feat: RTMP 서버 80 포트 추가 * feat: node_relay_session.js 파일 추가 --------- Co-authored-by: kkg5 <[email protected]>
boostcampwm2023 · Nov 15, 2023 · 305dba7 · 305dba7
1 parent f0f7421
commit 305dba7
Show file tree

Hide file tree

Showing 4 changed files with 91 additions and 2 deletions.
diff --git a/.github/workflows/deploy.yml b/.github/workflows/deploy.yml
@@ -89,5 +89,5 @@ jobs:
             docker pull ${{ secrets.NCP_CONTAINER_REGISTRY }}/rtmp-server
             docker stop $(docker ps -a -q)
             docker rm $(docker ps -a -q)
-            docker run -d -p 1935:1935 --env-file ${{ secrets.ENV_PATH }} ${{ secrets.NCP_CONTAINER_REGISTRY }}/rtmp-server
+            docker run -d -p 1935:1935 -p 80:80 --env-file ${{ secrets.ENV_PATH }} ${{ secrets.NCP_CONTAINER_REGISTRY }}/rtmp-server
             docker image prune -f
diff --git a/server/rtmp-server/Dockerfile b/server/rtmp-server/Dockerfile
@@ -11,6 +11,8 @@ RUN npm i
 
 COPY ./server/rtmp-server/. .
 
-EXPOSE 1935
+RUN mv ./node_relay_session.js ./node_modules/node-media-server/src/node_relay_session.js
+
+EXPOSE 1935 80
 
 CMD ["node","app.js"]
diff --git a/server/rtmp-server/app.js b/server/rtmp-server/app.js
@@ -1,4 +1,5 @@
 require("dotenv").config();
+
 const NodeMediaServer = require("node-media-server");
 
 const config = {
@@ -9,6 +10,10 @@ const config = {
     ping: 30,
     ping_timeout: 60,
   },
+  http: {
+    port: 80,
+    allow_origin: "*",
+  },
   relay: {
     ffmpeg: "/usr/local/bin/ffmpeg",
     tasks: [
@@ -22,4 +27,5 @@ const config = {
 };
 
 const nms = new NodeMediaServer(config);
+
 nms.run();
diff --git a/server/rtmp-server/node_relay_session.js b/server/rtmp-server/node_relay_session.js
@@ -0,0 +1,81 @@
+//
+//  Created by Mingliang Chen on 18/3/16.
+//  illuspas[a]gmail.com
+//  Copyright (c) 2018 Nodemedia. All rights reserved.
+//
+const Logger = require("./node_core_logger");
+const NodeCoreUtils = require("./node_core_utils");
+
+const EventEmitter = require("events");
+const { spawn } = require("child_process");
+
+const RTSP_TRANSPORT = ["udp", "tcp", "udp_multicast", "http"];
+
+class NodeRelaySession extends EventEmitter {
+  constructor(conf) {
+    super();
+    this.conf = conf;
+    this.id = NodeCoreUtils.generateNewSessionID();
+    this.ts = (Date.now() / 1000) | 0;
+    this.TAG = "relay";
+  }
+
+  run() {
+    let format = this.conf.ouPath.startsWith("rtsp://") ? "rtsp" : "flv";
+    let argv = [
+      "-re",
+      "-i",
+      this.conf.inPath,
+      "-c:v",
+      "libx264",
+      "-preset",
+      "veryfast",
+      "-tune",
+      "zerolatency",
+      "-c:a",
+      "aac",
+      "-ar",
+      "44100",
+      "-f",
+      format,
+      this.conf.ouPath,
+    ];
+    if (this.conf.inPath[0] === "/" || this.conf.inPath[1] === ":") {
+      argv.unshift("-1");
+      argv.unshift("-stream_loop");
+    }
+
+    if (this.conf.inPath.startsWith("rtsp://") && this.conf.rtsp_transport) {
+      if (RTSP_TRANSPORT.indexOf(this.conf.rtsp_transport) > -1) {
+        argv.unshift(this.conf.rtsp_transport);
+        argv.unshift("-rtsp_transport");
+      }
+    }
+
+    Logger.log("[relay task] id=" + this.id, "cmd=ffmpeg", argv.join(" "));
+
+    this.ffmpeg_exec = spawn(this.conf.ffmpeg, argv);
+    this.ffmpeg_exec.on("error", (e) => {
+      Logger.ffdebug(e);
+    });
+
+    this.ffmpeg_exec.stdout.on("data", (data) => {
+      Logger.ffdebug(`FF_LOG:${data}`);
+    });
+
+    this.ffmpeg_exec.stderr.on("data", (data) => {
+      Logger.ffdebug(`FF_LOG:${data}`);
+    });
+
+    this.ffmpeg_exec.on("close", (code) => {
+      Logger.log("[relay end] id=" + this.id, "code=" + code);
+      this.emit("end", this.id);
+    });
+  }
+
+  end() {
+    this.ffmpeg_exec.kill();
+  }
+}
+
+module.exports = NodeRelaySession;