51 lines
1.3 KiB
Go
51 lines
1.3 KiB
Go
package redisInit
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/go-redis/redis/v8"
|
|
"gopkg.in/yaml.v3"
|
|
"os"
|
|
"time"
|
|
"toutoukan/model/config"
|
|
)
|
|
|
|
var RedisClient *redis.Client
|
|
var Ctx = context.Background()
|
|
|
|
func RedisInit() error {
|
|
// 读取配置文件
|
|
data, err := os.ReadFile("./config.yaml")
|
|
if err != nil {
|
|
return fmt.Errorf("读取配置文件失败: %w", err)
|
|
}
|
|
|
|
// 解析配置
|
|
var cfg config.Config
|
|
if err := yaml.Unmarshal(data, &cfg); err != nil {
|
|
return fmt.Errorf("解析YAML失败: %w", err)
|
|
}
|
|
|
|
// 创建Redis客户端
|
|
RedisClient = redis.NewClient(&redis.Options{
|
|
Addr: fmt.Sprintf("%s:%d", cfg.Redis.Host, cfg.Redis.Port), // 地址:端口
|
|
Password: cfg.Redis.Password, // 密码，无密码则为空
|
|
// 连接池配置
|
|
PoolSize: 10, // 连接池大小
|
|
MinIdleConns: 2, // 最小空闲连接数
|
|
MaxConnAge: 30 * time.Minute, // 连接最大存活时间
|
|
IdleTimeout: 5 * time.Minute, // 空闲连接超时时间
|
|
ReadTimeout: 3 * time.Second, // 读取超时时间
|
|
WriteTimeout: 3 * time.Second, // 写入超时时间
|
|
})
|
|
|
|
// 测试连接
|
|
_, err = RedisClient.Ping(Ctx).Result()
|
|
if err != nil {
|
|
return fmt.Errorf("Redis连接失败: %w", err)
|
|
}
|
|
|
|
fmt.Println("Redis连接初始化成功")
|
|
return nil
|
|
}
|