/
conn.go
59 lines (50 loc) · 1.04 KB
/
conn.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
package server
import (
"errors"
"net"
"runtime"
"sync/atomic"
"github.com/josexy/logx"
"github.com/josexy/mini-ss/util/logger"
)
var errClosed = errors.New("connection closed")
type onceCloseConn struct {
net.Conn
closed uint32
}
func (c *onceCloseConn) Close() error {
if atomic.LoadUint32(&c.closed) != 0 {
return errClosed
}
atomic.StoreUint32(&c.closed, 1)
return c.Conn.Close()
}
type Conn struct {
conn *onceCloseConn
server Server
}
func newConn(conn net.Conn, server Server) Conn {
return Conn{
conn: &onceCloseConn{Conn: conn},
server: server,
}
}
func (c *Conn) close() error {
return c.conn.Close()
}
func (c *Conn) serve() {
defer func() {
// if an error occurs, close the client connection
c.close()
if err := recover(); err != nil {
buf := stackTraceBufferPool.Get()
n := runtime.Stack(*buf, false)
logger.Logger.Error("connection recovery",
logx.Error("err", err.(error)),
logx.String("stackbuf", string((*buf)[:n])),
)
stackTraceBufferPool.Put(buf)
}
}()
c.server.Serve(c)
}