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.Viper
	logger *zap.Logger
}

func NewHandler(conf *viper.Viper, logger *zap.Logger) *Handler {
	return &Handler{conf: conf, logger: logger}
}


//RoleHandler 源码
type RoleHandler struct {
	*Handler
	roleService *service.RoleService
}

func NewRoleController(controller *Handler, roleService *service.RoleService) *RoleHandler {
	return &RoleHandler{
		Handler:     controller,
		roleService: roleService,
	}
}

//RoleService 源码
type RoleService struct {
	*Service
	roleDao *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.Viper
	logger *zap.Logger
}

func NewDao(conf *viper.Viper, logger *zap.Logger) *Dao {
	return &Dao{conf: conf, logger: logger}
}

3 provider 提供者

provider/provider.go

复制代码
package provider

import (
	"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 wireinject

package wire

import (
	"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 !wireinject

package wire

import (
	"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 main

import (
	"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)
	// 创建logger
	logger, 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的示例工程

相关推荐
华仔啊8 分钟前
工作5年没碰过分布式锁,是我太菜还是公司太稳?网友:太真实了!
java·后端
摸鱼仙人~13 分钟前
深入理解 MyBatis-Plus 的 `BaseMapper`
java·开发语言·mybatis
卿·静23 分钟前
Node.js对接即梦AI实现“千军万马”视频
前端·javascript·人工智能·后端·node.js
ITKEY_39 分钟前
flutter日期选择国际化支持
开发语言·javascript·flutter
刃神太酷啦1 小时前
C++ 异常处理机制:从基础到实践的全面解析----《Hello C++ Wrold!》(20)--(C/C++)
java·c语言·开发语言·c++·qt·算法·leetcode
q567315231 小时前
告别低效:构建健壮R爬虫的工程思维
开发语言·爬虫·r语言
SamDeepThinking1 小时前
在 Cursor IDE 中配置 SQLTools 连接 MySQL 数据库指南(Windows 11)
后端·ai编程·cursor
武子康1 小时前
大数据-92 Spark 深入解析 Spark Standalone 模式:组件构成、提交流程与性能优化
大数据·后端·spark
枫叶丹41 小时前
【Qt开发】显示类控件(一)-> QLabel
开发语言·qt
青梅主码1 小时前
麦肯锡最新发布报告《想打破生产力天花板吗?重新思考完成工作的方式》:与其一味调整组织结构,不如从根本上简化工作流程,释放更大价值
后端