首页 > 编程笔记 > Go语言笔记

Go语言工厂模式自动注册——管理多个包的结构体

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

完整代码的结构如下:

.
└── src
    └── chapter08
        └── clsfactory
            ├── main.go
            └── base
                └── factory.go
            └── cls1
                └── reg.go
            └── cls2
                └── reg.go

本套教程所有源码下载地址:https://pan.baidu.com/s/1ORFVTOLEYYqDhRzeq0zIiQ    提取密码:hfyf
类工厂(具体文件:…/chapter08/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")
    }
}
这个包叫base,负责处理注册和使用工厂的基础代码,该包不会引用任何外部的包。

以下是对代码的说明:
类1及注册代码(具体文件:…/chapter08/clsfactory/cls1/reg.go)
package cls1

import (
    "chapter08/clsfactory/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)
    })
}
上面的代码展示了Class1的工厂及产品定义过程。
类2及注册代码(具体文件:…/chapter08/clsfactory/cls2/reg.go)
package cls2

import (
    "chapter08/clsfactory/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)
    })
}
Class2 的注册与 Class1 的定义和注册过程类似。

类工程主流程(具体文件:…/chapter08/clsfactory/main.go)
package main

import (
    "chapter08/clsfactory/base"
    _ "chapter08/clsfactory/cls1"  // 匿名引用cls1包, 自动注册
    _ "chapter08/clsfactory/cls2"  // 匿名引用cls2包, 自动注册
)

func main() {

    // 根据字符串动态创建一个Class1实例
    c1 := base.Create("Class1")
    c1.Do()

    // 根据字符串动态创建一个Class2实例
    c2 := base.Create("Class2")
    c2.Do()

}
下面是对代码的说明:
执行下面的指令进行编译:

export GOPATH=/home/davy/golangbook/code
go install chapter08/clsfactory
$GOPATH/bin/clsfactory

代码输出如下:

Class1
Class2

所有教程

优秀文章