/
kcp_server.go
56 lines (51 loc) · 1.07 KB
/
kcp_server.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
package lucky
import (
"github.com/helloh2o/lucky/log"
"github.com/xtaci/kcp-go"
"net"
"runtime/debug"
"sync"
)
type kcpServer struct {
mu sync.Mutex
addr string
ln net.Listener
processor Processor
}
// NewKcpServer return a *kcpServer
func NewKcpServer(addr string, processor Processor) (s *kcpServer, err error) {
ts := new(kcpServer)
ts.addr = addr
ts.ln, err = kcp.ListenWithOptions(addr, nil, 10, 3)
if processor == nil {
panic("processor must be set.")
}
ts.processor = processor
if err != nil {
return nil, err
}
return ts, err
}
// Run kcp server
func (s *kcpServer) Run() error {
log.Release("Starting kcp server on %s", s.addr)
for {
conn, err := s.ln.Accept()
if err != nil {
return err
}
go s.Handle(conn)
}
}
// Handle goroutine handle connection
func (s *kcpServer) Handle(conn net.Conn) {
defer func() {
if r := recover(); r != nil {
log.Error("PANIC %v TCP handle, stack %s", r, string(debug.Stack()))
}
}()
var ic IConnection
// 可靠的UDP协议, like tcp
ic = NewKcpConn(conn, s.processor)
ic.ReadMsg()
}