package ledis import ( "container/list" "strings" "sync" "time" ) const ( pingPeriod time.Duration = 3 * time.Second ) type Config struct { Addr string MaxIdleConns int ReadBufferSize int WriteBufferSize int } type Client struct { sync.Mutex cfg *Config conns *list.List } func getProto(addr string) string { if strings.Contains(addr, "/") { return "unix" } else { return "tcp" } } func NewClient(cfg *Config) *Client { c := new(Client) c.cfg = cfg if c.cfg.ReadBufferSize == 0 { c.cfg.ReadBufferSize = 4096 } if c.cfg.WriteBufferSize == 0 { c.cfg.WriteBufferSize = 4096 } c.conns = list.New() return c } func (c *Client) Do(cmd string, args ...interface{}) (interface{}, error) { co := c.get() r, err := co.Do(cmd, args...) c.put(co) return r, err } func (c *Client) Close() { c.Lock() defer c.Unlock() for c.conns.Len() > 0 { e := c.conns.Front() co := e.Value.(*Conn) c.conns.Remove(e) co.finalize() } } func (c *Client) Get() *Conn { return c.get() } func (c *Client) get() *Conn { c.Lock() if c.conns.Len() == 0 { c.Unlock() return c.newConn(c.cfg.Addr) } else { e := c.conns.Front() co := e.Value.(*Conn) c.conns.Remove(e) c.Unlock() return co } } func (c *Client) put(conn *Conn) { c.Lock() if c.conns.Len() >= c.cfg.MaxIdleConns { c.Unlock() conn.finalize() } else { conn.lastActive = time.Now() c.conns.PushFront(conn) c.Unlock() } }