goioc:一个使用 Go 写的简易的 ioc 框架

简介: goioc 介绍快速上手接口介绍使用 goioc如何使用生命周期实例化获取对象结构体字段依赖注入Dispose 接口反射形式使用 goioc如何使用接口、结构体、结构体指针不会自动注入本身

goioc 介绍

goioc 是一个基于 GO 语言编写的依赖注入框架,基于反射进行编写。

  • 支持泛型;
  • 简单易用的 API;
  • 简易版本的对象生命周期管理,作用域内对象具有生命;
  • 延迟加载,在需要的时候才会实例化对象;
  • 支持结构体字段注入,多层注入;
  • 对象实例化线程安全,作用域内只会被执行一次。

下载依赖:

go get -u github.com/whuanle/goioc v2.0.0

快速上手

定义接口:

type IAnimal interface {
  Println(s string)
}

实现接口:

type Dog struct {
}
func (my Dog) Println(s string) {
  fmt.Println(s)
}

依赖注入以及使用:

// 注册容器
var sc goioc.IServiceCollection = &ServiceCollection{}
// 注入服务
goioc.AddServiceOf[IAnimal, Dog](sc, goioc.Scope)
// 构建提供器
p := sc.Build()
// 获取服务
obj := goioc.Get[IAnimal](p)

接口介绍

IServiceCollection 是一个容器接口,通过此接口,将需要进行依赖注入的对象注册到容器中。

IServiceProvider 是一个服务提供器,当服务注册到容器后,构建一个服务提供器,IServiceProvider 可以管理服务的生命周期以及提供服务。

IDispose 接口用于声明此对象在 IServiceProvider 结束时,需要执行接口释放对象。

// IDispose 释放接口
type IDispose interface {
  // Dispose 释放资源
  Dispose()
}

除此之外,goioc 中还定义了部分扩展函数,如泛型注入等,代码量不多,简单易用。

使用 goioc

如何使用

注入的服务有两种形式,第一种是 B:A,即 B 实现了 A,使用的时候获取 A ;第二种是注入 B,使用的时候获取 B。

// 第一种
AddServiceOf[A,B]()
// 第二种
AddService[B]()

A 可以是接口或结构体,只要 B 实现了 A 即可。

定义一个接口:

type IAnimal interface {
  Println(s string)
}

实现这个接口:

type Dog struct {
  Id int
}
func (my Dog) Println(s string) {
  fmt.Println(s)
}

当使用依赖注入框架时,我们可以将接口和实现分开,甚至放到两个模块中,可以随时替换接口的实现。

注册服务和获取服务的代码示例如下:

func Demo() {
  sc := &ServiceCollection{}
  goioc.AddServiceOf[IAnimal, Dog](sc, goioc.Scope)
  p := sc.Build()
  animal := goioc.GetI[IAnimal](p)
  animal.Println("test")
}

下面讲解编码过程。

首先创建 IServiceCollection 容器,容器中可以注册服务。

sc := &ServiceCollection{}

然后通过接口注入服务:

goioc.AddServiceOf[IAnimal, Dog](sc, goioc.Scope)

这个函数是泛型方法。如果不使用泛型,则注入过程麻烦得多。

注册完毕后,开始构建提供器:

p := sc.Build()

然后获取服务:

animal := goioc.GetI[IAnimal](p)
  animal.Println("test")

生命周期

goioc 中定义了三个生命周期:

const (
  Transient ServiceLifetime = iota
  Scope
  Singleton
)

Transient:瞬时模式,每次获取到的都是新的对象;

Scope:作用域模式,同一个 Provider 中获取到的是同一个对象。

Singleton:单例模式,同一个 ServiceCollection 获取到的是同一个对象,也就是所有 Provider 获取到的都是同一个对象。

如果是单例模式(Singleton),那么无论多少次 Build,对象始终是同一个:

在注册服务的时候,需要注明对象生命周期。

goioc.AddServiceOf[IAnimal, Dog](sc, goioc.Scope)

生命周期为 scope 的注入,同一个 Provider 中,获取到的对象是一样的。

sc := &ServiceCollection{}
  goioc.AddServiceOf[IAnimal, Dog](sc, goioc.Scope)
  p := sc.Build()
    // 第一次获取对象
  animal1 := goioc.GetI[IAnimal](p)
  if animal1 == nil {
    t.Errorf("service is nil!")
  }
  animal1.Println("test")
    // 第二次获取对象
  animal2 := goioc.GetI[IAnimal](p)
  if animal2 == nil {
    t.Errorf("service is nil!")
  }
    // animal1 和 animal2 引用了同一个对象
  if animal1 != animal2 {
    t.Errorf("animal1 != animal2")
  }

实例一,Scope 生命周期的对象,在同一个提供器下获取到的都是同一个对象。

sc := &ServiceCollection{}
    goioc.AddServiceHandlerOf[IAnimal, Dog](sc, goioc.Scope, func(provider goioc.IServiceProvider) interface{} {
    return &Dog{
      Id: 3,
    }
  })
  p := sc.Build()
  // 第一次获取
  a := goioc.GetI[IAnimal](p)
  if v := a.(*Dog); v == nil {
    t.Errorf("service is nil!")
  }
  v := a.(*Dog)
  if v.Id != 2 {
    t.Errorf("Life cycle error")
  }
  v.Id = 3
  // 第二次获取
  aa := goioc.GetI[IAnimal](p)
  v = aa.(*Dog)
  if v.Id != 3 {
    t.Errorf("Life cycle error")
  }
  // 重新构建的 scope,不是同一个对象
  pp := sc.Build()
  aaa := goioc.GetI[IAnimal](pp)
  v = aaa.(*Dog)
  if v.Id != 2 {
    t.Errorf("Life cycle error")
  }

实例二, ServiceCollection 构建的提供器,单例模式下获取到的都是同一个对象。

sc := &ServiceCollection{}
  goioc.AddServiceHandler[Dog](sc, goioc.Singleton, func(provider goioc.IServiceProvider) interface{} {
    return &Dog{
      Id: 2,
    }
  })
  p := sc.Build()
  b := goioc.GetS[Dog](p)
  if b.Id != 2 {
    t.Errorf("Life cycle error")
  }
  b.Id = 3
  bb := goioc.GetS[Dog](p)
  if b.Id != bb.Id {
    t.Errorf("Life cycle error")
  }
  ppp := sc.Build()
  bbb := goioc.GetS[Dog](ppp)
  if b.Id != bbb.Id {
    t.Errorf("Life cycle error")
  }

实例化

由开发者决定如何实例化一个对象。

主要由注册形式决定,有四个泛型函数实现注册服务:

// AddService 注册对象
func AddService[T any](con IServiceCollection, lifetime ServiceLifetime)
// AddServiceHandler 注册对象,并自定义如何初始化实例
func AddServiceHandler[T any](con IServiceCollection, lifetime ServiceLifetime, f func(provider IServiceProvider) interface{})
// AddServiceOf 注册对象,注册接口或父类型及其实现,serviceType 必须实现了 baseType
func AddServiceOf[I any, T any](con IServiceCollection, lifetime ServiceLifetime)
// AddServiceHandlerOf 注册对象,注册接口或父类型及其实现,serviceType 必须实现了 baseType,并自定义如何初始化实例
func AddServiceHandlerOf[I any, T any](con IServiceCollection, lifetime ServiceLifetime, f func(provider IServiceProvider) interface{})

AddService[T any]:只注册可被实例化的对象:

AddService[T any]
goioc.AddService[Dog](sc, goioc.Scope)

AddServiceHandler 注册一个接口或结构体,自定义实例化。

func(provider goioc.IServiceProvider) interface{} 函数会在实例化对象时执行。

goioc.AddServiceHandler[Dog](sc, goioc.Scope, func(provider goioc.IServiceProvider) interface{} {
    return &Dog{
      Id: 1,
    }
  })

在实例化时,如果这个对象还依赖其他服务,则可以通过 goioc.IServiceProvider 来获取其他依赖。

例如下面示例中,一个依赖另一个对象,可以自定义实例化函数,从容器中取出其他依赖对象,然后构建一个新的对象。

goioc.AddServiceHandler[Dog](sc, goioc.Scope, func(provider goioc.IServiceProvider) interface{} {
    a := goioc.GetI[IA](provider)
    return &Dog{
      Id: 1,
            A:  a,
    }
  })
goioc.AddServiceHandler[Dog](sc, goioc.Scope, func(provider goioc.IServiceProvider) interface{} {
    config := goioc.GetI[Config](provider)
    if config.Enable == false
    return &Dog{
      Id: 1,
    }
  })

获取对象

前面提到,我们可以注入 [A,B],或者 [B]

那么获取的时候就有三种函数:

// Get 获取对象
func Get[T any](provider IServiceProvider) interface{} 
// GetI 根据接口获取对象
func GetI[T interface{}](provider IServiceProvider) T 
// GetS 根据结构体获取对象
func GetS[T interface{} | struct{}](provider IServiceProvider) *T

Get[T any] 获取接口或结构体,返回 interface{}

GetI[T interface{}] 获取的是一个接口实例。

GetS[T interface{} | struct{}] 获取的是一个结构体实例。

以上三种方式,返回的都是对象的引用,即指针。

sc := &ServiceCollection{}
  goioc.AddService[Dog](sc, goioc.Scope)
  goioc.AddServiceOf[IAnimal, Dog](sc, goioc.Scope)
  p := sc.Build()
  a := goioc.Get[IAnimal](p)
  b := goioc.Get[Dog](p)
  c := goioc.GetI[IAnimal](p)
  d := goioc.GetS[Dog](p)

结构体字段依赖注入

结构体中的字段,可以自动注入和转换实例。

如结构体 Animal 的定义,其使用了其它结构体,goioc 可以自动注入 Animal 对应字段,要被注入的字段必须是接口或者结构体。

// 结构体中包含了其它对象
type Animal struct {
  Dog IAnimal `ioc:"true"`
}

要对需要自动注入的字段设置 tag 中包含ioc:"true" 才会生效。

示例代码:

sc := &ServiceCollection{}
  goioc.AddServiceHandlerOf[IAnimal, Dog](sc, goioc.Scope, func(provider goioc.IServiceProvider) interface{} {
    return &Dog{
      Id: 666,
    }
  })
  goioc.AddService[Animal](sc, goioc.Scope)
  p := sc.Build()
  a := goioc.GetS[Animal](p)
  if dog := a.Dog.(*Dog); dog.Id != 666 {
    t.Errorf("service is nil!")
  }

goioc 可以自动给你的结构体字段进行自动依赖注入。

注意,goioc 的字段注入转换逻辑是这样的。

如果 obj 要转换为接口,则是使用:

animal := (*obj).(IAnimal)

如果 obj 要转换为结构体,则是:

animal := (*obj).(*Animal)

Dispose 接口

反射形式使用 goioc

如何使用

goioc 的原理是反射,ioc 使用了大量的反射机制实现依赖注入,但是因为 Go 的反射比较难用,导致操作十分麻烦,因此使用泛型包装一层可以降低使用难度。

当然,也可以直接使用原生的反射方式进行依赖注入。

首先反射通过反射获取 reflect.Type

// 获取 reflect.Type
  imy := reflect.TypeOf((*IAnimal)(nil)).Elem()
  my := reflect.TypeOf((*Dog)(nil)).Elem()

依赖注入:

// 创建容器
  sc := &ServiceCollection{}
  // 注入服务,生命周期为 scoped
  sc.AddServiceOf(goioc.Scope, imy, my)
  // 构建服务 Provider
  serviceProvider := sc.Build()

获取服务以及进行类型转换:

// 获取对象
  // *interface{} = &Dog{},因此需要处理指针
  obj, err := serviceProvider.GetService(imy)
  animal := (*obj).(IAnimal)

示例:

imy := reflect.TypeOf((*IAnimal)(nil)).Elem()
  my := reflect.TypeOf((*Dog)(nil)).Elem()
  var sc IServiceCollection = &ServiceCollection{}
  sc.AddServiceOf(goioc.Scope,imy, my)
  p := sc.Build()
  // 获取对象
  // *interface{} = &Dog{},因此需要处理指针
  obj1, _ := p.GetService(imy)
  obj2, _ := p.GetService(imy)
  fmt.Printf("obj1 = %p,obj2 = %p\r\n", (*obj1).(*Dog), (*obj2).(*Dog))
  if fmt.Sprintf("%p",(*obj1).(*Dog)) != fmt.Sprintf("%p",(*obj2).(*Dog)){
    t.Error("两个对象不是同一个")
  }

获取接口和结构体的 reflect.Type:

// 写法 1
    // 接口的 reflect.Type
  var animal IAnimal
    imy := reflect.TypeOf(&animal).Elem()
  my := reflect.TypeOf(Dog{})
// 写法 2
  // 获取 reflect.Type
  imy := reflect.TypeOf((*IAnimal)(nil)).Elem()
  my := reflect.TypeOf((*Dog)(nil)).Elem()

以上两种写法都可以使用,目的在于获取到接口和结构体的 reflect.Type。不过第一种方式会实例化结构体,消耗了一次内存,并且要获取接口的 reflect.Type,是不能直接有用 reflect.TypeOf(animal) 的,需要使用 reflect.TypeOf(&animal).Elem()

然后注入服务,其生命周期为 Scoped:

// 注入服务,生命周期为 scoped
  sc.AddServiceOf(goioc.Scope, imy, my)

当你需要 IAnimal 接口时,会自动注入 Dog 结构体给 IAnimal。

构建依赖注入服务提供器:

// 构建服务 Provider
  serviceProvider := sc.Build()

构建完成后,即可通过 Provider 对象获取需要的实例:

// 获取对象
  // *interface{}
  obj, err := serviceProvider.GetService(imy)
  if err != nil {
    panic(err)
  }
  // 转换为接口
  a := (*obj).(IAnimal)
  //  a := (*obj).(*Dog)

因为使用了依赖注入,我们使用时,只需要使用接口即可,不需要知道具体的实现。

完整的代码示例:

// 获取 reflect.Type
  imy := reflect.TypeOf((*IAnimal)(nil)).Elem()
  my := reflect.TypeOf((*Dog)(nil)).Elem()
  // 创建容器
  sc := &ServiceCollection{}
  // 注入服务,生命周期为 scoped
  sc.AddServiceOf(goioc.Scope, imy, my)
  // 构建服务 Provider
  serviceProvider := sc.Build()
  // 获取对象
  // *interface{} = &Dog{}
  obj, err := serviceProvider.GetService(imy)
  if err != nil {
    panic(err)
  }
  fmt.Println("obj 类型是", reflect.ValueOf(obj).Type())
  // *interface{} = &Dog{},因此需要处理指针
  animal := (*obj).(IAnimal)
  //  a := (*obj).(*Dog)
  animal.Println("测试")

接口、结构体、结构体指针

在结构体注入时,可以对需要的字段进行自动实例化赋值,而字段可能有以下情况:

// 字段是接口
type Animal1 struct {
  Dog IAnimal `ioc:"true"`
}
// 字段是结构体
type Animal2 struct {
  Dog Dog `ioc:"true"`
}
// 字段是结构体指针
type Animal3 struct {
  Dog *Dog `ioc:"true"`
}

首先注入前置的依赖对象:

// 获取 reflect.Type
  imy := reflect.TypeOf((*IAnimal)(nil)).Elem()
  my := reflect.TypeOf((*Dog)(nil)).Elem()
  // 创建容器
    p := &ServiceCollection{}
  // 注入服务,生命周期为 scoped
  p.AddServiceOf(goioc.Scope,imy, my)
    p.AddService(goioc.Scope, my)

然后将我们的一些对象注入进去:

t1 := reflect.TypeOf((*Animal1)(nil)).Elem()
  t2 := reflect.TypeOf((*Animal2)(nil)).Elem()
  t3 := reflect.TypeOf((*Animal3)(nil)).Elem()
  p.Ad(t1)
  p.AddServiceOf(goioc.Scope,t2)
  p.AddServiceOf(goioc.Scope,t3)

然后愉快地获取这些对象实例:

// 构建服务 Provider
  p := collection.Build()
  v1, _ := p.GetService(t1)
  v2, _ := p.GetService(t2)
  v3, _ := p.GetService(t3)
  fmt.Println(*v1)
  fmt.Println(*v2)
  fmt.Println(*v3)

打印对象信息:

&{0x3abdd8}
&{{}}
&{0x3abdd8}

可以看到,当你注入实例后,结构体字段可以是接口、结构体或结构体指针,goioc 会根据不同的情况注入对应的实例。

前面提到了对象是生命周期,这里有些地方需要注意。

如果字段是接口和结构体指针,那么 scope 生命周期时,注入的对象是同一个,可以参考前面的 v1、v3 的 Dog 字段,Dog 字段类型虽然不同,但是因为可以存储指针,因此注入的对象是同一个。如果字段是结构体,由于 Go 语言中结构体是值类型,因此给值类型赋值是,是值赋值,因此对象不是同一个了。

不会自动注入本身

下面是一个依赖注入过程:

// 获取 reflect.Type
  imy := reflect.TypeOf((*IAnimal)(nil)).Elem()
  my := reflect.TypeOf((*Dog)(nil)).Elem()
  // 创建容器
    sc := &ServiceCollection{}
  // 注入服务,生命周期为 scoped
  sc.AddServiceOf(goioc.Scope,imy, my)

此时,注册的服务是 IAnimal,你只能通过 IAnimal 获取实例,无法通过 Dog 获取实例。

如果你想获取 Dog,需要自行注入:

// 注入服务,生命周期为 scoped
  p.AddServiceOf(goioc.Scope,imy, my)
  p.AddService(my)

如果是结构体字段,则使用 IAnimal、Dog、*Dog 的形式都可以。

相关文章
|
2月前
|
JSON Go API
使用Go语言和Gin框架构建RESTful API:GET与POST请求示例
使用Go语言和Gin框架构建RESTful API:GET与POST请求示例
|
3月前
|
Cloud Native JavaScript API
一文读懂云原生 go-zero 微服务框架
一文读懂云原生 go-zero 微服务框架
|
2月前
|
消息中间件 NoSQL Go
PHP转Go系列 | ThinkPHP与Gin框架之Redis延时消息队列技术实践
【9月更文挑战第7天】在从 PHP 的 ThinkPHP 框架迁移到 Go 的 Gin 框架时,涉及 Redis 延时消息队列的技术实践主要包括:理解延时消息队列概念,其能在特定时间处理消息,适用于定时任务等场景;在 ThinkPHP 中使用 Redis 实现延时队列;在 Gin 中结合 Go 的 Redis 客户端库实现类似功能;Go 具有更高性能和简洁性,适合处理大量消息。迁移过程中需考虑业务需求及系统稳定性。
|
3月前
|
Go API 数据库
[go 面试] 分布式事务框架选择与实践
[go 面试] 分布式事务框架选择与实践
|
3月前
|
SQL JavaScript Go
Go Web 服务框架实现详解
Go Web 服务框架实现详解
|
3月前
|
存储 Prometheus 中间件
2020最佳人气项目之Go Web框架
2020最佳人气项目之Go Web框架
|
3月前
|
缓存 监控 中间件
构建高效的Go语言Web服务器:基于Fiber框架的性能优化实践
在追求极致性能的Web开发领域,Go语言(Golang)凭借其高效的并发处理能力、垃圾回收机制及简洁的语法赢得了广泛的青睐。本文不同于传统的性能优化教程,将深入剖析如何在Go语言环境下,利用Fiber这一高性能Web框架,通过精细化配置、并发策略调整及代码层面的微优化,构建出既快速又稳定的Web服务器。通过实际案例与性能测试数据对比,揭示一系列非直觉但极为有效的优化技巧,助力开发者在快节奏的互联网环境中抢占先机。
|
4月前
|
JSON 中间件 Go
Go语言Web框架Gin介绍
【7月更文挑战第19天】Gin是一个功能强大、高性能且易于使用的Go语言Web框架。它提供了路由、中间件、参数绑定等丰富的功能,帮助开发者快速构建高质量的Web应用。通过本文的介绍,你应该对Gin框架有了初步的了解,并能够使用它来开发简单的Web服务。随着你对Gin的深入学习和实践,你将能够利用它构建更复杂、更强大的Web应用。
|
3月前
|
API Go 数据安全/隐私保护
go-zero微服务框架的静态文件服务
【8月更文挑战第7天】`go-zero` 微服务框架支持多种静态文件服务实现方式。常用方法是利用 `Go` 标准库 `http.FileServer`。通过设置静态文件根目录并使用 `http.StripPrefix` 去除路径前缀,能确保 `/static/` 开头的请求正确返回文件。此外,结合 `go-zero` 的路由机制可更灵活地控制静态文件服务,例如仅在特定 API 路径 `/api/static` 下提供服务,从而实现精细化访问控制。
|
3月前
|
NoSQL Go API
简洁、轻量级的 Go API 框架
简洁、轻量级的 Go API 框架
49 0