forked from panjf2000/gnet
-
Notifications
You must be signed in to change notification settings - Fork 0
/
connection.go
68 lines (57 loc) · 1.76 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
// 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"
"syscall"
"github.com/panjf2000/gnet/ringbuffer"
"golang.org/x/sys/unix"
)
type conn struct {
fd int // file descriptor
lnidx int // listener index in the server lns list
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
addrIndex int // index of listening address
localAddr net.Addr // local addre
remoteAddr net.Addr // remote addr
loop *loop // connected loop
}
func (c *conn) ReadAll() ([]byte, []byte) {
return c.inBuf.PreReadAll()
}
func (c *conn) AdvanceBuffer(n int) {
c.inBuf.Advance(n)
}
func (c *conn) ResetBuffer() {
c.inBuf.Reset()
}
func (c *conn) write(buf []byte) {
n, err := syscall.Write(c.fd, buf)
if err != nil {
_, _ = c.outBuf.Write(buf)
c.loop.poller.ModReadWrite(c.fd)
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) AddrIndex() int { return c.addrIndex }
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))
}
}