mirror of
https://shylinux.com/x/icebergs
synced 2025-04-26 09:34:05 +08:00
81 lines
2.7 KiB
Go
81 lines
2.7 KiB
Go
package code
|
|
|
|
import (
|
|
"github.com/shylinux/icebergs"
|
|
"github.com/shylinux/toolkits"
|
|
|
|
"net/http"
|
|
_ "net/http/pprof"
|
|
"strings"
|
|
)
|
|
|
|
const (
|
|
PPROF = "pprof"
|
|
)
|
|
|
|
func init() {
|
|
Index.Merge(&ice.Context{
|
|
Configs: map[string]*ice.Config{
|
|
"pprof": {Name: "pprof", Help: "性能分析", Value: kit.Data(kit.MDB_SHORT, kit.MDB_NAME,
|
|
"stop", "ps aux|grep pprof|grep -v grep|cut -d' ' -f2|xargs -n1 kill",
|
|
)},
|
|
},
|
|
Commands: map[string]*ice.Command{
|
|
"pprof": {Name: "pprof run name time", Help: "性能分析", Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
if m.Show(cmd, arg...) {
|
|
return
|
|
}
|
|
|
|
switch arg[0] {
|
|
case "run":
|
|
m.Richs(cmd, nil, arg[1], func(key string, value map[string]interface{}) {
|
|
m.Gos(m.Spawn(), func(msg *ice.Message) {
|
|
m.Sleep("1s").Grows(cmd, kit.Keys(kit.MDB_HASH, key), "", "", func(index int, value map[string]interface{}) {
|
|
// 压测命令
|
|
m.Cmd(ice.WEB_FAVOR, "pprof", "shell", value[kit.MDB_TEXT], m.Cmdx(kit.Split(kit.Format(value[kit.MDB_TEXT]))))
|
|
})
|
|
})
|
|
|
|
// 启动监控
|
|
name := arg[1] + ".pd.gz"
|
|
value = value["meta"].(map[string]interface{})
|
|
msg := m.Cmd(ice.WEB_SPIDE, "self", "cache", "GET", kit.Select("/code/pprof/profile", value["remote"]), "seconds", kit.Select("5", arg, 2))
|
|
m.Cmd(ice.WEB_FAVOR, "pprof", "shell", "text", m.Cmdx(ice.CLI_SYSTEM, "go", "tool", "pprof", "-text", msg.Append("text")))
|
|
m.Cmd(ice.WEB_FAVOR, "pprof", "pprof", name, msg.Append("data"))
|
|
|
|
arg = kit.Simple("web", value[kit.MDB_TEXT], msg.Append("text"))
|
|
})
|
|
|
|
fallthrough
|
|
case "web":
|
|
// 展示结果
|
|
p := kit.Format("%s:%s", m.Conf(ice.WEB_SHARE, "meta.host"), m.Cmdx("tcp.getport"))
|
|
m.Cmd(ice.CLI_DAEMON, "go", "tool", "pprof", "-http="+p, arg[1:])
|
|
m.Cmd(ice.WEB_FAVOR, "pprof", "bin", arg[1], m.Cmd(ice.WEB_CACHE, "catch", "bin", arg[1]).Append("data"))
|
|
m.Cmd(ice.WEB_FAVOR, "pprof", "spide", arg[2], "http://"+p)
|
|
m.Echo(p)
|
|
|
|
case "stop":
|
|
m.Cmd(ice.CLI_SYSTEM, "sh", "-c", m.Conf(cmd, "meta.stop"))
|
|
|
|
case "add":
|
|
key := m.Rich(cmd, nil, kit.Data(
|
|
kit.MDB_NAME, arg[1], kit.MDB_TEXT, arg[2], "remote", arg[3],
|
|
))
|
|
|
|
for i := 4; i < len(arg)-1; i += 2 {
|
|
m.Grow(cmd, kit.Keys(kit.MDB_HASH, key), kit.Dict(
|
|
kit.MDB_NAME, arg[i], kit.MDB_TEXT, arg[i+1],
|
|
))
|
|
}
|
|
}
|
|
}},
|
|
"/pprof/": {Name: "/pprof/", Help: "性能分析", Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
m.R.URL.Path = strings.Replace("/code"+m.R.URL.Path, "code", "debug", 1)
|
|
http.DefaultServeMux.ServeHTTP(m.W, m.R)
|
|
m.Render(ice.RENDER_VOID)
|
|
}},
|
|
},
|
|
}, nil)
|
|
}
|