diff --git a/benchmark/benchmark_test.go b/benchmark/benchmark_test.go index cffb62a..f321b4d 100644 --- a/benchmark/benchmark_test.go +++ b/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)) } diff --git a/benchmark/go/go.go b/benchmark/go/go.go index dd8863b..88c446d 100644 --- a/benchmark/go/go.go +++ b/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() - defer cr.Table() + 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) } - cr.MarkAndRestart(fmt.Sprintf("go handle %d response", totalRequests)) + if dolog { + cr.MarkAndRestart(fmt.Sprintf("go handle %d response", totalRequests)) + } + return nil } diff --git a/benchmark/nest/nest.go b/benchmark/nest/nest.go index f6145b4..ca91609 100644 --- a/benchmark/nest/nest.go +++ b/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() - defer cr.Table() - + 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) } - cr.MarkAndRestart(fmt.Sprintf("nest handle %d response", totalRequests)) + if dolog { + cr.MarkAndRestart(fmt.Sprintf("nest handle %d response", totalRequests)) + } return nil }