150 lines
6.7 KiB
Go
150 lines
6.7 KiB
Go
// Package redis ...
|
|
//
|
|
// Description : redis ...
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 2022-06-15 11:52
|
|
package redis
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/go-redis/redis/v8"
|
|
|
|
"git.zhangdeman.cn/zhangdeman/logger"
|
|
)
|
|
|
|
// FullConfig 完整配置
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 12:23 2022/6/15
|
|
type FullConfig struct {
|
|
ExtraFieldList []string `json:"extra_field_list" yaml:"extra_field_list"` // 从上下文抽取的字段信息
|
|
Logger *logger.InputLogConfig `json:"logger" yaml:"logger"` // 日志配置
|
|
Master *Config `json:"master" yaml:"master"` // 主节点
|
|
Slave *Config `json:"slave" yaml:"slave"` // 从节点
|
|
}
|
|
|
|
// Config redis的配置
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 11:52 2022/6/15
|
|
type Config struct {
|
|
Network string `json:"network" yaml:"network"` // 连接方式 tcp 或 unix , 默认 tcp
|
|
Host string `yaml:"host" json:"host"` // 地址
|
|
Port int `yaml:"port" json:"port"` // 端口
|
|
|
|
// Dialer creates new network connection and has priority over
|
|
// Network and Addr options.
|
|
// Dialer func(ctx context.Context, network, addr string) (net.Conn, error)
|
|
|
|
// Hook that is called when new connection is established.
|
|
// OnConnect func(ctx context.Context, cn *redis.Conn) error
|
|
// Use the specified Username to authenticate the current connection
|
|
// with one of the connections defined in the ACL list when connecting
|
|
// to a Redis 6.0 instance, or greater, that is using the Redis ACL system.
|
|
Username string `json:"username" yaml:"username"` // 账号
|
|
// Optional password. Must match the password specified in the
|
|
// requirepass server configuration option (if connecting to a Redis 5.0 instance, or lower),
|
|
// or the User Password when connecting to a Redis 6.0 instance, or greater,
|
|
// that is using the Redis ACL system.
|
|
Password string `json:"password" yaml:"password"` // 密码
|
|
|
|
// Database to be selected after connecting to the server.
|
|
DB int `json:"db" yaml:"db"` // 选中的数据库
|
|
|
|
// Maximum number of retries before giving up.
|
|
// Default is 3 retries; -1 (not 0) disables retries.
|
|
MaxRetries int `json:"max_retries" yaml:"max_retries"` // 最大重试次数, 默认 3 , -1 为不重试
|
|
// Minimum backoff between each retry.
|
|
// Default is 8 milliseconds; -1 disables backoff.
|
|
MinRetryBackoff int64 `json:"min_retry_backoff" yaml:"min_retry_backoff"` // 最小重试的时间间隔, 默认 8ms, -1禁用
|
|
// Maximum backoff between each retry.
|
|
// Default is 512 milliseconds; -1 disables backoff.
|
|
MaxRetryBackoff int `json:"max_retry_backoff" yaml:"max_retry_backoff"` // 最大重试的时间间隔, 默认 512ms, -1禁用
|
|
|
|
// Dial timeout for establishing new connections.
|
|
// Default is 5 seconds.
|
|
DialTimeout int64 `json:"dial_timeout" yaml:"dial_timeout"` // 建立新连接的超时时间, 默认 5s
|
|
// Timeout for socket reads. If reached, commands will fail
|
|
// with a timeout instead of blocking. Use value -1 for no timeout and 0 for default.
|
|
// Default is 3 seconds.
|
|
ReadTimeout int64 `json:"read_timeout" yaml:"read_timeout"` // 读取超时时间，默认 3s , -1 不设置超时
|
|
// Timeout for socket writes. If reached, commands will fail
|
|
// with a timeout instead of blocking.
|
|
// Default is ReadTimeout.
|
|
WriteTimeout int64 `json:"write_timeout" yaml:"write_timeout"` // 写入超时时间, 默认值和 ReadTimeout 配置一致
|
|
|
|
// Type of connection pool.
|
|
// true for FIFO pool, false for LIFO pool.
|
|
// Note that fifo has higher overhead compared to lifo.
|
|
PoolFIFO bool `json:"pool_fifo" yaml:"pool_fifo"` // 连接吃采用 FIFO
|
|
// Maximum number of socket connections.
|
|
// Default is 10 connections per every available CPU as reported by runtime.GOMAXPROCS.
|
|
PoolSize int `json:"pool_size" yaml:"pool_size"` // 连接池大小, 默认 runtime.GOMAXPROCS * 10
|
|
// Minimum number of idle connections which is useful when establishing
|
|
// new connection is slow.
|
|
MinIdleConns int `json:"min_idle_conns" yaml:"min_idle_conns"` // 最小的 空闲 连接数数量
|
|
// Connection age at which client retires (closes) the connection.
|
|
// Default is to not close aged connections.
|
|
MaxConnAge int64 `json:"max_conn_age" yaml:"max_conn_age"` // 连接最大的存活时常
|
|
// Amount of time client waits for connection if all connections
|
|
// are busy before returning an error.
|
|
// Default is ReadTimeout + 1 second.
|
|
PoolTimeout int64 `json:"pool_timeout" yaml:"pool_timeout"` // 连接池无可用连接, 等待的时常, 默认值 ReadTimeout + 1
|
|
// Amount of time after which client closes idle connections.
|
|
// Should be less than server's timeout.
|
|
// Default is 5 minutes. -1 disables idle timeout check.
|
|
IdleTimeout int64 `json:"idle_timeout" yaml:"idle_timeout"` // 空闲链接的超时时间, 默认 5 分钟 -1 金童
|
|
// Frequency of idle checks made by idle connections reaper.
|
|
// Default is 1 minute. -1 disables idle connections reaper,
|
|
// but idle connections are still discarded by the client
|
|
// if IdleTimeout is set.
|
|
IdleCheckFrequency int64 `json:"idle_check_frequency" yaml:"idle_check_frequency"` // 空闲连接检查频率, 默认 1 分钟, -1 禁用, 注意 : 即使禁用，超过 IdleTimeout 时常的连接也不可被读取到
|
|
}
|
|
|
|
// Config2Options 将 自定义的配置文件转为 redis 包需要的配置格式
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 17:37 2022/6/15
|
|
func Config2Options(cfg *Config) *redis.Options {
|
|
option := &redis.Options{
|
|
Network: cfg.Network,
|
|
Addr: fmt.Sprintf("%v:%v", cfg.Host, cfg.Port),
|
|
Username: cfg.Username,
|
|
Password: cfg.Password,
|
|
DB: cfg.DB,
|
|
MaxRetries: cfg.MaxRetries,
|
|
MinRetryBackoff: time.Duration(cfg.MinRetryBackoff) * time.Millisecond,
|
|
MaxRetryBackoff: time.Duration(cfg.MaxRetryBackoff) * time.Millisecond,
|
|
DialTimeout: time.Duration(cfg.DialTimeout) * time.Millisecond,
|
|
ReadTimeout: time.Duration(cfg.ReadTimeout) * time.Millisecond,
|
|
WriteTimeout: time.Duration(cfg.WriteTimeout) * time.Millisecond,
|
|
PoolFIFO: cfg.PoolFIFO,
|
|
PoolSize: cfg.PoolSize,
|
|
MinIdleConns: cfg.MinIdleConns,
|
|
MaxConnAge: time.Duration(cfg.MaxConnAge) * time.Millisecond,
|
|
PoolTimeout: time.Duration(cfg.PoolTimeout) * time.Millisecond,
|
|
IdleTimeout: time.Duration(cfg.IdleTimeout) * time.Millisecond,
|
|
IdleCheckFrequency: time.Duration(cfg.IdleCheckFrequency) * time.Millisecond,
|
|
TLSConfig: nil,
|
|
Limiter: nil,
|
|
}
|
|
if len(option.Network) == 0 {
|
|
option.Network = NetworkTypeTCP
|
|
}
|
|
return option
|
|
}
|
|
|
|
const (
|
|
// NetworkTypeTCP ...
|
|
NetworkTypeTCP = "tcp"
|
|
// NetworkTypeUnix ...
|
|
NetworkTypeUnix = "unix"
|
|
)
|