samber/do: Go 泛型 DI

go get github.com/samber/do/v2

package main

import (
	`fmt`
	`github.com/samber/do/v2`
)

// ==========================================
// 1. 定义底层依赖: Repository
// ==========================================
type DBRepository struct{}

func (repo *DBRepository) GetUser() string {
	return `Alice`
}

// 构造函数：必须符合规范 func(i do.Injector) (T, error)
func NewDBRepository(i do.Injector) (*DBRepository, error) {
	fmt.Println(`--> 初始化 DBRepository`)
	return &DBRepository{}, nil
}

// ==========================================
// 2. 定义上层服务: Service (依赖 Repository)
// ==========================================
type UserService struct {
	repo *DBRepository
}

func (s *UserService) PrintUser() {
	fmt.Println(`User is:`, s.repo.GetUser())
}

// 构造函数：从 Injector 中取出所需依赖
func NewUserService(i do.Injector) (*UserService, error) {
	fmt.Println(`--> 初始化 UserService`)

	// 使用 do.MustInvoke 获取已注册的依赖
	repo := do.MustInvoke[*DBRepository](i)

	return &UserService{repo: repo}, nil
}

// ==========================================
// 3. 主函数
// ==========================================
func main() {
	// 创建一个新的 DI 容器
	injector := do.New()

	// 注册服务。
	// 注意：do.Provide 默认是懒加载 (Lazy loading)，所以这里的注册顺序不重要。
	do.Provide(injector, NewUserService)
	do.Provide(injector, NewDBRepository)

	fmt.Println(`容器初始化完成，准备提取服务...`)

	// 提取 UserService 并使用。
	// 此时才会触发 UserService 和 DBRepository 的实例化。
	userService := do.MustInvoke[*UserService](injector)
	userService.PrintUser()
}

容器初始化完成，准备提取服务...
--> 初始化 UserService
--> 初始化 DBRepository
User is: Alice

package main

import (
	`fmt`
	`github.com/samber/do/v2`
)

// 定义一个接口
type Notifier interface {
	Send(msg string)
}

// 实现该接口的具体结构体
type EmailNotifier struct{}

func (e *EmailNotifier) Send(msg string) {
	fmt.Println(`发送邮件通知:`, msg)
}

// 构造函数：返回的是接口类型 Notifier，而不是 *EmailNotifier
func NewEmailNotifier(i do.Injector) (Notifier, error) {
	return &EmailNotifier{}, nil
}

func main() {
	injector := do.New()

	// 注册接口与其实现
	do.Provide(injector, NewEmailNotifier)

	// 提取时，通过接口类型获取
	notifier := do.MustInvoke[Notifier](injector)
	notifier.Send(`系统异常警告！`)
}

package main

import (
	`fmt`
	`github.com/samber/do/v2`
)

type Database struct {
	conn string
}

// 实现 Shutdown() error 接口
func (db *Database) Shutdown() error {
	fmt.Println(`>>> 关闭数据库连接:`, db.conn)
	db.conn = ``
	return nil
}

func NewDatabase(i do.Injector) (*Database, error) {
	fmt.Println(`<<< 开启数据库连接`)
	return &Database{conn: `tcp://localhost:3306`}, nil
}

func main() {
	injector := do.New()
	do.Provide(injector, NewDatabase)

	// 触发实例化
	_ = do.MustInvoke[*Database](injector)

	fmt.Println(`程序运行中...`)

	// 销毁容器，自动调用所有已实例化服务的 Shutdown() 方法
	// 建议在 main 函数使用 defer injector.Shutdown()
	injector.Shutdown()

	fmt.Println(`程序退出`)
}

<<< 开启数据库连接
程序运行中...
>>> 关闭数据库连接: tcp://localhost:3306
程序退出

package main

import (
	`fmt`
	`github.com/samber/do/v2`
)

type Config struct {
	Port int
}

type RandomGenerator struct{}

func NewRandomGenerator(i do.Injector) (*RandomGenerator, error) {
    fmt.Println(`创建新的 Generator 实例`)
	return &RandomGenerator{}, nil
}

func main() {
	injector := do.New()

	// 1. ProvideValue: 直接注入一个已存在的对象/值 (非常适合配置项注入)
	cfg := &Config{Port: 8080}
	do.ProvideValue(injector, cfg)

	// 2. ProvideTransient: 瞬态服务注入 (每次 Invoke 都会执行构造函数创建一个新实例)
	do.ProvideTransient(injector, NewRandomGenerator)

	// 测试获取 Config
	loadedConfig := do.MustInvoke[*Config](injector)
	fmt.Println(`Config Port:`, loadedConfig.Port)

	// 测试获取 Transient 服务
	_ = do.MustInvoke[*RandomGenerator](injector) // 打印: 创建新的 Generator 实例
	_ = do.MustInvoke[*RandomGenerator](injector) // 打印: 创建新的 Generator 实例
}