icebergs/base/gdb/gdb.go

package gdb

import (
	"fmt"
	"os"
	"os/signal"
	"syscall"
	"time"

	ice "shylinux.com/x/icebergs"
	"shylinux.com/x/icebergs/base/mdb"
	kit "shylinux.com/x/toolkits"
	"shylinux.com/x/toolkits/logs"
)

type Frame struct{ s chan os.Signal }

func (f *Frame) listen(m *ice.Message, s int, arg ...string) {
	signal.Notify(f.s, syscall.Signal(s))
	mdb.HashCreate(m, SIGNAL, s, arg)
}
func (f *Frame) Begin(m *ice.Message, arg ...string) {
	f.s = make(chan os.Signal, 10)
}
func (f *Frame) Start(m *ice.Message, arg ...string) {
	kit.If(ice.Info.PidPath, func(p string) {
		if f, p, e := logs.CreateFile(p); e == nil {
			defer f.Close()
			fmt.Fprint(f, os.Getpid())
			m.Logs("save", PID, p)
		}
	})
	t := kit.Duration(mdb.Conf(m, TIMER, kit.Keym(TICK)))
	enable := mdb.Conf(m, TIMER, kit.Keym("enable")) == ice.TRUE
	for {
		select {
		case <-time.Tick(t):
			if enable {
				m.Cmd(TIMER, HAPPEN)
			}
		case s, ok := <-f.s:
			if !ok {
				return
			}
			m.Cmd(SIGNAL, HAPPEN, SIGNAL, s)
		}
	}
}
func (f *Frame) Close(m *ice.Message, arg ...string) { close(f.s) }

const GDB = "gdb"

var Index = &ice.Context{Name: GDB, Help: "事件模块", Commands: ice.Commands{
	ice.CTX_INIT: {Hand: func(m *ice.Message, arg ...string) { ice.Info.Load(m, TIMER, ROUTINE) }},
	ice.CTX_EXIT: {Hand: func(m *ice.Message, arg ...string) { ice.Info.Save(m, TIMER, ROUTINE) }},
}}

func init() { ice.Index.Register(Index, &Frame{}, SIGNAL, EVENT, TIMER, ROUTINE) }