Go-Spring 实战第 10 课 —— 依赖注入的方式：字段注入和构造函数注入

上一篇咱们把 Go-Spring 使用 IoC 容器的初衷和背景讲清楚了，内容偏抽象。从本文开始，咱们讲 Go-Spring IoC 容器的实际用法。

这一篇先介绍依赖注入怎么用，以及依赖注入会发生在哪些地方。其实也很简单：一是字段注入，二是构造函数注入。

字段注入

字段注入，就是把依赖写在结构体字段上，通过 autowire（或 inject）标签告诉 Go-Spring 要注入什么。

这种写法最简单也最直观。看个例子：

go 复制代码

type UserService struct{}

type UserController struct {
	Service *UserService `autowire:""`
}

func init() {
	gs.Provide(new(UserService))
	gs.Provide(new(UserController))
}

在上面的代码中，autowire:"" 标签没有值，表示只需要按类型注入。因为这里 UserController.Service 的类型是 *UserService，容器里也只有一个 *UserService，所以不需要写名字。

如果同一个类型有多个 Bean，就需要在标签里写明要注入的 Bean 名称。示例如下：

go 复制代码

type Repository struct {
	DB *DataSource `autowire:"slave"`
}

func init() {
	gs.Provide(NewMasterDataSource).Name("master")
	gs.Provide(NewSlaveDataSource).Name("slave")
	gs.Provide(new(Repository))
}

在上面的代码中，我们注册了 master 和 slave 两个 *DataSource 类型的 Bean。如果 Repository.DB 仍然只按类型注入，容器就会发现有两个相同类型的 *DataSource。这种情况下，我们就需要在 autowire 标签里写清楚要注入的 Bean 名称。

构造函数注入

构造函数注入，就是把依赖写在构造函数参数里。Go-Spring 创建 Bean 时，会先分析构造函数的参数类型，然后再根据参数类型从容器里找到对应的依赖，最后使用反射调用构造函数创建 Bean。

示例如下：

go 复制代码

type UserController struct {
	service *UserService
}

func NewUserController(service *UserService) *UserController {
	return &UserController{service: service}
}

func init() {
	gs.Provide(new(UserService))
	gs.Provide(NewUserController)
}

使用构造函数注入有两个好处。第一，依赖全都在函数签名上，一眼就能看到这个对象需要什么。第二，单元测试可以直接调用 NewUserController(mockService)，不一定要启动完整的 IoC 容器来填充所有依赖。

我们也可以让构造函数返回 error，这样能暴露更详细的错误信息。

示例如下：

go 复制代码

func NewUserController(service *UserService) (*UserController, error) {
	if service == nil {
		return nil, errors.New("missing user service")
	}
	return &UserController{service: service}, nil
}

如果构造函数返回了非 nil 的 error，Go-Spring 就会认为这个 Bean 创建失败了，然后就会终止应用启动。像配置检查、连接初始化、参数校验这类动作，都可以放在构造函数里完成。如果这些动作失败了，就可以返回 error 来终止启动，这样比创建出一个不可用的对象要更清楚。

构造函数参数

对比上面的示例可以发现，字段注入可以通过名字指定要注入的 Bean，这是因为我们使用了 Go 提供的 Tag 机制。但是 Go 没有为构造函数参数提供 Tag 机制，那还能指定参数对应的 Bean 名称吗？

看个例子。

go 复制代码

func NewRepository(ds *DataSource) *Repository {
	return &Repository{ds: ds}
}

func init() {
	gs.Provide(NewMasterDataSource).Name("master")
	gs.Provide(NewSlaveDataSource).Name("slave")

	gs.Provide(NewRepository, gs.TagArg("slave"))
}

虽然 Go 不支持在构造函数参数上写 Tag，但是 Go-Spring 通过 gs.Provide 提供了为构造函数参数指定 Bean 名称的功能。gs.TagArg("slave") 的意思和字段上的 autowire:"slave" 一样，都是指定注入名为 slave 的 Bean。

如果构造函数的参数都是 Bean 注入，而且都可以通过类型唯一匹配，那么可以省略 TagArg。

示例如下：

go 复制代码

func init() {
	gs.Provide(NewDataSource)
	gs.Provide(NewRepository)
}

对于更复杂的构造函数参数注入，后面的章节会进行专门介绍，这里先不展开了。

哪种方式更好

我们推荐在所有情况下都使用构造函数注入，因为它具有更强的规则性和统一性。但在简单场景下使用字段注入，也没有问题。