122 lines
2.3 KiB
Plaintext
122 lines
2.3 KiB
Plaintext
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
|
||
}
|