forked from x/icebergs
76 lines
2.5 KiB
Go
76 lines
2.5 KiB
Go
package gdb
|
|
|
|
import (
|
|
"sync"
|
|
|
|
ice "shylinux.com/x/icebergs"
|
|
"shylinux.com/x/icebergs/base/mdb"
|
|
kit "shylinux.com/x/toolkits"
|
|
"shylinux.com/x/toolkits/logs"
|
|
)
|
|
|
|
const (
|
|
LISTEN = "listen"
|
|
HAPPEN = "happen"
|
|
)
|
|
const EVENT = "event"
|
|
|
|
func init() {
|
|
Index.MergeCommands(ice.Commands{
|
|
EVENT: {Name: "event event id auto listen happen", Help: "事件流", Actions: ice.MergeActions(ice.Actions{
|
|
LISTEN: {Name: "listen event* cmd*", Help: "监听", Hand: func(m *ice.Message, arg ...string) {
|
|
mdb.ZoneInsert(m, m.OptionSimple(EVENT, ice.CMD))
|
|
list[m.Option(EVENT)]++
|
|
}},
|
|
HAPPEN: {Name: "happen event*", Help: "触发", Hand: func(m *ice.Message, arg ...string) {
|
|
defer m.Cost()
|
|
m.OptionCB(mdb.SELECT, "")
|
|
mdb.ZoneSelect(m.Spawn(ice.OptionFields("")), arg[1]).Table(func(value ice.Maps) {
|
|
m.Cmdy(kit.Split(value[ice.CMD]), arg[1], arg[2:], ice.OptionFields(""))
|
|
})
|
|
_waitMap.Range(func(key, cb ice.Any) bool { cb.(func(*ice.Message, ...string))(m, arg...); return true })
|
|
}},
|
|
}, mdb.ZoneAction(mdb.SHORT, EVENT, mdb.FIELDS, "time,id,cmd"), mdb.ClearOnExitHashAction())},
|
|
})
|
|
}
|
|
|
|
func EventsAction(arg ...string) ice.Actions {
|
|
list := kit.DictList(arg...)
|
|
return ice.Actions{ice.CTX_INIT: {Hand: func(m *ice.Message, arg ...string) {
|
|
for sub := range m.Target().Commands[m.CommandKey()].Actions {
|
|
kit.If(list[sub] == ice.TRUE, func() { Watch(m, sub) })
|
|
}
|
|
}}}
|
|
}
|
|
|
|
var list map[string]int = map[string]int{}
|
|
|
|
func Watch(m *ice.Message, key string, arg ...string) *ice.Message {
|
|
kit.If(len(arg) == 0, func() { arg = append(arg, m.ShortKey()) })
|
|
return m.Cmd(Prefix(EVENT), LISTEN, EVENT, key, ice.CMD, kit.Join(arg, ice.SP))
|
|
}
|
|
func Event(m *ice.Message, key string, arg ...ice.Any) *ice.Message {
|
|
if key = kit.Select(kit.Keys(m.CommandKey(), m.ActionKey()), key); list[key] == 0 {
|
|
return m
|
|
}
|
|
return m.Cmdy(Prefix(EVENT), HAPPEN, EVENT, key, arg, logs.FileLineMeta(-1))
|
|
}
|
|
func EventDeferEvent(m *ice.Message, key string, arg ...ice.Any) func(string, ...ice.Any) {
|
|
Event(m, key, arg...)
|
|
return func(key string, args ...ice.Any) { Event(m, key, args...) }
|
|
}
|
|
|
|
var _waitMap = sync.Map{}
|
|
|
|
func WaitEvent(m *ice.Message, key string, cb func(*ice.Message, ...string) bool) {
|
|
wg := sync.WaitGroup{}
|
|
h := kit.HashsUniq()
|
|
defer _waitMap.Delete(h)
|
|
_waitMap.Store(h, func(m *ice.Message, arg ...string) {
|
|
m.Info("WaitEvent %v %v", key, kit.FileLine(cb, 3))
|
|
kit.If((key == "" || m.Option(EVENT) == key) && cb(m, arg...), func() { wg.Done() })
|
|
})
|
|
wg.Add(1)
|
|
defer wg.Wait()
|
|
}
|