command_benchmark_test.go

package circuit

import (
	"context"
	"testing"

	"github.com/bunnier/circuit/breaker"
)

// run 原始的功能函数。
var run = func() string {
	return "ok"
}

// wrapRun 是包装后用于 command 用的功能函数。
var wrapRun = func(ctx context.Context, param interface{}) (interface{}, error) {
	return run(), nil
}

func BenchmarkDirectly(b *testing.B) {
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		run()
	}
	b.StopTimer()
}
func BenchmarkParallelDirectly(b *testing.B) {
	b.RunParallel(func(p *testing.PB) {
		for p.Next() {
			run()
		}
	})
}

func BenchmarkCutCommand(b *testing.B) {
	command := NewCommand("test", wrapRun)
	defer command.Close()
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		command.Execute(nil)
	}
	b.StopTimer()
}

func BenchmarkParallelCutCommand(b *testing.B) {
	command := NewCommand("test", wrapRun)
	defer command.Close()
	b.ResetTimer()
	b.RunParallel(func(p *testing.PB) {
		for p.Next() {
			command.Execute(nil)
		}
	})
	b.StopTimer()
}

func BenchmarkSreCommand(b *testing.B) {
	sreBreaker := breaker.NewSreBreaker("test")
	command := NewCommand("test", wrapRun, WithCommandBreaker(sreBreaker))
	defer command.Close()
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		command.Execute(nil)
	}
	b.StopTimer()
}

func BenchmarkParallelSreCommand(b *testing.B) {
	sreBreaker := breaker.NewSreBreaker("test")
	command := NewCommand("test", wrapRun, WithCommandBreaker(sreBreaker))
	defer command.Close()
	b.ResetTimer()
	b.RunParallel(func(p *testing.PB) {
		for p.Next() {
			command.Execute(nil)
		}
	})
	b.StopTimer()
}