ing
This commit is contained in:
crusader
parent
f653cbf551
commit
28723e5290
188
socket.go
188
socket.go
|
|
@ -14,8 +14,8 @@ type Socket interface {
|
|||
// ID returns the identity of the client.
|
||||
ID() string
|
||||
|
||||
// WaitRequest wait request of client.
|
||||
WaitRequest() (*SocketConn, error)
|
||||
// // WaitRequest wait request of client.
|
||||
// WaitRequest() (*SocketConn, error)
|
||||
|
||||
// Subprotocol returns the negotiated protocol for the connection.
|
||||
Subprotocol() string
|
||||
|
|
@ -161,7 +161,7 @@ func newSocket(socketHandler SocketHandler, socketCTX SocketContext, conn *webso
|
|||
return s
|
||||
}
|
||||
|
||||
type fasthttpSocket struct {
|
||||
type fasthttpWebSocket struct {
|
||||
*websocket.Conn
|
||||
|
||||
ctx SocketContext
|
||||
|
|
@ -169,40 +169,40 @@ type fasthttpSocket struct {
|
|||
|
||||
id string
|
||||
|
||||
sc *SocketConn
|
||||
// sc *SocketConn
|
||||
}
|
||||
|
||||
func (s *fasthttpSocket) Context() SocketContext {
|
||||
func (s *fasthttpWebSocket) Context() SocketContext {
|
||||
return s.ctx
|
||||
}
|
||||
|
||||
func (s *fasthttpSocket) ID() string {
|
||||
func (s *fasthttpWebSocket) ID() string {
|
||||
return s.id
|
||||
}
|
||||
|
||||
func (s *fasthttpSocket) WaitRequest() (*SocketConn, error) {
|
||||
if nil != s.sc {
|
||||
releaseSocketConn(s.sc)
|
||||
s.sc = nil
|
||||
}
|
||||
// func (s *fasthttpWebSocket) WaitRequest() (*SocketConn, error) {
|
||||
// if nil != s.sc {
|
||||
// releaseSocketConn(s.sc)
|
||||
// s.sc = nil
|
||||
// }
|
||||
|
||||
var mt int
|
||||
var err error
|
||||
var r io.Reader
|
||||
// var mt int
|
||||
// var err error
|
||||
// var r io.Reader
|
||||
|
||||
if mt, r, err = s.NextReader(); nil != err {
|
||||
return nil, err
|
||||
}
|
||||
// if mt, r, err = s.NextReader(); nil != err {
|
||||
// return nil, err
|
||||
// }
|
||||
|
||||
s.sc = retainSocketConn()
|
||||
s.sc.s = s
|
||||
s.sc.MessageType = mt
|
||||
s.sc.r = r
|
||||
// s.sc = retainSocketConn()
|
||||
// s.sc.s = s
|
||||
// s.sc.MessageType = mt
|
||||
// s.sc.r = r
|
||||
|
||||
return s.sc, nil
|
||||
}
|
||||
// return s.sc, nil
|
||||
// }
|
||||
|
||||
func (s *fasthttpSocket) NextWriter(messageType int) (io.WriteCloser, error) {
|
||||
func (s *fasthttpWebSocket) NextWriter(messageType int) (io.WriteCloser, error) {
|
||||
if 0 < s.sh.GetWriteTimeout() {
|
||||
s.SetWriteDeadline(time.Now().Add(s.sh.GetWriteTimeout() * time.Second))
|
||||
}
|
||||
|
|
@ -210,7 +210,7 @@ func (s *fasthttpSocket) NextWriter(messageType int) (io.WriteCloser, error) {
|
|||
return s.Conn.NextWriter(messageType)
|
||||
}
|
||||
|
||||
func (s *fasthttpSocket) WriteMessage(messageType int, data []byte) error {
|
||||
func (s *fasthttpWebSocket) WriteMessage(messageType int, data []byte) error {
|
||||
if 0 < s.sh.GetWriteTimeout() {
|
||||
s.SetWriteDeadline(time.Now().Add(s.sh.GetWriteTimeout() * time.Second))
|
||||
}
|
||||
|
|
@ -218,100 +218,100 @@ func (s *fasthttpSocket) WriteMessage(messageType int, data []byte) error {
|
|||
return s.Conn.WriteMessage(messageType, data)
|
||||
}
|
||||
|
||||
func (s *fasthttpSocket) Close() error {
|
||||
func (s *fasthttpWebSocket) Close() error {
|
||||
err := s.Conn.Close()
|
||||
releaseSocket(s)
|
||||
return err
|
||||
}
|
||||
|
||||
type SocketConn struct {
|
||||
net.Conn
|
||||
// type SocketConn struct {
|
||||
// net.Conn
|
||||
|
||||
s *fasthttpSocket
|
||||
// s *fasthttpWebSocket
|
||||
|
||||
MessageType int
|
||||
r io.Reader
|
||||
wc io.WriteCloser
|
||||
}
|
||||
// MessageType int
|
||||
// r io.Reader
|
||||
// wc io.WriteCloser
|
||||
// }
|
||||
|
||||
func (sc *SocketConn) Read(b []byte) (n int, err error) {
|
||||
return sc.r.Read(b)
|
||||
}
|
||||
// func (sc *SocketConn) Read(b []byte) (n int, err error) {
|
||||
// return sc.r.Read(b)
|
||||
// }
|
||||
|
||||
func (sc *SocketConn) Write(b []byte) (n int, err error) {
|
||||
if nil == sc.wc {
|
||||
var err error
|
||||
if sc.wc, err = sc.s.NextWriter(sc.MessageType); nil != err {
|
||||
return 0, err
|
||||
}
|
||||
}
|
||||
return sc.wc.Write(b)
|
||||
}
|
||||
// func (sc *SocketConn) Write(b []byte) (n int, err error) {
|
||||
// if nil == sc.wc {
|
||||
// var err error
|
||||
// if sc.wc, err = sc.s.NextWriter(sc.MessageType); nil != err {
|
||||
// return 0, err
|
||||
// }
|
||||
// }
|
||||
// return sc.wc.Write(b)
|
||||
// }
|
||||
|
||||
func (sc *SocketConn) Close() error {
|
||||
var err error
|
||||
if sc.wc != nil {
|
||||
err = sc.wc.Close()
|
||||
}
|
||||
sc.s.sc = nil
|
||||
releaseSocketConn(sc)
|
||||
return err
|
||||
}
|
||||
// func (sc *SocketConn) Close() error {
|
||||
// var err error
|
||||
// if sc.wc != nil {
|
||||
// err = sc.wc.Close()
|
||||
// }
|
||||
// sc.s.sc = nil
|
||||
// releaseSocketConn(sc)
|
||||
// return err
|
||||
// }
|
||||
|
||||
func (sc *SocketConn) LocalAddr() net.Addr {
|
||||
return sc.s.LocalAddr()
|
||||
}
|
||||
func (sc *SocketConn) RemoteAddr() net.Addr {
|
||||
return sc.s.RemoteAddr()
|
||||
}
|
||||
func (sc *SocketConn) SetDeadline(t time.Time) error {
|
||||
if err := sc.s.SetReadDeadline(t); nil != err {
|
||||
return err
|
||||
}
|
||||
if err := sc.s.SetWriteDeadline(t); nil != err {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
func (sc *SocketConn) SetReadDeadline(t time.Time) error {
|
||||
return sc.s.SetReadDeadline(t)
|
||||
}
|
||||
func (sc *SocketConn) SetWriteDeadline(t time.Time) error {
|
||||
return sc.s.SetWriteDeadline(t)
|
||||
}
|
||||
// func (sc *SocketConn) LocalAddr() net.Addr {
|
||||
// return sc.s.LocalAddr()
|
||||
// }
|
||||
// func (sc *SocketConn) RemoteAddr() net.Addr {
|
||||
// return sc.s.RemoteAddr()
|
||||
// }
|
||||
// func (sc *SocketConn) SetDeadline(t time.Time) error {
|
||||
// if err := sc.s.SetReadDeadline(t); nil != err {
|
||||
// return err
|
||||
// }
|
||||
// if err := sc.s.SetWriteDeadline(t); nil != err {
|
||||
// return err
|
||||
// }
|
||||
// return nil
|
||||
// }
|
||||
// func (sc *SocketConn) SetReadDeadline(t time.Time) error {
|
||||
// return sc.s.SetReadDeadline(t)
|
||||
// }
|
||||
// func (sc *SocketConn) SetWriteDeadline(t time.Time) error {
|
||||
// return sc.s.SetWriteDeadline(t)
|
||||
// }
|
||||
|
||||
var socketPool sync.Pool
|
||||
|
||||
func retainSocket() *fasthttpSocket {
|
||||
func retainSocket() *fasthttpWebSocket {
|
||||
v := socketPool.Get()
|
||||
if v == nil {
|
||||
return &fasthttpSocket{}
|
||||
return &fasthttpWebSocket{}
|
||||
}
|
||||
return v.(*fasthttpSocket)
|
||||
return v.(*fasthttpWebSocket)
|
||||
}
|
||||
|
||||
func releaseSocket(s *fasthttpSocket) {
|
||||
func releaseSocket(s *fasthttpWebSocket) {
|
||||
s.sh = nil
|
||||
s.sc = nil
|
||||
s.ctx = nil
|
||||
s.id = ""
|
||||
|
||||
socketPool.Put(s)
|
||||
}
|
||||
|
||||
var socketConnPool sync.Pool
|
||||
// var socketConnPool sync.Pool
|
||||
|
||||
func retainSocketConn() *SocketConn {
|
||||
v := socketConnPool.Get()
|
||||
if v == nil {
|
||||
return &SocketConn{}
|
||||
}
|
||||
return v.(*SocketConn)
|
||||
}
|
||||
// func retainSocketConn() *SocketConn {
|
||||
// v := socketConnPool.Get()
|
||||
// if v == nil {
|
||||
// return &SocketConn{}
|
||||
// }
|
||||
// return v.(*SocketConn)
|
||||
// }
|
||||
|
||||
func releaseSocketConn(sc *SocketConn) {
|
||||
sc.s = nil
|
||||
sc.r = nil
|
||||
sc.wc = nil
|
||||
// func releaseSocketConn(sc *SocketConn) {
|
||||
// sc.s = nil
|
||||
// sc.r = nil
|
||||
// sc.wc = nil
|
||||
|
||||
socketConnPool.Put(sc)
|
||||
}
|
||||
// socketConnPool.Put(sc)
|
||||
// }
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user