forked from x/icebergs
91 lines
2.5 KiB
Go
91 lines
2.5 KiB
Go
package tcp
|
|
|
|
import (
|
|
"net"
|
|
"path"
|
|
|
|
ice "shylinux.com/x/icebergs"
|
|
"shylinux.com/x/icebergs/base/aaa"
|
|
"shylinux.com/x/icebergs/base/cli"
|
|
"shylinux.com/x/icebergs/base/mdb"
|
|
"shylinux.com/x/icebergs/base/nfs"
|
|
kit "shylinux.com/x/toolkits"
|
|
)
|
|
|
|
func _port_right(m *ice.Message, arg ...string) string {
|
|
current := kit.Int(kit.Select(m.Config(CURRENT), arg, 0))
|
|
end := kit.Int(m.Config(END))
|
|
if current >= end {
|
|
current = kit.Int(m.Config(BEGIN))
|
|
}
|
|
|
|
for i := current; i < end; i++ {
|
|
if c, e := net.Dial(TCP, kit.Format(":%d", i)); e == nil {
|
|
m.Info("port exists %v", i)
|
|
c.Close()
|
|
continue
|
|
}
|
|
p := path.Join(m.Conf(cli.DAEMON, kit.Keym(nfs.PATH)), kit.Format(i))
|
|
if kit.FileExists(p) {
|
|
continue
|
|
}
|
|
nfs.MkdirAll(m, p)
|
|
|
|
m.Log_SELECT(PORT, i)
|
|
return m.Config(CURRENT, i)
|
|
}
|
|
return ""
|
|
}
|
|
|
|
const (
|
|
RANDOM = "random"
|
|
CURRENT = "current"
|
|
BEGIN = "begin"
|
|
END = "end"
|
|
)
|
|
const PORT = "port"
|
|
|
|
func init() {
|
|
Index.Merge(&ice.Context{Configs: map[string]*ice.Config{
|
|
PORT: {Name: PORT, Help: "端口", Value: kit.Data(BEGIN, 10000, CURRENT, 10000, END, 20000)},
|
|
}, Commands: map[string]*ice.Command{
|
|
PORT: {Name: "port port path auto", Help: "端口", Action: map[string]*ice.Action{
|
|
aaa.RIGHT: {Name: "right", Help: "分配", Hand: func(m *ice.Message, arg ...string) {
|
|
m.Echo(_port_right(m, arg...))
|
|
}},
|
|
nfs.TRASH: {Name: "trash", Help: "删除", Hand: func(m *ice.Message, arg ...string) {
|
|
if m.Option(PORT) != "" {
|
|
m.Cmd(nfs.TRASH, path.Join(ice.USR_LOCAL_DAEMON, m.Option(PORT)))
|
|
}
|
|
}},
|
|
}, Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
if len(arg) == 0 {
|
|
current := kit.Int(m.Config(BEGIN))
|
|
m.Option(nfs.DIR_ROOT, ice.USR_LOCAL_DAEMON)
|
|
m.Cmd(nfs.DIR, nfs.PWD, nfs.DIR_CLI_FIELDS).Tables(func(value map[string]string) {
|
|
bin := m.Cmd(nfs.DIR, path.Join(value[nfs.PATH], ice.BIN), nfs.DIR_CLI_FIELDS).Append(nfs.PATH)
|
|
if bin == "" {
|
|
bin = m.Cmd(nfs.DIR, path.Join(value[nfs.PATH], "sbin"), nfs.DIR_CLI_FIELDS).Append(nfs.PATH)
|
|
}
|
|
port := kit.Int(path.Base(value[nfs.PATH]))
|
|
if port > current {
|
|
current = port
|
|
}
|
|
|
|
m.Push(mdb.TIME, value[mdb.TIME])
|
|
m.Push(PORT, port)
|
|
m.Push(nfs.SIZE, value[nfs.SIZE])
|
|
m.Push(ice.BIN, bin)
|
|
})
|
|
m.SortInt(PORT)
|
|
m.PushAction(nfs.TRASH)
|
|
m.Config(CURRENT, current)
|
|
m.StatusTimeCount(m.ConfigSimple(BEGIN, CURRENT, END))
|
|
return
|
|
}
|
|
m.Option(nfs.DIR_ROOT, path.Join(m.Conf(cli.DAEMON, kit.Keym(nfs.PATH)), arg[0]))
|
|
m.Cmdy(nfs.DIR, arg[1:])
|
|
}},
|
|
}})
|
|
}
|