red-future/common
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 19 Wiki Activity

协程池单例模式

Browse Source
This commit is contained in:
Cold
2025-12-06 12:02:34 +08:00
committed by 张斌
parent 86661c687a
commit f7cb007491
2 changed files with 50 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
redis/redis.go
View File

@@ -10,16 +10,19 @@ import (
"github.com/gogf/gf/v2/util/gconv"
)
// GRedisClient GoFrame gredis 客户端，统一使用
// GRedisClient GoFrame gredis 客户端，统一使用（懒加载）
var GRedisClient *gredis.Redis
// RedisClient GRedisClient 的别名，保持向后兼容
var RedisClient *gredis.Redis
func init() {
// 初始化 GoFrame gredis 客户端
GRedisClient = g.Redis()
RedisClient = GRedisClient // 别名指向同一个客户端
// GetRedisClient 获取 Redis 客户端（懒加载）
func GetRedisClient() *gredis.Redis {
if GRedisClient == nil {
GRedisClient = g.Redis()
RedisClient = GRedisClient
}
return GRedisClient
}
// Stream 和消费者组常量
@@ -42,7 +45,7 @@ type StreamMessage struct {
// 使用 gredis Do() 方法执行 XGROUP CREATE 命令
func InitStreamGroup(ctx context.Context, streamKey, groupName string) error {
// XGROUP CREATE streamKey groupName 0 MKSTREAM
_, err := GRedisClient.Do(ctx, "XGROUP", "CREATE", streamKey, groupName, "0", "MKSTREAM")
_, err := GetRedisClient().Do(ctx, "XGROUP", "CREATE", streamKey, groupName, "0", "MKSTREAM")
if err != nil {
// 如果组已存在，忽略错误
errStr := err.Error()
@@ -63,7 +66,7 @@ func AddToStream(ctx context.Context, streamKey string, values map[string]interf
args = append(args, key, val)
}
result, err := GRedisClient.Do(ctx, "XADD", args...)
result, err := GetRedisClient().Do(ctx, "XADD", args...)
if err != nil {
return "", err
}
@@ -77,7 +80,7 @@ func AddToStream(ctx context.Context, streamKey string, values map[string]interf
// 使用 gredis Do() 方法执行 XREADGROUP 命令
func ReadFromStream(ctx context.Context, streamKey, groupName, consumerName string, count int64, blockMs int64) ([]StreamMessage, error) {
// XREADGROUP GROUP groupName consumerName COUNT count BLOCK blockMs STREAMS streamKey >
result, err := GRedisClient.Do(ctx,
result, err := GetRedisClient().Do(ctx,
"XREADGROUP", "GROUP", groupName, consumerName,
"COUNT", count,
"BLOCK", blockMs,
@@ -162,7 +165,7 @@ func AckMessage(ctx context.Context, streamKey, groupName string, messageIDs ...
args = append(args, id)
}
_, err := GRedisClient.Do(ctx, "XACK", args...)
_, err := GetRedisClient().Do(ctx, "XACK", args...)
return err
}
@@ -170,7 +173,7 @@ func AckMessage(ctx context.Context, streamKey, groupName string, messageIDs ...
// 使用 gredis Do() 方法执行 XLEN 命令
func GetStreamLength(ctx context.Context, streamKey string) (int64, error) {
// XLEN streamKey
result, err := GRedisClient.Do(ctx, "XLEN", streamKey)
result, err := GetRedisClient().Do(ctx, "XLEN", streamKey)
if err != nil {
return 0, err
}
@@ -191,7 +194,7 @@ type PendingMessage struct {
// 使用 gredis Do() 方法执行 XPENDING 命令
func GetPendingMessages(ctx context.Context, streamKey, groupName string, start, end string, count int64) ([]PendingMessage, error) {
// XPENDING streamKey groupName start end count
result, err := GRedisClient.Do(ctx, "XPENDING", streamKey, groupName, start, end, count)
result, err := GetRedisClient().Do(ctx, "XPENDING", streamKey, groupName, start, end, count)
if err != nil {
return nil, err
}
@@ -233,7 +236,7 @@ func ClaimPendingMessage(ctx context.Context, streamKey, groupName, consumerName
args = append(args, id)
}
result, err := GRedisClient.Do(ctx, "XCLAIM", args...)
result, err := GetRedisClient().Do(ctx, "XCLAIM", args...)
if err != nil {
return nil, err
}
@@ -287,7 +290,7 @@ func SetSessionLastActive(ctx context.Context, userId string) error {
timestamp := gtime.Now().Timestamp()
// SETEX key 7200 value (7200秒 = 2小时)
_, err := GRedisClient.Do(ctx, "SETEX", key, 7200, timestamp)
_, err := GetRedisClient().Do(ctx, "SETEX", key, 7200, timestamp)
return err
}
@@ -295,7 +298,7 @@ func SetSessionLastActive(ctx context.Context, userId string) error {
// 使用 gredis Get 方法
func GetSessionLastActive(ctx context.Context, userId string) (int64, error) {
key := SessionLastActiveKeyPrefix + userId + ":last_active"
result, err := GRedisClient.Get(ctx, key)
result, err := GetRedisClient().Get(ctx, key)
if err != nil {
return 0, err
}
@@ -337,7 +340,7 @@ func SetSessionCache(ctx context.Context, userId, sessionId string) error {
key := SessionLastActiveKeyPrefix + userId + ":session_id"
// SETEX key 604800 value (604800秒 = 7天)
_, err := GRedisClient.Do(ctx, "SETEX", key, 604800, sessionId)
_, err := GetRedisClient().Do(ctx, "SETEX", key, 604800, sessionId)
return err
}
@@ -345,7 +348,7 @@ func SetSessionCache(ctx context.Context, userId, sessionId string) error {
// 使用 gredis Get 方法
func GetSessionCache(ctx context.Context, userId string) (string, error) {
key := SessionLastActiveKeyPrefix + userId + ":session_id"
result, err := GRedisClient.Get(ctx, key)
result, err := GetRedisClient().Get(ctx, key)
if err != nil {
return "", err
}