forked from panjf2000/gnet
-
Notifications
You must be signed in to change notification settings - Fork 0
/
acceptor.go
79 lines (72 loc) · 1.54 KB
/
acceptor.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
// Copyright 2019 Andy Pan. All rights reserved.
// Use of this source code is governed by an MIT-style
// license that can be found in the LICENSE file.
// +build darwin netbsd freebsd openbsd dragonfly linux
package gnet
import (
"net"
"os"
"golang.org/x/sys/unix"
)
func (svr *server) acceptNewConnection(fd int) error {
nfd, sa, err := unix.Accept(fd)
if err != nil {
if err == unix.EAGAIN {
return nil
}
return err
}
if err := unix.SetNonblock(nfd, true); err != nil {
return err
}
lp := svr.subLoopGroup.next()
c := svr.connPool.Get().(*conn)
c.fd = nfd
c.loop = lp
c.sa = sa
_ = lp.loopOpen(c)
_ = lp.poller.Trigger(func() (err error) {
if err = lp.poller.AddRead(nfd); err == nil {
lp.connections[nfd] = c
return
}
return
})
return nil
}
func (ln *listener) close() {
if ln.f != nil {
sniffError(ln.f.Close())
}
if ln.ln != nil {
sniffError(ln.ln.Close())
}
if ln.pconn != nil {
sniffError(ln.pconn.Close())
}
if ln.network == "unix" {
sniffError(os.RemoveAll(ln.addr))
}
}
// system takes the net listener and detaches it from it's parent
// event loop, grabs the file descriptor, and makes it non-blocking.
func (ln *listener) system() error {
var err error
switch netln := ln.ln.(type) {
case nil:
switch pconn := ln.pconn.(type) {
case *net.UDPConn:
ln.f, err = pconn.File()
}
case *net.TCPListener:
ln.f, err = netln.File()
case *net.UnixListener:
ln.f, err = netln.File()
}
if err != nil {
ln.close()
return err
}
ln.fd = int(ln.f.Fd())
return unix.SetNonblock(ln.fd, true)
}