forked from x/icebergs
118 lines
3.3 KiB
Go
118 lines
3.3 KiB
Go
package tcp
|
|
|
|
import (
|
|
ice "github.com/shylinux/icebergs"
|
|
"github.com/shylinux/icebergs/base/mdb"
|
|
kit "github.com/shylinux/toolkits"
|
|
|
|
"net"
|
|
"strings"
|
|
)
|
|
|
|
type Listener struct {
|
|
m *ice.Message
|
|
h string
|
|
s *Stat
|
|
|
|
net.Listener
|
|
}
|
|
|
|
func (l Listener) Accept() (net.Conn, error) {
|
|
c, e := l.Listener.Accept()
|
|
l.s.nc += 1
|
|
l.m.Conf(SERVER, kit.Keys(kit.MDB_HASH, l.h, kit.MDB_META, "nconn"), l.s.nc)
|
|
|
|
ls := strings.Split(c.RemoteAddr().String(), ":")
|
|
if strings.Contains(c.RemoteAddr().String(), "[") {
|
|
ls = strings.Split(strings.TrimPrefix(c.RemoteAddr().String(), "["), "]:")
|
|
}
|
|
h := l.m.Cmdx(mdb.INSERT, CLIENT, "", mdb.HASH, HOST, ls[0], PORT, ls[1],
|
|
kit.MDB_NAME, l.m.Option(kit.MDB_NAME), kit.MDB_STATUS, kit.Select(ERROR, OPEN, e == nil), kit.MDB_ERROR, kit.Format(e))
|
|
|
|
return &Conn{m: l.m, h: h, s: &Stat{}, Conn: c}, e
|
|
}
|
|
func (l Listener) Close() error {
|
|
l.m.Cmd(mdb.MODIFY, SERVER, "", mdb.HASH, kit.MDB_HASH, l.h, kit.MDB_STATUS, CLOSE)
|
|
return l.Listener.Close()
|
|
}
|
|
|
|
const (
|
|
LISTEN_CB = "listen.cb"
|
|
LISTEN = "listen"
|
|
)
|
|
|
|
const SERVER = "server"
|
|
|
|
func init() {
|
|
Index.Merge(&ice.Context{
|
|
Configs: map[string]*ice.Config{
|
|
SERVER: {Name: SERVER, Help: "服务器", Value: kit.Data()},
|
|
},
|
|
Commands: map[string]*ice.Command{
|
|
SERVER: {Name: "server hash auto 清理", Help: "服务器", Action: map[string]*ice.Action{
|
|
LISTEN: {Name: "LISTEN port=9010 host=", Help: "监听", Hand: func(m *ice.Message, arg ...string) {
|
|
l, e := net.Listen(TCP, m.Option(HOST)+":"+m.Option(PORT))
|
|
h := m.Cmdx(mdb.INSERT, SERVER, "", mdb.HASH, PORT, m.Option(PORT), HOST, m.Option(HOST),
|
|
kit.MDB_NAME, m.Option(kit.MDB_NAME), kit.MDB_STATUS, kit.Select(ERROR, OPEN, e == nil), kit.MDB_ERROR, kit.Format(e))
|
|
|
|
l = &Listener{m: m, h: h, s: &Stat{}, Listener: l}
|
|
if e == nil {
|
|
defer l.Close()
|
|
}
|
|
|
|
switch cb := m.Optionv(LISTEN_CB).(type) {
|
|
case func(net.Listener, error):
|
|
cb(l, e)
|
|
case func(net.Listener):
|
|
m.Assert(e)
|
|
cb(l)
|
|
case func(net.Conn):
|
|
for {
|
|
c, e := l.Accept()
|
|
if e != nil {
|
|
break
|
|
}
|
|
cb(c)
|
|
}
|
|
case func(net.Conn, error):
|
|
for {
|
|
c, e := l.Accept()
|
|
if cb(c, e); e != nil {
|
|
break
|
|
}
|
|
}
|
|
default:
|
|
for {
|
|
c, e := l.Accept()
|
|
if e != nil {
|
|
break
|
|
}
|
|
|
|
b := make([]byte, 4096)
|
|
if n, e := c.Read(b); e == nil {
|
|
m.Info("nonce", string(b[:n]))
|
|
c.Write(b[:n])
|
|
}
|
|
c.Close()
|
|
}
|
|
}
|
|
}},
|
|
mdb.DELETE: {Name: "delete", Help: "删除", Hand: func(m *ice.Message, arg ...string) {
|
|
m.Cmdy(mdb.DELETE, SERVER, "", mdb.HASH, kit.MDB_HASH, m.Option(kit.MDB_HASH))
|
|
}},
|
|
mdb.PRUNES: {Name: "prunes", Help: "清理", Hand: func(m *ice.Message, arg ...string) {
|
|
m.Cmdy(mdb.PRUNES, SERVER, "", mdb.HASH, kit.MDB_STATUS, ERROR)
|
|
m.Cmdy(mdb.PRUNES, SERVER, "", mdb.HASH, kit.MDB_STATUS, CLOSE)
|
|
}},
|
|
}, Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
m.Option(mdb.FIELDS, kit.Select("time,hash,status,name,host,port,error,nconn", mdb.DETAIL, len(arg) > 0))
|
|
if m.Cmdy(mdb.SELECT, SERVER, "", mdb.HASH, kit.MDB_HASH, arg); len(arg) == 0 {
|
|
m.Table(func(index int, value map[string]string, head []string) {
|
|
m.PushButton(kit.Select("", "删除", value[kit.MDB_STATUS] == CLOSE))
|
|
})
|
|
}
|
|
}},
|
|
},
|
|
}, nil)
|
|
}
|