benchmark_test.go

package unleash_test

import (
	"fmt"
	"net/http"
	"testing"
	"time"

	"github.com/Unleash/unleash-go-sdk/v6"
	"github.com/Unleash/unleash-go-sdk/v6/api"
)

type mockStorage map[string]api.Feature

func (m mockStorage) Init(backupPath string, appName string) {}
func (m mockStorage) Load() (*api.FeatureResponse, error) {

	features := make([]api.Feature, 0, len(m))
	for _, f := range m {
		features = append(features, f)
	}

	return &api.FeatureResponse{
		Features: features,
		Segments: make([]api.Segment, 0),
	}, nil
}

func (m mockStorage) Persist(*api.FeatureResponse) error { return nil }

func ptr[T any](v T) *T {
	return &v
}

func BenchmarkFeatureToggleEvaluation(b *testing.B) {
	err := unleash.Initialize(
		unleash.WithListener(&unleash.NoopListener{}),
		unleash.WithAppName("go-benchmark"),
		unleash.WithUrl("https://app.unleash-hosted.com/demo/api/"),
		unleash.WithCustomHeaders(http.Header{"Authorization": {"Go-Benchmark:development.be6b5d318c8e77469efb58590022bb6416100261accf95a15046c04d"}}),
		unleash.WithStorage(mockStorage{
			"foo": api.Feature{
				Name:    "foo",
				Enabled: true,
				Dependencies: &[]api.Dependency{
					{Feature: "bar", Enabled: ptr(true)},
				},
			},
			"bar": api.Feature{
				Name:    "bar",
				Enabled: true,
			},
		}),
	)
	if err != nil {
		b.Fatal(err)
	}

	startTime := time.Now()
	b.ReportAllocs()

	for i := 0; i < b.N; i++ {
		_ = unleash.IsEnabled("foo", unleash.FeatureOptions{})
	}

	endTime := time.Now()

	// Calculate ns/op (nanoseconds per operation)
	nsPerOp := float64(endTime.Sub(startTime).Nanoseconds()) / float64(b.N)

	// Calculate operations per day
	opsPerSec := 1e9 / nsPerOp
	opsPerDay := opsPerSec * 60 * 60 * 24

	if b.N > 1000000 { // Only print if the number of iterations is large enough for a stable result
		opsPerDayBillions := opsPerDay / 1e9 // Convert to billions
		fmt.Printf("Final Estimated Operations Per Day: %.3f billion (%e)\n", opsPerDayBillions, opsPerDay)
	}
}