sorbet/pkg/ioc/ioc.go

package ioc

import (
	"context"
	"errors"
	"reflect"
)

var (
	ErrValueNotFound = errors.New("ioc: value not found")
)

var global = New()

// Fork 分支
func Fork() *Container {
	return global.Fork()
}

// Bind 绑定值到容器，有效类型：
//
// - 接口的具体实现值
// - 结构体的实例
// - 类型的值（尽量不要使用原始类型，而应该使用元素类型的变体）
func Bind(instance any) {
	global.Bind(instance)
}

// NamedBind 绑定具名值到容器
func NamedBind(name string, instance any) {
	global.NamedBind(name, instance)
}

// Factory 绑定工厂函数
func Factory(factory any, shared ...bool) error {
	return global.Factory(factory, shared...)
}

func MustFactory(factory any, shared ...bool) {
	err := Factory(factory, shared...)
	if err != nil {
		panic(err)
	}
}

// NamedFactory 绑定具名工厂函数
func NamedFactory(name string, factory any, shared ...bool) error {
	return global.NamedFactory(name, factory, shared...)
}

func MustNamedFactory(name string, factory any, shared ...bool) {
	err := NamedFactory(name, factory, shared...)
	if err != nil {
		panic(err)
	}
}

// Resolve 完成的注入
func Resolve(i any) error {
	return global.Resolve(i)
}

// Get 获取指定类型的值
func Get[T any](ctx context.Context) (*T, error) {
	return NamedGet[T](ctx, "")
}

func MustGet[T any](ctx context.Context) *T {
	return MustNamedGet[T](ctx, "")
}

// NamedGet 通过注入的名称获取指定类型的值
func NamedGet[T any](ctx context.Context, name string) (*T, error) {
	var abs T
	t := reflect.TypeOf(&abs)
	v := global.NamedGet(name, t)
	if !v.IsValid() {
		return nil, ErrValueNotFound
	}
	if x, ok := v.Interface().(*T); ok {
		return x, nil
	}
	return nil, ErrValueNotFound
}

func MustNamedGet[T any](ctx context.Context, name string) *T {
	v, err := NamedGet[T](ctx, name)
	if err != nil {
		panic(err)
	}
	return v
}

// Invoke 执行函数
func Invoke(f any) ([]reflect.Value, error) {
	return global.Invoke(f)
}