最重要的是Ensure方法，该方法扫描实例的所有export字段，并读取di标签，如果有该标签则启动注入。
判断di标签的类型来确定注入singleton或者prototype对象
测试
-
单例对象在整个容器中只有一个实例，所以不管在何处注入，获取到的指针一定是一样的。
实例对象是通过同一个工厂方法创建的，所以每个实例的指针不可以相同。
下面是测试入口代码，完整代码在github仓库，有兴趣的可以翻阅：
package main
import (
"di"
"database/sql"
"fmt"
"os"
_ "github.com/go-sql-driver/mysql"
"demo"
)
func main() {
container := di.NewContainer()
db, err := sql.Open("mysql", "root:root@tcp(localhost)/sampledb")
if err != nil {
fmt.Printf("error: %sn", err.Error())
os.Exit(1)
}
container.SetSingleton("db", db)
container.SetPrototype("b", func() (interface{}, error) {
return demo.NewB(), nil
})
a := demo.NewA()
if err := container.Ensure(a); err != nil {
fmt.Println(err)
return
}
// 打印指针，确保单例和实例的指针地址
fmt.Printf("db: %pndb1: %pnb: %pnb1: %pn", a.Db, a.Db1, &a.B, &a.B1)
}
执行之后打印出来的结果为：
db: 0xc4200b6140
db1: 0xc4200b6140
b: 0xc4200a0330
b1: 0xc4200a0338
可以看到两个db实例的指针一样，说明是同一个实例，而两个b的指针不同，说明不是一个实例。
写在最后
通过依赖注入可以很好的管理多个对象之间的实例化以及依赖关系，配合配置文件在应用初始化阶段将需要注入的实例注册到容器中，在应用的任何地方只需要在实例化时注入容器即可。没有额外依赖。
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持易采站长站。
