/
TCPServer.go
119 lines (107 loc) · 2.36 KB
/
TCPServer.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
package TCPServer
import (
"errors"
"github.com/kolonse/function"
"github.com/kolonse/logs"
"net"
"reflect"
)
type TCPServer struct {
Addr string
logger *logs.BeeLogger
newConnCB *function.Function
recvCB *function.Function
writeCB *function.Function
listener net.Listener
exit chan bool
}
func (ts *TCPServer) Register(opt ...interface{}) error {
if len(opt) == 0 {
return errors.New("参数为空")
}
if reflect.TypeOf(opt[0]).Kind() != reflect.String {
// 如果参数不正确 那么相当于设置失败
return errors.New("第一个参数必须为字符串 标识要设置的目标")
}
switch opt[0] {
case "newConnCB":
if len(opt) > 1 {
ts.newConnCB = opt[1].(*function.Function)
}
case "recvCB":
if len(opt) > 1 {
ts.recvCB = opt[1].(*function.Function)
}
case "writeCB":
if len(opt) > 1 {
ts.writeCB = opt[1].(*function.Function)
}
case "logger":
if len(opt) > 1 {
ts.logger = opt[1].(*logs.BeeLogger)
}
}
return nil
}
// TCP 服务开始接口
func (ts *TCPServer) Server() error {
// 启动 TCP 服务
listener, err := net.Listen("tcp", ts.Addr)
if err != nil {
ts.logger.Error("server addr:%s start fail,err:%v", ts.Addr, err.Error())
return err
}
ts.logger.Info("server addr:%s start success!", ts.Addr)
ts.listener = listener
go func() {
for {
//等待客户端接入
conn, err := listener.Accept()
if nil != err {
ts.logger.Warn("server addr:%s server over,err:%v", ts.Addr, err.Error())
break
}
tcpConn := NewTCPConn(conn, ts.recvCB, ts.writeCB)
if ts.newConnCB != nil {
ts.newConnCB.Call(tcpConn)
}
go tcpConn.Server()
}
ts.exit <- true
}()
return nil
}
func (ts *TCPServer) Stop() {
if ts.listener != nil {
ts.listener.Close()
}
}
// 停止函数 并带有一个回调通知退出完成
func (ts *TCPServer) StopFunc(cb func()) {
if ts.listener != nil {
go func() {
<-ts.exit
cb()
}()
ts.listener.Close()
}
}
// 默认的TCP 服务
var DefaultTCPServer *TCPServer
func NewTCPServer(addr string) *TCPServer {
logger := logs.NewLogger(10000)
err := logger.SetLogger("console", "")
if err != nil {
panic(err.Error())
}
logger.EnableFuncCallDepth(true)
logger.SetLogFuncCallDepth(3)
return &TCPServer{
Addr: addr,
logger: logger,
exit: make(chan bool, 1),
}
}
func init() {
DefaultTCPServer = NewTCPServer("0.0.0.0:9999")
}