redis/go-redis
1
0
Fork 0
mirror of https://github.com/redis/go-redis.git synced 2025-07-29 17:41:15 +03:00
Code Activity

Optimize pool.Remove.

Browse Source
This commit is contained in:
Vladimir Mihailenco
2016-03-12 12:41:02 +02:00
parent ad0739be99
commit fdd0fdf678
13 changed files with 126 additions and 128 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
internal/pool/conn.go
View File

@ -8,10 +8,12 @@ import (
const defaultBufSize = 4096
var noTimeout = time.Time{}
var noDeadline = time.Time{}
type Conn struct {
NetConn net.Conn
idx int
netConn net.Conn
Rd *bufio.Reader
Buf []byte
@ -22,7 +24,9 @@ type Conn struct {
func NewConn(netConn net.Conn) *Conn {
cn := &Conn{
NetConn: netConn,
idx: -1,
netConn: netConn,
Buf: make([]byte, defaultBufSize),
UsedAt: time.Now(),
@ -31,30 +35,35 @@ func NewConn(netConn net.Conn) *Conn {
return cn
}
func (cn *Conn) SetNetConn(netConn net.Conn) {
cn.netConn = netConn
cn.UsedAt = time.Now()
}
func (cn *Conn) Read(b []byte) (int, error) {
cn.UsedAt = time.Now()
if cn.ReadTimeout != 0 {
cn.NetConn.SetReadDeadline(cn.UsedAt.Add(cn.ReadTimeout))
cn.netConn.SetReadDeadline(cn.UsedAt.Add(cn.ReadTimeout))
} else {
cn.NetConn.SetReadDeadline(noTimeout)
cn.netConn.SetReadDeadline(noDeadline)
}
return cn.NetConn.Read(b)
return cn.netConn.Read(b)
}
func (cn *Conn) Write(b []byte) (int, error) {
cn.UsedAt = time.Now()
if cn.WriteTimeout != 0 {
cn.NetConn.SetWriteDeadline(cn.UsedAt.Add(cn.WriteTimeout))
cn.netConn.SetWriteDeadline(cn.UsedAt.Add(cn.WriteTimeout))
} else {
cn.NetConn.SetWriteDeadline(noTimeout)
cn.netConn.SetWriteDeadline(noDeadline)
}
return cn.NetConn.Write(b)
return cn.netConn.Write(b)
}
func (cn *Conn) RemoteAddr() net.Addr {
return cn.NetConn.RemoteAddr()
return cn.netConn.RemoteAddr()
}
func (cn *Conn) Close() error {
return cn.NetConn.Close()
return cn.netConn.Close()
}