63 lines
1.3 KiB
Plaintext
63 lines
1.3 KiB
Plaintext
|
package initMgr
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"sync"
|
|||
|
|
|
|||
|
|
"goutil/debugUtil"
|
|||
|
|
"goutil/logUtil"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// 初始化成功对象
|
|||
|
|
type InitSuccess struct {
|
|||
|
|
name string
|
|||
|
|
registerMap map[string]chan bool
|
|||
|
|
mutex sync.RWMutex
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 注册需要被通知的对象
|
|||
|
|
// name:唯一标识
|
|||
|
|
// ch:用于通知的通道
|
|||
|
|
func (this *InitSuccess) Register(name string, ch chan bool) {
|
|||
|
|
this.mutex.Lock()
|
|||
|
|
defer this.mutex.Unlock()
|
|||
|
|
|
|||
|
|
if _, exists := this.registerMap[name]; exists {
|
|||
|
|
panic(fmt.Errorf("InitSuccess.Register-%s-%s已经存在，请检查", this.name, name))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
this.registerMap[name] = ch
|
|||
|
|
|
|||
|
|
logUtil.DebugLog(fmt.Sprintf("InitSuccess.Register-%s，当前共有%d个注册", this.name, len(this.registerMap)))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 取消启动成功通知注册
|
|||
|
|
// name:唯一标识
|
|||
|
|
func (this *InitSuccess) Unregister(name string) {
|
|||
|
|
this.mutex.Lock()
|
|||
|
|
defer this.mutex.Unlock()
|
|||
|
|
|
|||
|
|
delete(this.registerMap, name)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 通知所有已注册的对象
|
|||
|
|
func (this *InitSuccess) Notify() {
|
|||
|
|
this.mutex.RLock()
|
|||
|
|
defer this.mutex.RUnlock()
|
|||
|
|
|
|||
|
|
for name, ch := range this.registerMap {
|
|||
|
|
ch <- true
|
|||
|
|
msg := fmt.Sprintf("通知:%s-%s初始化成功", this.name, name)
|
|||
|
|
debugUtil.Println(msg)
|
|||
|
|
logUtil.DebugLog(msg)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 创建初始化成功对象
|
|||
|
|
func NewInitSuccess(name string) *InitSuccess {
|
|||
|
|
return &InitSuccess{
|
|||
|
|
name: name,
|
|||
|
|
registerMap: make(map[string]chan bool),
|
|||
|
|
}
|
|||
|
|
}
|