随笔- 542 文章- 6 评论- 27 阅读- 29万

Go从入门到精通——包（package）——示例：工厂模式自动注册——管理多个包的结构体

示例：工厂模式自动注册——管理多个包的结构体

本例利用包的 init 特性，将 cls1 和 cls2 两个包注册到工厂，使用字符串创建这两个注册好的结构实例。

代码1-1 类工厂（具体文件：../clsfactory/base/factory.go）

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!")
    }
}

　　代码1-2 类1及注册代码（具体文件：../clsfactory/cls1/reg.go）

package cls1
 
import (
    "clsfactory/base"
    "fmt"
)
 
//定义类cls1
type Class1 struct {
}
 
//实现 class 接口
func (c *Class1) Do() {
    fmt.Println("Class1 class")
}
 
func init() {
 
    //在启动时注册 类1 工厂
    base.Register("Class1", func() base.Class { return new(Class) })
}　　

代码1-3 类2及注册代码（具体文件：../clsfactory/cls2/reg.go）

package cls2
 
import (
    "clsfactory/base"
    "fmt"
)
 
//定义类cls2
type Class2 struct {
}
 
//实现 Class 接口
func (c *Class2) Do(){
    fmt.Println("Class2 class")
}
 
func init(){
 
    //在启动时注册类2工厂
    base.Register("Class2",func() base.Class{ return new(Class2)})
}

　　代码1-4 类工程主流程（具体文件：../clsfactory/cls2/reg.go）

package main
 
import (
    "clsfactory/base"
    _ "clsfactory/cls1" //匿名引用 cls1 包，自动注册
    _ "clsfactory/cls2" //匿名引用 cls2 包，自动注册
)
 
func main() {
 
    //根据字符串动态创建一个 Class1 实例
    c1 := base.Create("Class1")
    c1.Do()
 
    //根据字符串动态创建一个 Class2 实例
    c2 := base.Create("Class2")
    c2.Do()
 
}