forked from panjf2000/gnet
-
Notifications
You must be signed in to change notification settings - Fork 0
/
connection.go
99 lines (85 loc) · 2.16 KB
/
connection.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
// Copyright 2019 Andy Pan. All rights reserved.
// Copyright 2018 Joshua J Baker. 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"
"github.com/panjf2000/gnet/ringbuffer"
"golang.org/x/sys/unix"
)
type conn struct {
fd int // file descriptor
inBuf *ringbuffer.RingBuffer
outBuf *ringbuffer.RingBuffer
sa unix.Sockaddr // remote socket address
opened bool // connection opened event fired
action Action // next user action
ctx interface{} // user-defined context
localAddr net.Addr // local addre
remoteAddr net.Addr // remote addr
loop *loop // connected loop
extra []byte
}
func (c *conn) ReadPair() (top, tail []byte) {
if c.inBuf.IsEmpty() {
top = c.extra
return
}
top, _ = c.inBuf.PreReadAll()
tail = c.extra
return
}
func (c *conn) ReadBytes() []byte {
return c.inBuf.WithBytes(c.extra)
}
func (c *conn) ResetBuffer() {
c.inBuf.Reset()
}
func (c *conn) AsyncWrite(buf []byte) {
_ = c.loop.poller.Trigger(func() error {
c.write(buf)
ringbuffer.Recycle(buf)
return nil
})
}
func (c *conn) open(buf []byte) {
n, err := unix.Write(c.fd, buf)
if err != nil {
_, _ = c.outBuf.Write(buf)
return
}
if n < len(buf) {
_, _ = c.outBuf.Write(buf[n:])
}
}
func (c *conn) write(buf []byte) {
if !c.outBuf.IsEmpty() {
_, _ = c.outBuf.Write(buf)
return
}
n, err := unix.Write(c.fd, buf)
if err != nil {
if err == unix.EAGAIN {
_, _ = c.outBuf.Write(buf)
_ = c.loop.poller.ModReadWrite(c.fd)
return
}
_ = c.loop.loopCloseConn(c, err)
return
}
if n < len(buf) {
_, _ = c.outBuf.Write(buf[n:])
_ = c.loop.poller.ModReadWrite(c.fd)
}
}
func (c *conn) Context() interface{} { return c.ctx }
func (c *conn) SetContext(ctx interface{}) { c.ctx = ctx }
func (c *conn) LocalAddr() net.Addr { return c.localAddr }
func (c *conn) RemoteAddr() net.Addr { return c.remoteAddr }
//func (c *conn) Wake() {
// if c.loop != nil {
// sniffError(c.loop.poller.Trigger(c))
// }
//}