/
acceptor_udp_linux.go
103 lines (86 loc) · 2.18 KB
/
acceptor_udp_linux.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
// +build linux
package server
import (
"github.com/ikilobyte/netman/eventloop"
"github.com/ikilobyte/netman/iface"
"github.com/ikilobyte/netman/util"
"golang.org/x/sys/unix"
"log"
)
type acceptorUdp struct {
packer iface.IPacker
connectMgr iface.IConnectManager
poller *eventloop.Poller
eventfd int
eventbuff []byte
connID int
options *Options
server *Server
}
func newAcceptorUdp(packer iface.IPacker, connectMgr iface.IConnectManager, options *Options, server *Server) iface.IAcceptor {
eventfd, err := unix.Eventfd(0, unix.EPOLL_CLOEXEC)
if err != nil {
log.Panicln(err)
}
poller, err := eventloop.NewPoller(connectMgr)
if err != nil {
log.Panicln(err)
}
return &acceptorUdp{
packer: packer,
connectMgr: connectMgr,
poller: poller,
eventfd: eventfd,
eventbuff: []byte{0, 0, 0, 0, 0, 0, 0, 1},
connID: -1,
options: options,
server: server,
}
}
//Run 启动,只用于接收新的"连接"
// UDP 没有连接的概念,但可以参考TCP,手动创建一个fd,结合epoll,达到多路复用
func (a *acceptorUdp) Run(listenerFD int, loop iface.IEventLoop) error {
// 添加eventfd,用于server退出
if err := a.poller.AddRead(a.eventfd, a.IncrementID()); err != nil {
return err
}
// 添加listener fd
// 虽然udp没有accept的概念,但是可以使用listener的方式创造一个连接
if err := a.poller.AddRead(listenerFD, a.IncrementID()); err != nil {
return err
}
for {
n, err := unix.EpollWait(a.poller.Epfd, a.poller.Events, -1)
if err != nil {
if err == unix.EAGAIN || err == unix.EINTR {
continue
}
return err
}
for i := 0; i < n; i++ {
event := a.poller.Events[i]
fd := int(event.Fd)
// close
if fd == a.eventfd {
_, _ = unix.Read(fd, a.eventbuff)
a.Close()
return nil
}
if _, err := a.makeUdpConnect(fd, loop); err != nil {
util.Logger.Errorln(err)
}
}
}
}
func (a *acceptorUdp) IncrementID() int {
a.connID += 1
return a.connID
}
func (a *acceptorUdp) Close() {
_ = a.poller.Remove(a.eventfd)
_ = unix.Close(a.eventfd)
_ = a.poller.Close()
}
func (a *acceptorUdp) Exit() {
_, _ = unix.Write(a.eventfd, a.eventbuff)
}