forked from x/icebergs
77 lines
2.3 KiB
Go
77 lines
2.3 KiB
Go
package gdb
|
|
|
|
import (
|
|
"os"
|
|
"os/signal"
|
|
"path"
|
|
"syscall"
|
|
|
|
ice "shylinux.com/x/icebergs"
|
|
"shylinux.com/x/icebergs/base/cli"
|
|
"shylinux.com/x/icebergs/base/mdb"
|
|
"shylinux.com/x/icebergs/base/nfs"
|
|
kit "shylinux.com/x/toolkits"
|
|
log "shylinux.com/x/toolkits/logs"
|
|
)
|
|
|
|
func _signal_listen(m *ice.Message, s int, arg ...string) {
|
|
if f, ok := m.Target().Server().(*Frame); ok {
|
|
m.Cmdy(mdb.INSERT, SIGNAL, "", mdb.HASH, arg)
|
|
signal.Notify(f.s, syscall.Signal(s))
|
|
}
|
|
}
|
|
func _signal_action(m *ice.Message, arg ...string) {
|
|
mdb.HashSelect(m.Spawn(), arg...).Table(func(index int, value map[string]string, head []string) {
|
|
m.Cmdy(kit.Split(value[ice.CMD]))
|
|
})
|
|
}
|
|
|
|
func SignalNotify(m *ice.Message, sig int, cb func()) {
|
|
ch := make(chan os.Signal)
|
|
signal.Notify(ch, syscall.Signal(sig))
|
|
m.Go(func() {
|
|
for {
|
|
if _, ok := <-ch; ok {
|
|
cb()
|
|
}
|
|
}
|
|
})
|
|
}
|
|
|
|
const (
|
|
LISTEN = "listen"
|
|
ACTION = "action"
|
|
)
|
|
const SIGNAL = "signal"
|
|
|
|
func init() {
|
|
Index.Merge(&ice.Context{Configs: map[string]*ice.Config{
|
|
SIGNAL: {Name: SIGNAL, Help: "信号器", Value: kit.Data(
|
|
kit.MDB_SHORT, SIGNAL, kit.MDB_FIELD, "time,signal,name,cmd",
|
|
nfs.PATH, path.Join(ice.VAR_RUN, "ice.pid"),
|
|
)},
|
|
}, Commands: map[string]*ice.Command{
|
|
ice.CTX_INIT: {Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
if log.LogDisable {
|
|
return // 禁用日志
|
|
}
|
|
m.Cmd(nfs.SAVE, kit.Select(m.Conf(SIGNAL, kit.META_PATH), m.Conf(cli.RUNTIME, kit.Keys(cli.CONF, cli.CTX_PID))),
|
|
m.Conf(cli.RUNTIME, kit.Keys(cli.HOST, cli.PID)))
|
|
|
|
m.Cmd(SIGNAL, LISTEN, SIGNAL, "3", kit.MDB_NAME, "退出", ice.CMD, "exit 0")
|
|
m.Cmd(SIGNAL, LISTEN, SIGNAL, "2", kit.MDB_NAME, "重启", ice.CMD, "exit 1")
|
|
}},
|
|
SIGNAL: {Name: "signal signal auto listen", Help: "信号器", Action: ice.MergeAction(map[string]*ice.Action{
|
|
LISTEN: {Name: "listen signal name cmd", Help: "监听", Hand: func(m *ice.Message, arg ...string) {
|
|
_signal_listen(m, kit.Int(m.Option(SIGNAL)), arg...)
|
|
}},
|
|
ACTION: {Name: "action signal", Help: "触发", Hand: func(m *ice.Message, arg ...string) {
|
|
_signal_action(m, m.Option(SIGNAL))
|
|
}},
|
|
}, mdb.HashAction()), Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
mdb.HashSelect(m, arg...).Sort(SIGNAL)
|
|
m.PushAction(ACTION, mdb.REMOVE)
|
|
}},
|
|
}})
|
|
}
|