54 lines
809 B
Go
54 lines
809 B
Go
package common
|
|
|
|
import "sync"
|
|
|
|
type PauseSub struct {
|
|
ctrl *CtrlWait
|
|
// 回调函数
|
|
items []chan int8
|
|
m sync.RWMutex
|
|
}
|
|
|
|
// Add 添加一个暂停项
|
|
func (p *PauseSub) Add(item chan int8) {
|
|
p.m.Lock()
|
|
defer p.m.Unlock()
|
|
p.items = append(p.items, item)
|
|
}
|
|
|
|
// Remove 移除一个暂停项
|
|
func (p *PauseSub) Remove(item chan int8) {
|
|
p.m.Lock()
|
|
defer p.m.Unlock()
|
|
for i, v := range p.items {
|
|
if v == item {
|
|
p.items = append(p.items[:i], p.items[i+1:]...)
|
|
}
|
|
}
|
|
}
|
|
|
|
// Run 开始监听
|
|
func (p *PauseSub) Run() {
|
|
p.ctrl.Open()
|
|
defer p.ctrl.Close()
|
|
|
|
for {
|
|
select {
|
|
case <-p.ctrl.C:
|
|
go func() {
|
|
p.m.RLock()
|
|
defer p.m.RUnlock()
|
|
for _, item := range p.items {
|
|
item <- 1
|
|
}
|
|
}()
|
|
}
|
|
}
|
|
}
|
|
|
|
func NewPauseSub(c *CtrlWait) *PauseSub {
|
|
return &PauseSub{
|
|
ctrl: c,
|
|
}
|
|
}
|