Gin实战准备-Go依赖注入：Wire如何让你的代码优雅如诗？

// 数据库模块
type Database struct{ DSN string }

func NewDatabase(dsn string) *Database {
    return &Database{DSN: dsn}
}

// 用户服务模块
type UserService struct {
    db *Database
}

func NewUserService(db *Database) *UserService {
    return &UserService{db: db}
}

//go:build wireinject

package main
import "github.com/google/wire"

func InitializeUserService() *UserService {
    wire.Build(
        NewDatabase,   // 数据库构造器
        NewUserService, // 用户服务构造器
    )
    return nil // 返回值无实际意义
}

wire

func InitializeUserService() *UserService {
    db := NewDatabase("mysql://root@localhost:3306")
    return NewUserService(db)
}

func main() {
    userService := InitializeUserService()
    fmt.Println("Database连接:", userService.db.DSN)
}

var SuperSet = wire.NewSet(
    NewDatabase, 
    NewUserService,
    wire.Struct(new(UserService), "db"), // 自动注入结构体字段
)

// 跨模块组合
var InfrastructureSet = wire.NewSet(
    database.Set, 
    cache.Set,
)

type Logger interface { Log(string) }

// 实现ConsoleLogger
wire.Bind(new(Logger), new(*ConsoleLogger)) // 接口与实现绑定

func InitializeUserService(dsn string) *UserService {
    wire.Build(
        NewDatabaseProvider(dsn), // 工厂函数传递参数
        NewUserService,
    )
    return nil
}

// 错误示例：循环依赖
func NewA(b *B) *A { ... }
func NewB(a *A) *B { ... } // 编译时报错！

// 解决方案：引入接口解耦
type AInterface interface{ ... }
wire.Bind(new(AInterface), new(*A))