mirror
/
ledisdb
mirror of https://github.com/ledisdb/ledisdb.git
1
0
Fork 1
Code Issues Packages Projects Releases Wiki Activity

new conn with size

This commit is contained in:
siddontang 2014-10-23 13:49:25 +08:00
parent e3765035ce
commit 66d5ef4541
1 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
client/go/ledis/conn.go
View File

@ -25,6 +25,9 @@ type Conn struct {
br *bufio.Reader br *bufio.Reader
bw *bufio.Writer bw *bufio.Writer
rSize int
wSize int
lastActive time.Time lastActive time.Time
// Scratch space for formatting argument length. // Scratch space for formatting argument length.
@ -39,6 +42,16 @@ func NewConn(addr string) *Conn {
co := new(Conn) co := new(Conn)
co.addr = addr co.addr = addr
co.rSize = 4096
co.wSize = 4096
return co
}
func NewConnSize(addr string, readSize int, writeSize int) *Conn {
co := NewConn(addr)
co.rSize = readSize
co.wSize = writeSize
return co return co
} }
@ -119,13 +132,13 @@ func (c *Conn) connect() error {
if c.br != nil { if c.br != nil {
c.br.Reset(c.c) c.br.Reset(c.c)
} else { } else {
c.br = bufio.NewReader(c.c) c.br = bufio.NewReaderSize(c.c, c.rSize)
} }
if c.bw != nil { if c.bw != nil {
c.bw.Reset(c.c) c.bw.Reset(c.c)
} else { } else {
c.bw = bufio.NewWriter(c.c) c.bw = bufio.NewWriterSize(c.c, c.wSize)
} }
return nil return nil
@ -371,9 +384,8 @@ func (c *Conn) readReply() (interface{}, error) {
} }
func (c *Client) newConn(addr string) *Conn { func (c *Client) newConn(addr string) *Conn {
co := new(Conn) co := NewConn(addr)
co.client = c co.client = c
co.addr = addr
return co return co
} }