首页 > 文章列表 > golang框架如何实现DI(依赖注入)?

golang框架如何实现DI(依赖注入)?

golang DI
459 2024-08-31

依赖注入(DI)是一种设计模式,它将对象与其依赖项分离开来。在 Go 中,Wire 框架使用反射来自动注入依赖项:定义提供程序函数创建依赖项。在 Wire 文件中声明需要注入的类型。使用 wire.Build() 函数构建依赖项。例如,使用 UserRepository 的 UserService 可以通过 Wire 注入 UserRepository 来实现。

golang框架如何实现DI(依赖注入)?

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,请执行以下步骤:

  1. 声明一个 Provider 函数,用于创建依赖项:
func NewRepository() *Repository {
  return &Repository{}
}
  1. wire.go 文件中声明需要注入的类型:
package main

import (
  "github.com/google/wire"
  "github.com/my/app/repository"
)

var userRepositorySet = wire.NewSet(repository.NewRepository)
  1. 使用 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,
)