Source of conn.go - Zabbix

Raw file
Source viewDiff to previous
5 KB
xxxxxxxxxx
// get returns a connection with given uri if it exists and also updates lastTimeAccess, otherwise returns nil.
 
/*
** Copyright (C) 2001-2025 Zabbix SIA
**
** This program is free software: you can redistribute it and/or modify it under the terms of
** the GNU Affero General Public License as published by the Free Software Foundation, version 3.
**
** This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
** without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
** See the GNU Affero General Public License for more details.
**
** You should have received a copy of the GNU Affero General Public License along with this program.
** If not, see <https://www.gnu.org/licenses/>.
**/
​
package memcached
​
import (
    "context"
    "sync"
    "time"
​
    "github.com/memcachier/mc/v3"
    "golang.zabbix.com/sdk/log"
    "golang.zabbix.com/sdk/uri"
    "golang.zabbix.com/sdk/zbxerr"
)
​
const poolSize = 1
​
type MCClient interface {
    Stats(key string) (mc.McStats, error)
    NoOp() error
}
​
type MCConn struct {
    client         mc.Client
    lastTimeAccess time.Time
}
​
// stubConn for testing
type stubConn struct {
    NoOpFunc  func() error
    StatsFunc func(key string) (mc.McStats, error)
}
​
func (c *stubConn) Stats(key string) (mc.McStats, error) {
    return c.StatsFunc(key)
}
​
func (c *stubConn) NoOp() error {
    return c.NoOpFunc()
}
​
// Stats wraps the mc.Client.StatsWithKey function.
func (conn *MCConn) Stats(key string) (stats mc.McStats, err error) {
    res, err := conn.client.StatsWithKey(key)
    if err != nil {
        return nil, err
    }
​
    if len(res) == 0 {
        return nil, zbxerr.ErrorEmptyResult
    }
​
    if len(res) > 1 {
        panic("unexpected result")
    }
​
    // get the only entry of stats
    for _, stats = range res {
        break
    }
​
    return stats, err
}
​
// NoOp wraps the mc.Client.NoOp function.
func (conn *MCConn) NoOp() error {
    return conn.client.NoOp()
}
​
// updateAccessTime updates the last time a connection was accessed.
func (conn *MCConn) updateAccessTime() {
    conn.lastTimeAccess = time.Now()
}
​
// ConnManager is thread-safe structure for manage connections.
type ConnManager struct {
    sync.Mutex
    connMutex   sync.Mutex
    connections map[uri.URI]*MCConn