Go 搭建一个Web 服务器(1):IOC工厂

由于go是一门新的语言。已经造好的轮子还很少，再加上本人很是喜欢造轮子。

所以今天就来和大家聊聊用go实现的IOC工厂。

之所以第一个分享的代码就是IOC工厂，是因为我个人认为IOC工厂是解耦的第一步

它基本上可以说是实现其他优雅的模式所需要最最基础的对象。

我的工厂支持 单例，装饰以及普通 对象的实现

下面和大家分享一下代码。

以下代码均可以在下面的链接中看到

《IOCFactory所在地址》

装饰者接口

package Common
type IDecorater interface {
    SetPackage(interface{})
}

普通注册上下文 结构

package Common
import (
    "reflect"
)
type registContext struct {
    instType InstanceType
    bType    reflect.Type
}

装饰者注册上下文 结构

package Common
type decorateRegistcontext struct {
    currentContext *registContext
    nextContext    *decorateRegistcontext
}

工厂的具体实现

package Common
import (
    "fmt"
    "log"
    "reflect"
    "sync"
)
type InstanceType int
var once sync.Once
const (
    InstanceType_Singleton InstanceType = 1 << iota
    InstanceType_Normal
)
type interfaceArrayValue map[string]typeArrayValue
type typeArrayValue interface{}
type IOCFactory struct {
    array map[string]interfaceArrayValue
}
var inst *IOCFactory
func GetIOCFactory() *IOCFactory {
    once.Do(initFactory)
    return inst
}
func initFactory() {
    inst = new(IOCFactory)
    inst.array = make(map[string]interfaceArrayValue)
}
func (this *IOCFactory) Regist(i reflect.Type, t reflect.Type,
    instType InstanceType) error {
    return this.RegistByName("default", i, t, instType)
}
func (this *IOCFactory) RegistByName(key string, i reflect.Type,
    t reflect.Type, instType InstanceType) error {
    if !this.checkIsImplementInterface(i, t) {
        return fmt.Errorf("regist type error")
    }
    var pArray = this.getPArray(i)
    pArray[key] = this.createNormalRegistContext(i, t, instType)
    return nil
}
func (this *IOCFactory) RegistDecorate(i reflect.Type, t reflect.Type,
    instType InstanceType) {
    this.RegistDecorateByName("default", i, t, instType)
}
func (this *IOCFactory) RegistDecorateByName(key string, i reflect.Type,
    t reflect.Type, instType InstanceType) {
    pArray := this.getPArray(i)
    rContext, err := this.getRegistContext(key, i)
    if err != nil {
        log.Printf(err.Error())
        return
    }
    if !this.checkIsImplementInterface(i, t) {
        fmt.Printf("regist type error")
        return
    }
    dContext := new(decorateRegistcontext)
    dContext.currentContext = this.createNormalRegistContext(i, t, instType)
    var cContext *decorateRegistcontext
    if rContext != nil {
        idType := reflect.TypeOf((*IDecorater)(nil)).Elem()
        if !this.checkIsImplementInterface(idType, t) {
            log.Printf("struct not implement interface IDecorater can't regist as a decorater")
            return
        }
        switch rContext.(type) {
        case *registContext:
            cContext = new(decorateRegistcontext)
            cContext.currentContext = rContext.(*registContext)
        case *decorateRegistcontext:
            cContext = rContext.(*decorateRegistcontext)
        }
        dContext.nextContext = cContext
    }
    pArray[key] = dContext
}
func (this *IOCFactory) Get(i reflect.Type) (interface{}, error) {
    return this.GetByName("default", i)
}
func (this *IOCFactory) GetByName(key string, i reflect.Type) (interface{}, error) {
    var returnValue interface{}
    if iContext, err := this.getRegistContext(key, i); err != nil {
        return nil, err
    } else {
        switch iContext.(type) {
        case *registContext:
            regContext := iContext.(*registContext)
            returnValue = this.createNewInst(regContext)
            if regContext.instType == InstanceType_Singleton {
                pArray := this.getPArray(i)
                pArray[key] = returnValue
            }
        case *decorateRegistcontext:
            drContext := iContext.(*decorateRegistcontext)
            returnValue = this.createNewDecorateInst(drContext)
        default:
            returnValue = iContext
        }
    }
    return returnValue, nil
}
func (this *IOCFactory) GetRegistCount(i reflect.Type) int {
    var returnValue int
    if i != nil {
        pArray := this.getPArray(i)
        returnValue = len(pArray)
    } else {
        for _, array := range this.array {
            returnValue += len(array)
        }
    }
    return returnValue
}
func (this *IOCFactory) GetAll(i reflect.Type) map[string]interface{} {
    pArray := this.getPArray(i)
    returnValue := make(map[string]interface{}, len(pArray))
    count := 0
    for key, _ := range pArray {
        if obj, err := this.GetByName(key, i); err == nil {
            returnValue[key] = obj
        } else {
            fmt.Printf("%s\n", err.Error())
        }
        count++
    }
    return returnValue
}
func (this *IOCFactory) getRegistContext(key string, i reflect.Type) (interface{}, error) {
    var pArray = this.getPArray(i)
    if len(pArray) == 0 {
        return nil, fmt.Errorf("interface named \"%s\" not regist any type", i.Name())
    }
    return pArray[key], nil
}
func (this *IOCFactory) getPArray(i reflect.Type) interfaceArrayValue {
    pName := i.Name()
    if this.array[pName] == nil {
        this.array[pName] = make(interfaceArrayValue)
    }
    return this.array[pName]
}
func (this *IOCFactory) createNormalRegistContext(i reflect.Type,
    t reflect.Type, instType InstanceType) *registContext {
    returnValue := new(registContext)
    returnValue.bType = t
    returnValue.instType = instType
    return returnValue
}
func (this *IOCFactory) checkIsImplementInterface(i reflect.Type, instType reflect.Type) bool {
    return instType.Implements(i)
}
func (this *IOCFactory) createNewInst(context *registContext) interface{} {
    returnValue := reflect.New(context.bType.Elem())
    var i interface{} = returnValue.Interface()
    return i
}
func (this *IOCFactory) createNewDecorateInst(context *decorateRegistcontext) interface{} {
    returnValue := this.createNewInst(context.currentContext)
    if returnValue == nil {
        return nil
    }
    if context.nextContext != nil {
        tPackage := this.createNewDecorateInst(context.nextContext)
        id := returnValue.(IDecorater)
        tId := id
        tId.SetPackage(tPackage)
    }
    return returnValue
}

本文出自 “技术源于专注” 博客，转载请与作者联系！