perf_test.go

package module

import (
	"context"
	"testing"
)

type DB struct {
	target string
}

func NewDB(ctx context.Context) (*DB, error) {
	return &DB{
		target: "localhost.db",
	}, nil
}

type Cache struct {
	fallback *DB
}

func NewCache(ctx context.Context) (*Cache, error) {
	db := moduleDB.Value(ctx)
	return &Cache{
		fallback: db,
	}, nil
}

var (
	moduleDB     = New[*DB]()
	moduleCache  = New[*Cache]()
	provideDB    = moduleDB.ProvideWithFunc(NewDB)
	provideCache = moduleCache.ProvideWithFunc(NewCache)
)

func BenchmarkThroughModuleValue(b *testing.B) {
	repo := NewRepo()
	repo.Add(provideDB)
	repo.Add(provideCache)

	ctx, err := repo.InjectTo(context.Background())
	if err != nil {
		b.Fatal("create context error:", err)
	}

	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		var _ *Cache = moduleCache.Value(ctx)
	}
}

func BenchmarkSimpleContextValue(b *testing.B) {
	ctx := context.Background()

	db, err := NewDB(ctx)
	if err != nil {
		b.Fatal("create db error:", err)
	}
	ctx = context.WithValue(ctx, moduleDB.moduleKey, db)

	cache, err := NewCache(ctx)
	if err != nil {
		b.Fatal("create cache error:", err)
	}
	ctx = context.WithValue(ctx, moduleCache.moduleKey, cache)

	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		var _ *Cache = ctx.Value(moduleCache.moduleKey).(*Cache)
	}
}