61 lines
1.3 KiB
Go
61 lines
1.3 KiB
Go
|
package myRedis
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"fmt"
|
|||
|
|
"github.com/redis/go-redis/v9"
|
|||
|
|
"log"
|
|||
|
|
"sync"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type SimpleRedis struct {
|
|||
|
|
Host string
|
|||
|
|
Password string
|
|||
|
|
hosts map[int]*Hosts
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type Hosts struct {
|
|||
|
|
clientOnce sync.Once
|
|||
|
|
client *redis.Client
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func NewSimpleRedis(host string, password string) *SimpleRedis {
|
|||
|
|
return &SimpleRedis{Host: host, Password: password, hosts: map[int]*Hosts{}}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (t *SimpleRedis) connectRedis(index int) *redis.Client {
|
|||
|
|
if t.hosts[index] == nil {
|
|||
|
|
t.hosts[index] = &Hosts{
|
|||
|
|
clientOnce: sync.Once{},
|
|||
|
|
client: nil,
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
t.hosts[index].clientOnce.Do(func() {
|
|||
|
|
redisClient := redis.NewClient(&redis.Options{
|
|||
|
|
Addr: t.Host,
|
|||
|
|
Password: t.Password, // no password set
|
|||
|
|
DB: index, // use default DB
|
|||
|
|
//连接池容量以闲置链接数量
|
|||
|
|
PoolSize: 15,
|
|||
|
|
MinIdleConns: 10,
|
|||
|
|
})
|
|||
|
|
pong, err := redisClient.Ping(context.Background()).Result()
|
|||
|
|
if err != nil {
|
|||
|
|
panic(fmt.Errorf("connect error:%s", err))
|
|||
|
|
}
|
|||
|
|
log.Println(fmt.Sprintf("redis newClient success, index：%d, pong: %s", index, pong))
|
|||
|
|
|
|||
|
|
t.hosts[index].client = redisClient
|
|||
|
|
})
|
|||
|
|
return t.hosts[index].client
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (t *SimpleRedis) ConnectDefaultRedis() *Client {
|
|||
|
|
return t.GetRedisClient(0)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (t *SimpleRedis) GetRedisClient(index int) *Client {
|
|||
|
|
return NewClient(t.connectRedis(index))
|
|||
|
|
}
|