在 go 框架中自定义依赖注入容器,步骤如下:定义接口和服务结构以实现依赖注入。创建 di 容器结构,其中包含注册依赖项的映射。使用 register 方法注册依赖项接口和其实现。使用 resolve 方法解析注册的依赖项。
在 Go 框架中自定义依赖注入容器
依赖注入 (DI) 容器是一个轻量级库,负责将依赖项注入到类实例中。自定义 DI 容器可提供更高的控制力和灵活性。
步骤:
立即学习“go语言免费学习笔记(深入)”;
-
定义接口:定义服务的接口,以方便依赖注入。
type GreetService interface { Greet(name string) string }
创建服务:实现接口并创建服务结构。
type GreetServiceImpl struct {} func (s *GreetServiceImpl) Greet(name string) string { return fmt.Sprintf("Hello, %s!", name) }
定义注入容器:创建一个结构体来表示 DI 容器。
type Container struct { bindings map[reflect.Type]reflect.Value }
注册依赖项:使用 Register 方法将依赖项接口注册到容器中。
func (c *Container) Register(key interface{}, impl interface{}) { k := reflect.TypeOf(key) i := reflect.ValueOf(impl) c.bindings[k] = i }
解析依赖项:使用 Resolve 方法解析注册的依赖项。
func (c *Container) Resolve(key interface{}) interface{} { k := reflect.TypeOf(key) return c.bindings[k].Interface() }
实战案例:
package main
import (
"fmt"
"reflect"
)
type GreetService interface {
Greet(name string) string
}
type GreetServiceImpl struct{}
func (s *GreetServiceImpl) Greet(name string) string {
return fmt.Sprintf("Hello, %s!", name)
}
type Container struct {
bindings map[reflect.Type]reflect.Value
}
func (c *Container) Register(key interface{}, impl interface{}) {
k := reflect.TypeOf(key)
i := reflect.ValueOf(impl)
c.bindings[k] = i
}
func (c *Container) Resolve(key interface{}) interface{} {
k := reflect.TypeOf(key)
return c.bindings[k].Interface()
}
func main() {
c := Container{}
c.Register(new(GreetService), &GreetServiceImpl{})
greetService := c.Resolve(new(GreetService)).(GreetService)
fmt.Println(greetService.Greet("John"))
}