在 go-zero 中优雅使用 Google Wire 实现依赖注入

背景介绍

在微服务架构中，依赖注入（Dependency Injection, DI） 是实现模块解耦、提升可测试性和可维护性的关键技术。Go 语言由于其静态类型和简洁语法，天然不支持传统的反射式 DI 框架，因此我们需要借助工具来实现编译期的依赖注入。

本文将分享如何在 go-zero 框架 中结合 Google Wire 实现优雅的依赖注入，提升项目的可维护性和扩展性。

为什么选择 go-zero + Wire？

go-zero 是一个功能完备的微服务框架，提供了丰富的工具链和约定优于配置的开发体验。
Wire 是 Google 开源的编译期依赖注入工具，利用 Go 的类型系统生成初始化代码，避免运行时反射带来的性能损耗。

两者结合可以实现：

✅ 清晰的依赖关系管理
✅ 自动化的初始化流程
✅ 更好的单元测试支持

项目结构示例

复制代码

├── goleaf.api
├── goleaf.go
├── internal
│   ├── config
│   │   └── config.go
│   ├── handler
│   │   ├── biztag
│   │   │   └── get_biz_tag_handler.go
│   │   └── routes.go
│   ├── infra
│   │   ├── db.go
│   │   ├── etcd.go
│   │   └── metrics_plugins.go
│   ├── logic
│   │   └── biztag
│   │       └── get_biz_tag_logic.go
│   ├── manager
│   │   └── biz_tag.go
│   ├── model
│   │   └── leaf_alloc.go
│   ├── repo
│   │   └── biz_tag.go
│   ├── svc
│       ├── service_context.go

步骤一：定义依赖关系

`repo/biz_tag.go`

go 复制代码

type BizTagRepo interface {
	GetBizTagList(ctx context.Context, db *gorm.DB, offset, limit int) (int64, []*model.LeafAlloc, error)
	GetBizTagByID(ctx context.Context, db *gorm.DB, id uint) (*model.LeafAlloc, error)
	CreateBizTag(ctx context.Context, db *gorm.DB, bizTag, description string, maxID, step int64) (uint, error)
	UpdateBizTag(ctx context.Context, db *gorm.DB, id uint, bizTag *model.LeafAlloc) error
	DeleteBizTag(ctx context.Context, db *gorm.DB, id uint) error
	GetBizTagByName(ctx context.Context, db *gorm.DB, name string) (*model.LeafAlloc, error)
}

type bizTagRepo struct {
}

func NewBizTagRepo() BizTagRepo {
	return &bizTagRepo{}
}

`manager/biz_tag.go`

go 复制代码

type BizTagManager interface {
	GetBizTagById(ctx context.Context, id uint) (*types.BizTagData, error)
}

type bizTagManager struct {
	bizTagRepo repo.BizTagRepo
	db         *gorm.DB
}

func NewBizTagManager(db *gorm.DB, bizTagRepo repo.BizTagRepo) BizTagManager {
	return &bizTagManager{
		db:         db,
		bizTagRepo: bizTagRepo,
	}
}

`logic/biztag/get_biz_tag_logic.go`

go 复制代码

type GetBizTagLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext

	bizTagManager manager.BizTagManager
}

func NewGetBizTagLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetBizTagLogic {
	return &GetBizTagLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,

		bizTagManager: manager.NewBizTagManager(svcCtx.DB, svcCtx.BizTagRepo),
	}
}

步骤二：使用 Wire 生成依赖注入代码

`wire/wire_set.go`

go 复制代码

package svc

import (
	"github.com/google/wire"

	"github.com/colinrs/goleaf/internal/manager"
	"github.com/colinrs/goleaf/internal/repo"
)

var infraWireSet = wire.NewSet(
	initDB,
	initRedis,
	initAsynqProducer,
	initAsynqServer)

var repoWireSet = wire.NewSet(
	repo.NewBizTagRepo)

var managerWireSet = wire.NewSet(
	manager.NewBizTagManager)

`wire/wire.go`

go 复制代码

//go:build wireinject
// +build wireinject

package svc

import (
	"github.com/google/wire"

	"github.com/zeromicro/go-zero/core/stores/redis"
	"gorm.io/gorm"

	"github.com/colinrs/goleaf/internal/config"
	"github.com/colinrs/goleaf/internal/manager"
	"github.com/colinrs/goleaf/internal/repo"
	"github.com/colinrs/goleaf/pkg/asynqueue"
)

func NewServiceContext(c config.Config) *ServiceContext {
	panic(wire.Build(
		infraWireSet,
		repoWireSet,
		managerWireSet,
		InitServiceContext,
	))
	return &ServiceContext{}
}

func InitServiceContext(c config.Config,
	// 基础设施
	db *gorm.DB,
	redisClient *redis.Redis,
	producer asynqueue.Producer,
	server asynqueue.Server,
	// repo
	bizTagRepo repo.BizTagRepo,
	// manager / service
	bizTagManager manager.BizTagManager) *ServiceContext {
	return &ServiceContext{
		Config:        c,
		DB:            db,
		RedisClient:   redisClient,
		Producer:      producer,
		Server:        server,
		BizTagRepo:    bizTagRepo,
		BizTagManager: bizTagManager,
	}
}

生成代码

安装 Wire：

bash 复制代码

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

Makefile 示例：

makefile 复制代码

wire: 
	wire check ./internal/svc
	wire ./internal/svc

Wire 生成代码示例

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 svc

import (
	"github.com/colinrs/goleaf/internal/config"
	"github.com/colinrs/goleaf/internal/manager"
	"github.com/colinrs/goleaf/internal/repo"
	"github.com/colinrs/goleaf/pkg/asynqueue"
	"github.com/zeromicro/go-zero/core/stores/redis"
	"gorm.io/gorm"
)

// Injectors from wire.go:

func NewServiceContext(c config.Config) *ServiceContext {
	db := initDB(c)
	redis := initRedis(c)
	producer := initAsynqProducer(c)
	server := initAsynqServer(c)
	bizTagRepo := repo.NewBizTagRepo()
	bizTagManager := manager.NewBizTagManager(db, bizTagRepo)
	serviceContext := InitServiceContext(c, db, redis, producer, server, bizTagRepo, bizTagManager)
	return serviceContext
}

// wire.go:

func InitServiceContext(c config.Config,

	db *gorm.DB,
	redisClient *redis.Redis,
	producer asynqueue.Producer,
	server asynqueue.Server,

	bizTagRepo repo.BizTagRepo,

	bizTagManager manager.BizTagManager) *ServiceContext {
	return &ServiceContext{
		Config:        c,
		DB:            db,
		RedisClient:   redisClient,
		Producer:      producer,
		Server:        server,
		BizTagRepo:    bizTagRepo,
		BizTagManager: bizTagManager,
	}
}

go-zero 原生方式 vs 接入 Wire 的方式对比

类别	go-zero 原生方式	接入 Wire 后的方式	备注说明
context.Context 的传递	结构体间接传递	函数参数显式传递	Wire 更倾向函数式风格
日志 logx 的使用	每次通过 ctx 创建	函数参数中使用 ctx	更贴近标准库风格
logic 层依赖对象	每次请求创建	使用单例对象	提升性能与一致性
依赖关系管理	分散在各层	集中在构建函数	更易维护
初始化流程	手动初始化	自动生成	减少重复代码
测试友好性	手动 mock	替换构造函数	更适合单元测试
性能影响	重复创建对象	共享对象	可优化资源使用
可维护性	多处同步修改	构造函数集中修改	提升可读性
学习成本	简单直接	需理解 Wire 机制	初期成本高，长期收益大
工具支持	goctl 自动生成	Wire 手动编写	可结合使用

优势总结

✅ 清晰的依赖图：通过 wire.Build 显式声明依赖关系
✅ 编译期安全：避免运行时错误
✅ 易于测试：可替换依赖进行单元测试
✅ 与 go-zero 高度兼容：增强可维护性

注意事项

❗ Wire 不支持循环依赖，设计时需避免
❗ 生成代码需加入版本控制，避免 CI/CD 构建失败
✅ 推荐将 wire.go 和 wire_gen.go 放在单独目录中，保持整洁

结语

在实际项目中，go-zero + Wire 的组合可以帮助我们构建更清晰、更可维护的服务架构。希望这篇分享能为你的项目架构设计提供参考。