fix manual benchmarks (#2249)

bench/proxy/Dockerfile

@@ -1,9 +1,10 @@
-FROM golang:1.20-alpine3.18
+ARG BASE_IMAGE
+FROM $BASE_IMAGE
 
 RUN apk add --no-cache \
     ffmpeg
 
-RUN wget -O /video.mkv http://jell.yfish.us/media/jellyfish-10-mbps-hd-h264.mkv
+RUN wget -O /video.mkv http://www.larmoire.info/jellyfish/media/jellyfish-10-mbps-hd-h264.mkv
 
 WORKDIR /s
 

bench/proxy/start.sh

@@ -1,13 +1,16 @@
 #!/bin/sh -e
 
-PROXY_COUNT=20
+PROXY_COUNT=50
 PROXY_PROTOCOL=tcp
 
 #####################################################
 # source
 
 CONF=""
+CONF="${CONF}rtmp: no\n"
 CONF="${CONF}hls: no\n"
+CONF="${CONF}webrtc: no\n"
+CONF="${CONF}srt: no\n"
 CONF="${CONF}rtspAddress: :8555\n"
 CONF="${CONF}rtpAddress: :8002\n"
 CONF="${CONF}rtcpAddress: :8003\n"
@@ -15,7 +18,7 @@ CONF="${CONF}paths:\n"
 CONF="${CONF}  all:\n"
 echo -e "$CONF" > /source.conf
 
-RTSP_RTMP=no /mediamtx /source.conf &
+/mediamtx /source.conf &
 
 sleep 1
 
@@ -28,7 +31,10 @@ sleep 1
 # proxy
 
 CONF=""
+CONF="${CONF}rtmp: no\n"
 CONF="${CONF}hls: no\n"
+CONF="${CONF}webrtc: no\n"
+CONF="${CONF}srt: no\n"
 CONF="${CONF}pprof: yes\n"
 CONF="${CONF}paths:\n"
 for i in $(seq 1 $PROXY_COUNT); do
@@ -38,7 +44,7 @@ for i in $(seq 1 $PROXY_COUNT); do
 done
 echo -e "$CONF" > /proxy.conf
 
-RTSP_RTMP=no /mediamtx /proxy.conf &
+/mediamtx /proxy.conf &
 
 sleep 5
 

bench/publish/Dockerfile

@@ -1,9 +1,10 @@
-FROM golang:1.20-alpine3.18
+ARG BASE_IMAGE
+FROM $BASE_IMAGE
 
 RUN apk add --no-cache \
     ffmpeg
 
-RUN wget -O /video.mkv http://jell.yfish.us/media/jellyfish-10-mbps-hd-h264.mkv
+RUN wget -O /video.mkv http://www.larmoire.info/jellyfish/media/jellyfish-10-mbps-hd-h264.mkv
 
 WORKDIR /s
 

bench/publish/start.sh

@@ -1,6 +1,6 @@
 #!/bin/sh -e
 
-PUBLISHER_COUNT=20
+PUBLISHER_COUNT=50
 PUBLISHER_PROTOCOL=tcp
 
 #####################################################

bench/read/Dockerfile

@@ -1,9 +1,10 @@
-FROM golang:1.20-alpine3.18
+ARG BASE_IMAGE
+FROM $BASE_IMAGE
 
 RUN apk add --no-cache \
     ffmpeg
 
-RUN wget -O /video.mkv http://jell.yfish.us/media/jellyfish-10-mbps-hd-h264.mkv
+RUN wget -O /video.mkv http://www.larmoire.info/jellyfish/media/jellyfish-10-mbps-hd-h264.mkv
 
 WORKDIR /s
 

bench/read/start.sh

@@ -1,6 +1,6 @@
 #!/bin/sh -e
 
-READER_COUNT=20
+READER_COUNT=50
 READER_PROTOCOL=tcp
 
 #####################################################
@@ -26,7 +26,7 @@ sleep 1
 for i in $(seq 1 $READER_COUNT); do
     ffmpeg  -hide_banner -loglevel error \
     -rtsp_transport $READER_PROTOCOL \
-    -i rtsp://localhost:8554/source -c copy -f mpegts -y /dev/null &
+    -i rtsp://localhost:8554/source -c copy -f null - &
 done
 
 sleep 5

scripts/bench.mk

@@ -1,3 +1,4 @@
 bench:
-	docker build -q . -f bench/$(NAME)/Dockerfile -t temp
+	docker build -q . -f bench/$(NAME)/Dockerfile -t temp \
+	--build-arg BASE_IMAGE=$(BASE_IMAGE)
 	docker run --rm -it -p 9999:9999 temp