Go语言工厂模式自动注册——管理多个包的结构体
发布时间:2022-08-29 21:09:38
本例利用包的 init 特性,将 cls1 和 cls2 两个包注册到工厂,使用字符串创建这两个注册好的结构实例。
package base
// 类接口
type Class interface {
Do()
}
var (
// 保存注册好的工厂
factoryByName = make(map[string]func() Class)
)
// 注册一个类生成工厂
func Register(name string, factory func() Class) {
factoryByName[name] = factory
}
// 根据名称创建对应的类
func Create(name string) Class {
if f, ok := factoryByName[name]; ok {
return f()
} else {
panic("name not found")
}
}
package cls1
import (
"base"
"fmt"
)
// 类1
type Class1 struct {
}
// 实现Class接口
func (c *Class1) Do() {
fmt.Println("Class1")
}
func init() {
// 在启动时注册类1工厂
base.Register("Class1", func() base.Class {
return new(Class1)
})
}
package cls2
import (
"base"
"fmt"
)
// 类2
type Class2 struct {
}
// 实现Class接口
func (c *Class2) Do() {
fmt.Println("Class2")
}
func init() {
// 在启动时注册类2工厂
base.Register("Class2", func() base.Class {
return new(Class2)
})
}
package main
import (
"base"
_ "cls1" // 匿名引用cls1包,自动注册
_ "cls2" // 匿名引用cls2包,自动注册
)
func main() {
// 根据字符串动态创建一个Class1实例
c1 := base.Create("Class1")
c1.Do()
// 根据字符串动态创建一个Class2实例
c2 := base.Create("Class2")
c2.Do()
}