一个通过注解来实现依赖注入的golang工具

digo工具地址：https://github.com/werbenhu/digo

快速开始

更多示例请参考：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
", 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

生成依赖注入代码

打开命令行执行下面命令,digogen将会根据注解自动生成digo.generated.go源码文件.

digogen

运行代码

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

注解详情

@provider

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

• 示例

// @provider({"id":"main.db"})

• 支持的参数：

如果获取实例，通过digo.Provide(providerId)可以获取到某一个provider的实例

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

@inject

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

• 示例

// @inject({"param":"db", "id":"main.db"})

• 支持的参数：

pkg在什么时候需要使用，比如我们需要引入一个包 github.com/xxx/tool/v1 , 我们使用包名的时候是这样使用的 *tool.Struct， 而不是 *v1.Struct，那我们需要显示指明需要导入github.com/xxx/tool/v1包

// @inject({"param":"tool", "id":"main.tool", "pkg":"github.com/xxx/tool/v1"})

@group

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

• 示例

// @group({"id":"main.controllers"})

• 支持的参数：

如果获取组的所有实例，通过digo.Members(groupId)可以获取到组的所有实例

ctrls, err := digo.Members("main.controllers")
if err == nil {
    for _, controller := range ctrls {
        // TODO:
    }
}