mirror of
https://github.com/linux-do/new-api.git
synced 2025-09-17 16:06:38 +08:00
108 lines
2.5 KiB
Go
108 lines
2.5 KiB
Go
package common
|
||
|
||
import (
|
||
"context"
|
||
"github.com/go-redis/redis/v8"
|
||
"os"
|
||
"time"
|
||
)
|
||
|
||
var RDB *redis.Client
|
||
var RedisEnabled = true
|
||
|
||
// InitRedisClient This function is called after init()
|
||
func InitRedisClient() (err error) {
|
||
if os.Getenv("REDIS_CONN_STRING") == "" {
|
||
RedisEnabled = false
|
||
SysLog("REDIS_CONN_STRING not set, Redis is not enabled")
|
||
return nil
|
||
}
|
||
if os.Getenv("SYNC_FREQUENCY") == "" {
|
||
RedisEnabled = false
|
||
SysLog("SYNC_FREQUENCY not set, Redis is disabled")
|
||
return nil
|
||
}
|
||
SysLog("Redis is enabled")
|
||
opt, err := redis.ParseURL(os.Getenv("REDIS_CONN_STRING"))
|
||
if err != nil {
|
||
FatalLog("failed to parse Redis connection string: " + err.Error())
|
||
}
|
||
RDB = redis.NewClient(opt)
|
||
|
||
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||
defer cancel()
|
||
|
||
_, err = RDB.Ping(ctx).Result()
|
||
if err != nil {
|
||
FatalLog("Redis ping test failed: " + err.Error())
|
||
}
|
||
return err
|
||
}
|
||
|
||
func ParseRedisOption() *redis.Options {
|
||
opt, err := redis.ParseURL(os.Getenv("REDIS_CONN_STRING"))
|
||
if err != nil {
|
||
FatalLog("failed to parse Redis connection string: " + err.Error())
|
||
}
|
||
return opt
|
||
}
|
||
|
||
func RedisSet(key string, value string, expiration time.Duration) error {
|
||
ctx := context.Background()
|
||
return RDB.Set(ctx, key, value, expiration).Err()
|
||
}
|
||
|
||
func RedisGet(key string) (string, error) {
|
||
ctx := context.Background()
|
||
return RDB.Get(ctx, key).Result()
|
||
}
|
||
|
||
func RedisExpire(key string, expiration time.Duration) error {
|
||
ctx := context.Background()
|
||
return RDB.Expire(ctx, key, expiration).Err()
|
||
}
|
||
|
||
func RedisGetEx(key string, expiration time.Duration) (string, error) {
|
||
ctx := context.Background()
|
||
return RDB.GetSet(ctx, key, expiration).Result()
|
||
}
|
||
|
||
func RedisDel(key string) error {
|
||
ctx := context.Background()
|
||
return RDB.Del(ctx, key).Err()
|
||
}
|
||
|
||
func RedisDecrease(key string, value int64) error {
|
||
|
||
// 检查键的剩余生存时间
|
||
ttlCmd := RDB.TTL(context.Background(), key)
|
||
ttl, err := ttlCmd.Result()
|
||
if err != nil {
|
||
// 失败则尝试直接减少
|
||
return RDB.DecrBy(context.Background(), key, value).Err()
|
||
}
|
||
|
||
// 如果剩余生存时间大于0，则进行减少操作
|
||
if ttl > 0 {
|
||
ctx := context.Background()
|
||
// 开始一个Redis事务
|
||
txn := RDB.TxPipeline()
|
||
|
||
// 减少余额
|
||
decrCmd := txn.DecrBy(ctx, key, value)
|
||
if err := decrCmd.Err(); err != nil {
|
||
return err // 如果减少失败，则直接返回错误
|
||
}
|
||
|
||
// 重新设置过期时间，使用原来的过期时间
|
||
txn.Expire(ctx, key, ttl)
|
||
|
||
// 执行事务
|
||
_, err = txn.Exec(ctx)
|
||
return err
|
||
} else {
|
||
_ = RedisDel(key)
|
||
}
|
||
return nil
|
||
}
|