From 6d98532f199efe9de95e30bd47fd383c07947023 Mon Sep 17 00:00:00 2001
From: aler9 <46489434+aler9@users.noreply.github.com>
Date: Wed, 6 Jan 2021 23:24:05 +0100
Subject: [PATCH] convert stress test into benchmark tests

---
 Makefile                                | 6 +++---
 {stress/read => bench/proxy}/Dockerfile | 2 +-
 {stress => bench}/proxy/start.sh        | 6 +++++-
 {stress => bench}/publish/Dockerfile    | 2 +-
 {stress => bench}/publish/start.sh      | 4 +++-
 {stress/proxy => bench/read}/Dockerfile | 2 +-
 {stress => bench}/read/start.sh         | 4 +++-
 7 files changed, 17 insertions(+), 9 deletions(-)
 rename {stress/read => bench/proxy}/Dockerfile (91%)
 rename {stress => bench}/proxy/start.sh (86%)
 rename {stress => bench}/publish/Dockerfile (90%)
 rename {stress => bench}/publish/start.sh (84%)
 rename {stress/proxy => bench/read}/Dockerfile (91%)
 rename {stress => bench}/read/start.sh (87%)

diff --git a/Makefile b/Makefile
index 3690c827..565e27b9 100644
--- a/Makefile
+++ b/Makefile
@@ -13,7 +13,7 @@ help:
 	@echo "  format         format source files"
 	@echo "  test           run tests"
 	@echo "  lint           run linters"
-	@echo "  stress NAME=n  run stress environment"
+	@echo "  bench NAME=n  run bench environment"
 	@echo "  run            run app"
 	@echo "  release        build release assets"
 	@echo "  dockerhub      build and push docker hub images"
@@ -60,8 +60,8 @@ lint:
 	$(GO_LINT_IMAGE) \
 	golangci-lint run -v
 
-stress:
-	docker build -q . -f stress/$(NAME)/Dockerfile -t temp
+bench:
+	docker build -q . -f bench/$(NAME)/Dockerfile -t temp
 	docker run --rm -it -p 9999:9999 temp
 
 define DOCKERFILE_RUN
diff --git a/stress/read/Dockerfile b/bench/proxy/Dockerfile
similarity index 91%
rename from stress/read/Dockerfile
rename to bench/proxy/Dockerfile
index 1e8264f1..5269346d 100644
--- a/stress/read/Dockerfile
+++ b/bench/proxy/Dockerfile
@@ -13,7 +13,7 @@ RUN go mod download
 COPY . ./
 RUN go build -o /rtsp-simple-server .
 
-COPY stress/read/start.sh /
+COPY bench/proxy/start.sh /
 RUN chmod +x /start.sh
 
 ENTRYPOINT [ "/start.sh" ]
diff --git a/stress/proxy/start.sh b/bench/proxy/start.sh
similarity index 86%
rename from stress/proxy/start.sh
rename to bench/proxy/start.sh
index e0e21db6..1283ba06 100644
--- a/stress/proxy/start.sh
+++ b/bench/proxy/start.sh
@@ -34,4 +34,8 @@ for i in $(seq 1 $PROXY_COUNT); do
 done
 echo -e "$CONF" > /proxy.conf
 
-/rtsp-simple-server /proxy.conf
+/rtsp-simple-server /proxy.conf &
+
+sleep 5
+
+go tool pprof -text http://localhost:9999/debug/pprof/profile?seconds=15
diff --git a/stress/publish/Dockerfile b/bench/publish/Dockerfile
similarity index 90%
rename from stress/publish/Dockerfile
rename to bench/publish/Dockerfile
index 91cfcc41..24ec9f08 100644
--- a/stress/publish/Dockerfile
+++ b/bench/publish/Dockerfile
@@ -13,7 +13,7 @@ RUN go mod download
 COPY . ./
 RUN go build -o /rtsp-simple-server .
 
-COPY stress/publish/start.sh /
+COPY bench/publish/start.sh /
 RUN chmod +x /start.sh
 
 ENTRYPOINT [ "/start.sh" ]
diff --git a/stress/publish/start.sh b/bench/publish/start.sh
similarity index 84%
rename from stress/publish/start.sh
rename to bench/publish/start.sh
index f34e5178..e537a339 100644
--- a/stress/publish/start.sh
+++ b/bench/publish/start.sh
@@ -20,4 +20,6 @@ for i in $(seq 1 $PUBLISHER_COUNT); do
     -rtsp_transport $PUBLISHER_PROTOCOL rtsp://localhost:8554/source$i &
 done
 
-wait
+sleep 5
+
+go tool pprof -text http://localhost:9999/debug/pprof/profile?seconds=15
diff --git a/stress/proxy/Dockerfile b/bench/read/Dockerfile
similarity index 91%
rename from stress/proxy/Dockerfile
rename to bench/read/Dockerfile
index 99991422..25abb09a 100644
--- a/stress/proxy/Dockerfile
+++ b/bench/read/Dockerfile
@@ -13,7 +13,7 @@ RUN go mod download
 COPY . ./
 RUN go build -o /rtsp-simple-server .
 
-COPY stress/proxy/start.sh /
+COPY bench/read/start.sh /
 RUN chmod +x /start.sh
 
 ENTRYPOINT [ "/start.sh" ]
diff --git a/stress/read/start.sh b/bench/read/start.sh
similarity index 87%
rename from stress/read/start.sh
rename to bench/read/start.sh
index 35930b87..be575a82 100644
--- a/stress/read/start.sh
+++ b/bench/read/start.sh
@@ -27,4 +27,6 @@ for i in $(seq 1 $READER_COUNT); do
     -i rtsp://localhost:8554/source -c copy -f mpegts -y /dev/null &
 done
 
-wait
+sleep 5
+
+go tool pprof -text http://localhost:9999/debug/pprof/profile?seconds=15