ledisdb/client/go/ledis/client.go

package ledis

import (
	"container/list"
	"strings"
	"sync"
	"time"
)

const (
	pingPeriod time.Duration = 3 * time.Second
)

type Config struct {
	Addr         string
	MaxIdleConns int
}

type Client struct {
	sync.Mutex

	cfg   *Config
	proto string

	conns *list.List
}

func NewClient(cfg *Config) *Client {
	c := new(Client)

	c.cfg = cfg

	if strings.Contains(cfg.Addr, "/") {
		c.proto = "unix"
	} else {
		c.proto = "tcp"
	}

	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()
	} 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()
	}
}