在 go 框架中,可以使用依赖注入实现代码复用。具体步骤如下:安装依赖项 go get github.com/google/wire创建接口和实现创建依赖图,绑定接口到实现获取已注入实例实战案例,将依赖项注入到 api 路由器中
在 Go 框架中通过依赖注入实现代码复用
依赖注入是一种设计模式,它通过在运行时向对象提供其依赖项来实现代码复用。在 Go 框架中,我们可以使用依赖注入库来简化代码重复问题。
1. 安装依赖项
立即学习“go语言免费学习笔记(深入)”;
go get github.com/google/wire
2. 创建接口和实现
// UserService 接口
type UserService interface {
GetUser(id int) (*User, error)
CreateUser(u *User) (*User, error)
}
// UserServiceImpl 实现 UserService 接口
type UserServiceImpl struct {
userRepository UserRepository
}
func (s *UserService) GetUser(id int) (*User, error) {
return s.userRepository.Get(id)
}
func (s *UserService) CreateUser(u *User) (*User, error) {
return s.userRepository.Create(u)
}
3. 创建依赖图
func CreateUserService(wire.ProviderSet) (UserService, error) {
wire.Bind(new(UserService), new(*UserServiceImpl))
wire.Bind(new(UserRepository), new(*UserRepositoryImpl))
return new(UserServiceImpl), nil
}
在这个依赖图中,我们绑定了 UserService 接口到 UserServiceImpl 实现,并绑定了 UserRepository 接口到 UserRepositoryImpl 实现。
4. 获取已注入实例
我们可以使用注入器来获取已注入的实例。
func main() {
injector, err := CreateUserService(wire.NewSet())
if err != nil {
log.Fatal(err)
}
userService := injector.Get(new(UserService))
_ = userService // 使用已注入实例
}
5. 实战案例
假设有一个 API 路由器需要一个 UserService 实例:
func GetUsers(w http.ResponseWriter, r *http.Request) {
userService := injector.Get(new(UserService))
users, err := userService.GetUsers()
if err != nil {
http.Error(w, http.StatusInternalServerError, err)
return
}
// 写入响应
}
通过使用依赖注入,我们可以轻松地将 UserService 实例注入到路由器中,而无需显式创建或传递它。这简化了代码,提高了可读性和可维护性。