This commit is contained in:
2026-09-15 16:02:45 +08:00
parent 112c5f8e46
commit ade2453083
70 changed files with 1441 additions and 707 deletions
-23
View File
@@ -22,15 +22,6 @@ end
return v
`)
// notificationUnreadDecrScript 未读数 -1,保证结果不低于 0
var notificationUnreadDecrScript = redis.NewScript(`
local v = redis.call('GET', KEYS[1])
if v and tonumber(v) > 0 then
return redis.call('DECRBY', KEYS[1], 1)
end
return 0
`)
// NotificationUnreadIncr 用户未读消息数 +1,返回最新值
func NotificationUnreadIncr(ctx context.Context, userID int64) (int64, error) {
ctx, span := global.Tracer.Start(ctx, "redisdao.NotificationUnreadIncr")
@@ -45,20 +36,6 @@ func NotificationUnreadIncr(ctx context.Context, userID int64) (int64, error) {
return val, nil
}
// NotificationUnreadDecr 用户未读消息数 -1,返回扣减后的值(不低于 0)
func NotificationUnreadDecr(ctx context.Context, userID int64) (int64, error) {
ctx, span := global.Tracer.Start(ctx, "redisdao.NotificationUnreadDecr")
defer span.End()
key := fmt.Sprintf(redismodel.KeyNotificationUnread, userID)
val, err := notificationUnreadDecrScript.Run(ctx, global.RedisEngine, []string{key}).Int64()
if err != nil {
span.RecordError(err)
return 0, err
}
return val, nil
}
// NotificationUnreadGet 获取用户未读消息数,不存在时返回 0
func NotificationUnreadGet(ctx context.Context, userID int64) (int64, error) {
ctx, span := global.Tracer.Start(ctx, "redisdao.NotificationUnreadGet")
+67
View File
@@ -0,0 +1,67 @@
package redisdao
import (
"context"
"fmt"
"time"
"git.lzcwy.cn/zhengchiliu/zt-closet-backend/global"
"git.lzcwy.cn/zhengchiliu/zt-closet-backend/internal/redismodel"
)
// VirtualPayUnpaidSet 标记某虚拟支付订单「最近一次查单确认未支付」,TTL 内不再重复查询。
func VirtualPayUnpaidSet(ctx context.Context, orderNo string, ttl time.Duration) error {
ctx, span := global.Tracer.Start(ctx, "redisdao.VirtualPayUnpaidSet")
defer span.End()
key := fmt.Sprintf(redismodel.KeyVirtualPayUnpaid, orderNo)
if err := global.RedisEngine.Set(ctx, key, 1, ttl).Err(); err != nil {
span.RecordError(err)
return err
}
return nil
}
// VirtualPayUnpaidDel 清除负缓存(订单已支付/已处理后调用,让后续流程不再跳过它)。
func VirtualPayUnpaidDel(ctx context.Context, orderNo string) error {
ctx, span := global.Tracer.Start(ctx, "redisdao.VirtualPayUnpaidDel")
defer span.End()
key := fmt.Sprintf(redismodel.KeyVirtualPayUnpaid, orderNo)
if err := global.RedisEngine.Del(ctx, key).Err(); err != nil {
span.RecordError(err)
return err
}
return nil
}
// VirtualPayUnpaidBatchGet 批量判断哪些订单命中负缓存。
// 用一次 MGET 代替 N 次 GET,避免逐单往返 Redis 放大单轮耗时。
func VirtualPayUnpaidBatchGet(ctx context.Context, orderNos []string) (map[string]bool, error) {
ctx, span := global.Tracer.Start(ctx, "redisdao.VirtualPayUnpaidBatchGet")
defer span.End()
result := make(map[string]bool, len(orderNos))
if len(orderNos) == 0 {
return result, nil
}
keys := make([]string, 0, len(orderNos))
for _, no := range orderNos {
keys = append(keys, fmt.Sprintf(redismodel.KeyVirtualPayUnpaid, no))
}
vals, err := global.RedisEngine.MGet(ctx, keys...).Result()
if err != nil {
span.RecordError(err)
return nil, err
}
for i, v := range vals {
if i >= len(orderNos) {
break
}
if v != nil {
result[orderNos[i]] = true
}
}
return result, nil
}