icebergs/base/gdb/routine.go

package gdb

import (
	ice "github.com/shylinux/icebergs"
	"github.com/shylinux/icebergs/base/mdb"
	kit "github.com/shylinux/toolkits"

	"strings"
)

const (
	INNER = "inner"
)
const ROUTINE = "routine"

func init() {
	Index.Merge(&ice.Context{
		Configs: map[string]*ice.Config{
			ROUTINE: {Name: ROUTINE, Help: "协程池", Value: kit.Data()},
		},
		Commands: map[string]*ice.Command{
			ROUTINE: {Name: "routine hash auto 清理", Help: "协程池", Action: map[string]*ice.Action{
				mdb.CREATE: {Name: "create fileline status", Help: "创建", Hand: func(m *ice.Message, arg ...string) {
					m.Cmdy(mdb.INSERT, ROUTINE, "", mdb.HASH, arg)
				}},
				mdb.MODIFY: {Name: "modify", Help: "编辑", Hand: func(m *ice.Message, arg ...string) {
					m.Cmdy(mdb.MODIFY, ROUTINE, "", mdb.HASH, kit.MDB_HASH, m.Option(kit.MDB_HASH), arg)
				}},
				mdb.REMOVE: {Name: "remove", Help: "删除", Hand: func(m *ice.Message, arg ...string) {
					m.Cmdy(mdb.DELETE, ROUTINE, "", mdb.HASH, kit.MDB_HASH, m.Option(kit.MDB_HASH))
				}},
				mdb.PRUNES: {Name: "prunes", Help: "清理", Hand: func(m *ice.Message, arg ...string) {
					m.Option(mdb.FIELDS, "time,hash,status,fileline")
					m.Cmdy(mdb.PRUNES, ROUTINE, "", mdb.HASH, kit.MDB_STATUS, STOP)
				}},

				INNER: {Name: "inner", Help: "源码", Hand: func(m *ice.Message, arg ...string) {
					switch kit.Select("", arg, 0) {
					case kit.SSH_RUN:
						m.Cmdy(INNER, arg[1:])
					default:
						ls := kit.Split(m.Option("fileline"), ":")
						switch kit.Split(ls[0], "/")[0] {
						case "usr":
							ls[0] = strings.TrimPrefix(ls[0], "usr/icebergs/")
						case "icebergs":
							ls[0] = strings.TrimPrefix(ls[0], "icebergs/")
						}

						m.PushPlugin(INNER, INNER, kit.SSH_RUN)
						m.Push("args", kit.Format([]string{"usr/icebergs/", ls[0], ls[1]}))
					}
				}},
			}, Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
				m.Option(mdb.FIELDS, kit.Select("time,hash,status,fileline", mdb.DETAIL, len(arg) > 0))
				m.Cmdy(mdb.SELECT, ROUTINE, "", mdb.HASH, kit.MDB_HASH, arg)
				m.PushAction(INNER, mdb.REMOVE)
			}},
		},
	}, nil)
}