From 79771bebb414137e6dc34dd2fb582bf06fa7cdf3 Mon Sep 17 00:00:00 2001
From: Urko <urko@fungimail.com>
Date: Wed, 5 Apr 2023 19:22:20 +0200
Subject: [PATCH] feat: refactor benchmark

---
 benchmark/go/go.go     |  3 +--
 benchmark/main.go      | 16 +++++++++-------
 benchmark/nest/nest.go |  3 +--
 3 files changed, 11 insertions(+), 11 deletions(-)

diff --git a/benchmark/go/go.go b/benchmark/go/go.go
index 865b451..dd8863b 100644
--- a/benchmark/go/go.go
+++ b/benchmark/go/go.go
@@ -44,7 +44,7 @@ func doRequest(wg *sync.WaitGroup, host, employeeID string, errChan chan error)
 	return nil
 }
 
-func Benchmark() error {
+func Benchmark(totalRequests int) error {
 	cr := crono.New()
 	defer cr.Table()
 	host := "http://127.0.0.1:45654"
@@ -52,7 +52,6 @@ func Benchmark() error {
 	wg := &sync.WaitGroup{}
 	errChan := make(chan error, 1)
 	wg.Add(1)
-	totalRequests := 5000
 	go func(errChan chan error) {
 		defer wg.Done()
 		defer func() {
diff --git a/benchmark/main.go b/benchmark/main.go
index c92f5e6..59835f7 100644
--- a/benchmark/main.go
+++ b/benchmark/main.go
@@ -7,7 +7,8 @@ import (
 	"strings"
 	"time"
 
-	go_benchmark "gitea.urkob.com/urko/ess-etl-go/benchmark/go"
+	// go_benchmark "gitea.urkob.com/urko/ess-etl-go/benchmark/go"
+	nest_benchmark "gitea.urkob.com/urko/ess-etl-go/benchmark/nest"
 )
 
 func main() {
@@ -21,11 +22,12 @@ func main() {
 	defer f.Close()
 	log.SetOutput(f)
 
-	if err := go_benchmark.Benchmark(); err != nil {
-		log.Fatalln("go_benchmark.Benchmark", err)
-	}
-
-	// if err := nest_benchmark.Benchmark(); err != nil {
-	// 	log.Fatalln("nest_benchmark.Benchmark", err)
+	totalRequests := 5000
+	// if err := go_benchmark.Benchmark(totalRequests); err != nil {
+	// 	log.Fatalln("go_benchmark.Benchmark", err)
 	// }
+
+	if err := nest_benchmark.Benchmark(totalRequests); err != nil {
+		log.Fatalln("nest_benchmark.Benchmark", err)
+	}
 }
diff --git a/benchmark/nest/nest.go b/benchmark/nest/nest.go
index f10fb0a..f6145b4 100644
--- a/benchmark/nest/nest.go
+++ b/benchmark/nest/nest.go
@@ -70,7 +70,7 @@ func doRequest(wg *sync.WaitGroup, host string, employeeID int, errChan chan err
 	return nil
 }
 
-func Benchmark() error {
+func Benchmark(totalRequests int) error {
 	cr := crono.New()
 	defer cr.Table()
 
@@ -79,7 +79,6 @@ func Benchmark() error {
 	wg := &sync.WaitGroup{}
 	errChan := make(chan error, 1)
 	wg.Add(1)
-	totalRequests := 5000
 	go func(errChan chan error) {
 		defer wg.Done()
 		defer func() {