feat: add benchmark

benchmark/benchmark_test.go

@@ -10,9 +10,42 @@ import (
 
 	go_benchmark "gitea.urkob.com/urko/ess-etl-go/benchmark/go"
 	nest_benchmark "gitea.urkob.com/urko/ess-etl-go/benchmark/nest"
+	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 )
 
+func BenchmarkGo(b *testing.B) {
+	require.NoError(b, os.RemoveAll("./dump"))
+	require.NoError(b, os.MkdirAll("./dump", os.ModeTemporary))
+	log.SetFlags(log.Lmicroseconds)
+	logFileName := fmt.Sprintf("%s.txt", time.Now().Format(strings.ReplaceAll(time.RFC1123Z, ":", "_")))
+	f, err := os.OpenFile("./dump/"+logFileName, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0o644)
+	require.NoError(b, err)
+
+	defer f.Close()
+	log.SetOutput(f)
+
+	for i := 0; i < b.N; i++ {
+		assert.NoError(b, go_benchmark.Benchmark(1, false))
+	}
+
+}
+func BenchmarkNest(b *testing.B) {
+	require.NoError(b, os.RemoveAll("./dump"))
+	require.NoError(b, os.MkdirAll("./dump", os.ModeTemporary))
+	log.SetFlags(log.Lmicroseconds)
+	logFileName := fmt.Sprintf("%s.txt", time.Now().Format(strings.ReplaceAll(time.RFC1123Z, ":", "_")))
+	f, err := os.OpenFile("./dump/"+logFileName, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0o644)
+	require.NoError(b, err)
+
+	defer f.Close()
+	log.SetOutput(f)
+
+	for i := 0; i < b.N; i++ {
+		assert.NoError(b, nest_benchmark.Benchmark(1, false))
+	}
+}
+
 func TestGoXRequestes(t *testing.T) {
 	log.SetFlags(log.Lmicroseconds)
 	logFileName := fmt.Sprintf("%s.txt", time.Now().Format(strings.ReplaceAll(time.RFC1123Z, ":", "_")))
@@ -25,7 +58,7 @@ func TestGoXRequestes(t *testing.T) {
 	log.SetOutput(f)
 
 	totalRequests := 1000
-	require.NoError(t, go_benchmark.Benchmark(totalRequests))
+	require.NoError(t, go_benchmark.Benchmark(totalRequests, true))
 	require.NoError(t, os.RemoveAll("./temp"))
 
 }
@@ -40,5 +73,5 @@ func TestNestXRequests(t *testing.T) {
 	log.SetOutput(f)
 
 	totalRequests := 1000
-	require.NoError(t, nest_benchmark.Benchmark(totalRequests))
+	require.NoError(t, nest_benchmark.Benchmark(totalRequests, true))
 }

benchmark/go/go.go

@@ -44,21 +44,22 @@ func doRequest(wg *sync.WaitGroup, host, employeeID string, errChan chan error)
 	return nil
 }
 
-func Benchmark(totalRequests int) error {
+func Benchmark(totalRequests int, dolog bool) error {
 	cr := crono.New()
+	if dolog {
 		defer cr.Table()
+	}
 	host := "http://127.0.0.1:45654"
 
 	wg := &sync.WaitGroup{}
 	errChan := make(chan error, 1)
-	wg.Add(1)
+	wg.Add(totalRequests + 1)
 	go func(errChan chan error) {
 		defer wg.Done()
 		defer func() {
 			errChan <- nil
 		}()
 		for i := 0; i < totalRequests; i++ {
-			wg.Add(1)
 			go doRequest(wg, host, "975135", errChan)
 		}
 	}(errChan)
@@ -68,6 +69,9 @@ func Benchmark(totalRequests int) error {
 	if err := <-errChan; err != nil {
 		return fmt.Errorf("Benchmark: %s", err)
 	}
+	if dolog {
 		cr.MarkAndRestart(fmt.Sprintf("go handle %d response", totalRequests))
+	}
+
 	return nil
 }

benchmark/nest/nest.go

@@ -70,10 +70,11 @@ func doRequest(wg *sync.WaitGroup, host string, employeeID int, errChan chan err
 	return nil
 }
 
-func Benchmark(totalRequests int) error {
+func Benchmark(totalRequests int, dolog bool) error {
 	cr := crono.New()
+	if dolog {
 		defer cr.Table()
-
+	}
 	host := "http://127.0.0.1:3458/graphql"
 
 	wg := &sync.WaitGroup{}
@@ -95,6 +96,8 @@ func Benchmark(totalRequests int) error {
 	if err := <-errChan; err != nil {
 		return fmt.Errorf("Benchmark: %s", err)
 	}
+	if dolog {
 		cr.MarkAndRestart(fmt.Sprintf("nest handle %d response", totalRequests))
+	}
 	return nil
 }