golang依赖注入工具

特性

  • 使用注释中的注解
  • 自动代码生成
  • 自动检测循环依赖
  • 编译时期依赖注入
  • 自动初始化
  • 支持实例组的管理

快速开始

更多示例请参考：examples

编写代码和注解

package main

import (
	"log"

	"github.com/werbenhu/digo"
)

// @provider({"id":"main.db.url"})
func NewDbUrl() string {
	return "localhost:3306"
}

type Db struct {
	url string
}

// @provider({"id":"main.db"})
// @inject({"param":"url", "id":"main.db.url"})
func NewDb(url string) *Db {
	return &Db{
		url: url,
	}
}

type App struct {
	Db *Db
}

// @provider({"id":"main.app"})
// @inject({"param":"db", "id":"main.db"})
func NewApp(db *Db) *App {
	return &App{
		Db: db,
	}
}

func (a *App) Start() {
	log.Printf("app strat, db:%s\n", a.Db.url)
}

func main() {
	app, err := digo.Provide("main.app")
	if err == nil {
		app.(*App).Start()
	}
}

安装digogen工具

go install github.com/werbenhu/digo/digogen@v1.0.0

生成依赖注入代码

digogendigo.generated.go
digogen

运行代码

go run .\digo.generated.go .\main.go

注解详情

@provider

@provider注解表示是一个实例提供者，该实例是一个单例

  • 示例
// @provider({"id":"main.db"})
  • 支持的参数：
参数类型是否必需说明
idstring是实例的id
digo.Provide(providerId)
app, err := digo.Provide("main.app")
if err == nil {
	app.(*App).Start()
}

@inject

@inject注解表示注入一个实例到某个参数, @inject注解必须和@provider或者@group二者中的一个同时存在.

  • 示例
// @inject({"param":"db", "id":"main.db"})
  • 支持的参数：
参数类型是否必需说明
paramstring是指明哪个参数需要注入实例
idstring是指明需要注入的实例id
pkgstring否该参数需要引入特定的包
github.com/xxx/tool/v1github.com/xxx/tool/v1
// @inject({"param":"tool", "id":"main.tool", "pkg":"github.com/xxx/tool/v1"})

@group

@group注解表示将实例注册到一个组

  • 示例
// @group({"id":"main.controllers"})
  • 支持的参数：
参数类型是否必需说明
idstring是组的id
digo.Members(groupId)
ctrls, err := digo.Members("main.controllers")
if err == nil {
    for _, controller := range ctrls {
        // TODO:
    }
}