src: Add DMA localagent

[barometer.git] / src / dma / vendor / github.com / go-redis / redis / internal / pool / pool_sticky.go

diff --git a/src/dma/vendor/github.com/go-redis/redis/internal/pool/pool_sticky.go b/src/dma/vendor/github.com/go-redis/redis/internal/pool/pool_sticky.go
new file mode 100644 (file)
index 0000000..91bd913
--- /dev/null
+++ b/src/dma/vendor/github.com/go-redis/redis/internal/pool/pool_sticky.go
@@ -0,0 +1,109 @@
+package pool
+
+import "sync"
+
+type StickyConnPool struct {
+       pool     *ConnPool
+       reusable bool
+
+       cn     *Conn
+       closed bool
+       mu     sync.Mutex
+}
+
+var _ Pooler = (*StickyConnPool)(nil)
+
+func NewStickyConnPool(pool *ConnPool, reusable bool) *StickyConnPool {
+       return &StickyConnPool{
+               pool:     pool,
+               reusable: reusable,
+       }
+}
+
+func (p *StickyConnPool) NewConn() (*Conn, error) {
+       panic("not implemented")
+}
+
+func (p *StickyConnPool) CloseConn(*Conn) error {
+       panic("not implemented")
+}
+
+func (p *StickyConnPool) Get() (*Conn, error) {
+       p.mu.Lock()
+       defer p.mu.Unlock()
+
+       if p.closed {
+               return nil, ErrClosed
+       }
+       if p.cn != nil {
+               return p.cn, nil
+       }
+
+       cn, err := p.pool.Get()
+       if err != nil {
+               return nil, err
+       }
+
+       p.cn = cn
+       return cn, nil
+}
+
+func (p *StickyConnPool) putUpstream() {
+       p.pool.Put(p.cn)
+       p.cn = nil
+}
+
+func (p *StickyConnPool) Put(cn *Conn) {}
+
+func (p *StickyConnPool) removeUpstream() {
+       p.pool.Remove(p.cn)
+       p.cn = nil
+}
+
+func (p *StickyConnPool) Remove(cn *Conn) {
+       p.removeUpstream()
+}
+
+func (p *StickyConnPool) Len() int {
+       p.mu.Lock()
+       defer p.mu.Unlock()
+
+       if p.cn == nil {
+               return 0
+       }
+       return 1
+}
+
+func (p *StickyConnPool) IdleLen() int {
+       p.mu.Lock()
+       defer p.mu.Unlock()
+
+       if p.cn == nil {
+               return 1
+       }
+       return 0
+}
+
+func (p *StickyConnPool) Stats() *Stats {
+       return nil
+}
+
+func (p *StickyConnPool) Close() error {
+       p.mu.Lock()
+       defer p.mu.Unlock()
+
+       if p.closed {
+               return ErrClosed
+       }
+       p.closed = true
+
+       if p.cn != nil {
+               if p.reusable {
+                       p.putUpstream()
+               } else {
+                       p.removeUpstream()
+               }
+       }
+
+       return nil
+}