122 lines
2.3 KiB
Go
122 lines
2.3 KiB
Go
|
package initMgr
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"sync"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
. "Framework/startMgr"
|
|||
|
|
"goutil/logUtil"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
var (
|
|||
|
|
funcMap = make(map[string]*FuncItem)
|
|||
|
|
mutex sync.Mutex
|
|||
|
|
operateName = "Init"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// 注册方法(如果名称重复会panic)
|
|||
|
|
// name:方法名称（唯一标识）
|
|||
|
|
// moduleType:模块类型
|
|||
|
|
// definition:方法定义
|
|||
|
|
func Register(name string, moduleType ModuleType, definition func() error) {
|
|||
|
|
mutex.Lock()
|
|||
|
|
defer mutex.Unlock()
|
|||
|
|
|
|||
|
|
if _, exists := funcMap[name]; exists {
|
|||
|
|
panic(fmt.Sprintf("%s已经存在，请重新命名", name))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
funcMap[name] = NewFuncItem(name, moduleType, definition)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 调用一个方法
|
|||
|
|
// name:方法名称（唯一标识）
|
|||
|
|
// 返回值：
|
|||
|
|
// 错误对象
|
|||
|
|
func CallOne(name string) (err error) {
|
|||
|
|
mutex.Lock()
|
|||
|
|
defer mutex.Unlock()
|
|||
|
|
|
|||
|
|
if item, exists := funcMap[name]; !exists {
|
|||
|
|
panic(fmt.Sprintf("%s不存在", name))
|
|||
|
|
} else {
|
|||
|
|
// 调用方法
|
|||
|
|
err = item.Call2(operateName)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 调用任意数量的方法
|
|||
|
|
// nameList:任意数量的方法名称
|
|||
|
|
// 返回值：
|
|||
|
|
// 错误列表
|
|||
|
|
func CallAny(nameList ...string) (errList []error) {
|
|||
|
|
mutex.Lock()
|
|||
|
|
defer mutex.Unlock()
|
|||
|
|
|
|||
|
|
for _, name := range nameList {
|
|||
|
|
if item, exists := funcMap[name]; !exists {
|
|||
|
|
panic(fmt.Sprintf("%s不存在", name))
|
|||
|
|
} else {
|
|||
|
|
// 调用方法
|
|||
|
|
if err := item.Call2(operateName); err != nil {
|
|||
|
|
errList = append(errList, err)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 调用所有方法
|
|||
|
|
// 返回值：
|
|||
|
|
// 错误列表
|
|||
|
|
func CallAll() (errList []error) {
|
|||
|
|
mutex.Lock()
|
|||
|
|
defer mutex.Unlock()
|
|||
|
|
|
|||
|
|
startTime := time.Now()
|
|||
|
|
defer func() {
|
|||
|
|
logUtil.InfoLog(fmt.Sprintf("%s 执行总时间:%s", operateName, time.Since(startTime)))
|
|||
|
|
}()
|
|||
|
|
|
|||
|
|
for _, item := range funcMap {
|
|||
|
|
// 调用方法
|
|||
|
|
if err := item.Call2(operateName); err != nil {
|
|||
|
|
errList = append(errList, err)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 按照模块类型进行调用
|
|||
|
|
// moduleType:模块类型
|
|||
|
|
// 返回值：
|
|||
|
|
// errList:错误列表
|
|||
|
|
func CallType(moduleType ModuleType) (errList []error) {
|
|||
|
|
mutex.Lock()
|
|||
|
|
defer mutex.Unlock()
|
|||
|
|
|
|||
|
|
startTime := time.Now()
|
|||
|
|
defer func() {
|
|||
|
|
logUtil.InfoLog(fmt.Sprintf("%v %s 执行总时间:%s", moduleType, operateName, time.Since(startTime)))
|
|||
|
|
}()
|
|||
|
|
|
|||
|
|
for _, item := range funcMap {
|
|||
|
|
|
|||
|
|
if item.ModuleType != moduleType {
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 调用方法
|
|||
|
|
if err := item.Call2(operateName); err != nil {
|
|||
|
|
errList = append(errList, err)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return
|
|||
|
|
}
|