依赖注入(di)是一种设计模式,它将对象与其依赖项分离开来。在 go 中,wire 框架使用反射来自动注入依赖项:定义提供程序函数创建依赖项。在 wire 文件中声明需要注入的类型。使用 wire.build() 函数构建依赖项。例如,使用 userrepository 的 userservice 可以通过 wire 注入 userrepository 来实现。
Go 框架中的依赖注入 (DI)
依赖注入 (DI) 是一种设计模式,用于将对象及其依赖项分离开来。它使代码更具可测试性、可维护性和可重用性。
在 Go 中,有几种框架支持 DI,包括:
- [Dingo](https://github.com/vinkdong/dingo)
- [Constructor](https://github.com/rightscale/constructor)
- [Wire](https://github.com/google/wire)
Wire 框架
Wire 是一个流行的 Go DI 框架,它使用反射来自动为结构类型注入依赖项。要使用 Wire,请执行以下步骤:
- 声明一个 Provider 函数,用于创建依赖项:
func NewRepository() *Repository {
return &Repository{}
}
- 在 wire.go 文件中声明需要注入的类型:
package main
import (
"github.com/google/wire"
"github.com/my/app/repository"
)
var userRepositorySet = wire.NewSet(repository.NewRepository)
- 使用 wire.Build() 函数来构建依赖项:
func main() {
wire.Build(userRepositorySet)
}
实战案例
假设我们有一个 UserService,它依赖于 UserRepository:
type UserService struct {
repo *UserRepository
}
func NewUserService(repo *UserRepository) *UserService {
return &UserService{repo: repo}
}
使用 Wire 注入 UserRepository:
package main
import (
"github.com/google/wire"
"github.com/my/app/repository"
"github.com/my/app/service"
)
var userServiceSet = wire.NewSet(
repository.NewRepository,
wire.Bind(new(service.UserRepository), new(*repository.Repository)),
service.NewUserService,
)