golang wire 依赖注入

本文主要是介绍golang wire 依赖注入，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

1 安装wire

1.1 通过命令直接安装

go install github.com/google/wire/cmd/wire@latest

1.2 通过go get方式安装

go get github.com/google/wire/cmd/wire

进入目录编译

cd C:\Users\leell\go\pkg\mod\github.com\google\wire@v0.6.0\cmd\wire
go build

然后将wire.exe移动到 C:\Users\leell\go\bin目录下

具体目录通过go env进行查看

C:\Users\leell\go>go env
set GO111MODULE=on
set GOARCH=amd64
set GOBIN=C:\Users\leell\go\bin

2  业务源码

//Handler 源码
type Handler struct {conf   *viper.Viperlogger *zap.Logger
}func NewHandler(conf *viper.Viper, logger *zap.Logger) *Handler {return &Handler{conf: conf, logger: logger}
}//RoleHandler 源码
type RoleHandler struct {*HandlerroleService *service.RoleService
}func NewRoleController(controller *Handler, roleService *service.RoleService) *RoleHandler {return &RoleHandler{Handler:     controller,roleService: roleService,}
}//RoleService 源码
type RoleService struct {*ServiceroleDao *dao.RoleDao
}func NewRoleService(service *Service, roleDao *dao.RoleDao) *RoleService {return &RoleService{Service: service,roleDao: roleDao,}
}//RoleDao源码
type RoleDao struct {*Dao
}func NewRoleDao(dao *Dao) *RoleDao {return &RoleDao{Dao: dao}
}//Dao 源码
type Dao struct {conf   *viper.Viperlogger *zap.Logger
}func NewDao(conf *viper.Viper, logger *zap.Logger) *Dao {return &Dao{conf: conf, logger: logger}
}

3 provider 提供者

provider/provider.go

package providerimport ("github.com/google/wire""wire-first/dao""wire-first/handler""wire-first/server""wire-first/service"
)var ServerSet = wire.NewSet(server.NewServerHttp)var HandlerSet = wire.NewSet(handler.NewHandler, handler.NewUserController, handler.NewRoleController)var ServiceSet = wire.NewSet(service.NewService, service.NewUserService, service.NewRoleService)var DaoSet = wire.NewSet(dao.NewDao, dao.NewUserDao, dao.NewRoleDao)

4 wire提供生成构建代码

wire/wire.go

//go:build wireinject
// +build wireinjectpackage wireimport ("net/http""github.com/google/wire""github.com/spf13/viper""go.uber.org/zap""wire-first/provider"
)// wire.go 初始化模块
func NewApp(*viper.Viper, *zap.Logger) (*http.Server, error) {panic(wire.Build(provider.ServerSet,provider.HandlerSet,provider.ServiceSet,provider.DaoSet,))
}

5 通过wire生成gen代码 

PS G:\github\wire-first> wire  gen wire/wire.go
wire: command-line-arguments: wrote G:\github\wire-first\wire\wire_gen.go

6 结果查看

wire/wire_gen.go

// Code generated by Wire. DO NOT EDIT.//go:generate go run -mod=mod github.com/google/wire/cmd/wire
//go:build !wireinject
// +build !wireinjectpackage wireimport ("github.com/spf13/viper""go.uber.org/zap""net/http""wire-first/dao""wire-first/handler""wire-first/server""wire-first/service"
)// Injectors from wire.go:// wire.go 初始化模块
func NewApp(viperViper *viper.Viper, logger *zap.Logger) (*http.Server, error) {handlerHandler := handler.NewHandler(viperViper, logger)serviceService := service.NewService(viperViper, logger)daoDao := dao.NewDao(viperViper, logger)userDao := dao.NewUserDao(daoDao)userService := service.NewUserService(serviceService, userDao)userHandler := handler.NewUserController(handlerHandler, userService)roleDao := dao.NewRoleDao(daoDao)roleService := service.NewRoleService(serviceService, roleDao)roleHandler := handler.NewRoleController(handlerHandler, roleService)httpServer := server.NewServerHttp(viperViper, logger, userHandler, roleHandler)return httpServer, nil
}

7 main.go代码

package mainimport ("github.com/alecthomas/kingpin/v2""go.uber.org/zap""wire-first/config""wire-first/wire"
)var (cfgPath = kingpin.Flag("config", "the path of the config file").Default("app.yml").String()
)func main() {kingpin.Parse()conf := config.NewConfig(*cfgPath)// 创建loggerlogger, err := zap.NewDevelopment()if err != nil {panic(err)}defer logger.Sync()app, err := wire.NewApp(conf, logger)if err != nil {logger.Error("Initialization failed",zap.Error(err),)}logger.Info("Server's running", zap.String("address", app.Addr))if err := app.ListenAndServe(); err != nil {logger.Error("Server Error",zap.String("key", "value"),zap.Error(err),)}
}

源码：wire-first: 依赖注入库wire的示例工程

这篇关于golang wire 依赖注入的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---