129 lines
3.6 KiB
Go
129 lines
3.6 KiB
Go
package benchmark
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"strings"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
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 init() {
|
|
if err := os.RemoveAll("./dump"); err != nil {
|
|
panic(fmt.Errorf("os.RemoveAll: %s", err))
|
|
}
|
|
if err := os.MkdirAll("./dump", os.ModeTemporary); err != nil {
|
|
panic(fmt.Errorf("os.MkdirAll: %s", err))
|
|
}
|
|
}
|
|
|
|
const bmNumberOfRequests = 100
|
|
const numberOfRequestsAtOnce = 1000
|
|
const seconds = 15
|
|
|
|
func BenchmarkGo(b *testing.B) {
|
|
log.SetFlags(log.Lmicroseconds)
|
|
logFileName := fmt.Sprintf("BenchmarkGo-%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)
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
start := time.Now()
|
|
assert.NoError(b, go_benchmark.BenchmarkNoLog(bmNumberOfRequests))
|
|
b.Logf("Request took %v", time.Since(start))
|
|
b.Logf("i %d | b.N %d", i, b.N)
|
|
}
|
|
}
|
|
func BenchmarkNest(b *testing.B) {
|
|
log.SetFlags(log.Lmicroseconds)
|
|
logFileName := fmt.Sprintf("BenchmarkNest-%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)
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
start := time.Now()
|
|
assert.NoError(b, nest_benchmark.BenchmarkNoLog(bmNumberOfRequests))
|
|
b.Logf("Request took %v", time.Since(start))
|
|
b.Logf("i %d | b.N %d", i, b.N)
|
|
}
|
|
}
|
|
|
|
func TestGoRequestsPerSecondFor15s(t *testing.T) {
|
|
log.SetFlags(log.Lmicroseconds)
|
|
logFileName := fmt.Sprintf("GoRequestsPerSecondFor15s-%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(t, err)
|
|
|
|
defer f.Close()
|
|
log.SetOutput(f)
|
|
|
|
var wg sync.WaitGroup
|
|
wg.Add(seconds)
|
|
for i := 0; i < seconds; i++ {
|
|
go func() {
|
|
defer wg.Done()
|
|
require.NoError(t, go_benchmark.Benchmark(numberOfRequestsAtOnce))
|
|
}()
|
|
time.Sleep(time.Second)
|
|
}
|
|
wg.Wait()
|
|
}
|
|
|
|
func TestNestRequestsPerSecondFor15s(t *testing.T) {
|
|
log.SetFlags(log.Lmicroseconds)
|
|
logFileName := fmt.Sprintf("NestRequestsPerSecondFor15s-%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(t, err)
|
|
|
|
defer f.Close()
|
|
log.SetOutput(f)
|
|
|
|
var wg sync.WaitGroup
|
|
wg.Add(seconds)
|
|
for i := 0; i < seconds; i++ {
|
|
go func() {
|
|
defer wg.Done()
|
|
require.NoError(t, nest_benchmark.Benchmark(numberOfRequestsAtOnce))
|
|
}()
|
|
time.Sleep(time.Second)
|
|
}
|
|
wg.Wait()
|
|
}
|
|
|
|
func TestGoRequests(t *testing.T) {
|
|
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(t, err)
|
|
|
|
defer f.Close()
|
|
log.SetOutput(f)
|
|
|
|
require.NoError(t, go_benchmark.Benchmark(numberOfRequestsAtOnce))
|
|
}
|
|
|
|
func TestNestRequests(t *testing.T) {
|
|
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(t, err)
|
|
|
|
defer f.Close()
|
|
log.SetOutput(f)
|
|
|
|
require.NoError(t, nest_benchmark.Benchmark(numberOfRequestsAtOnce))
|
|
}
|