56 lines
1.6 KiB
Go
56 lines
1.6 KiB
Go
package model
|
||
|
||
import (
|
||
"context"
|
||
"errors"
|
||
"fmt"
|
||
"github.com/go-redis/redis/v8"
|
||
"samba/pkg/xtime"
|
||
"samba/util/rdbkey"
|
||
)
|
||
|
||
type UserStreakOp struct {
|
||
rdb *redis.Client
|
||
}
|
||
|
||
func NewUserStreakOp() *UserStreakOp {
|
||
return &UserStreakOp{
|
||
rdb: rdbGameLog,
|
||
}
|
||
}
|
||
|
||
// SetLossStreak 设置用户连败次数
|
||
func (op *UserStreakOp) SetLossStreak(uid int64, gameType string, count int) error {
|
||
return op.rdb.HSet(context.TODO(), rdbkey.UserLossStreakKey(), fmt.Sprintf("%d-%s", uid, gameType), count).Err()
|
||
}
|
||
|
||
// IncLossStreak 增加用户连败次数
|
||
func (op *UserStreakOp) IncLossStreak(uid int64, gameType string, inc int) (updated int64, err error) {
|
||
return op.rdb.HIncrBy(context.TODO(), rdbkey.UserLossStreakKey(), fmt.Sprintf("%d-%s", uid, gameType), int64(inc)).Result()
|
||
}
|
||
|
||
// UpdateBankruptStreak 更新连续破产次数
|
||
func (op *UserStreakOp) UpdateBankruptStreak(uid int64) (updated int64, err error) {
|
||
// 获取上次更新的时间
|
||
lastTime, err := op.rdb.HGet(context.TODO(), rdbkey.UserBankruptStreakKey(), fmt.Sprintf("%d-timestamp", uid)).Int64()
|
||
if err != nil && !errors.Is(err, redis.Nil) {
|
||
return 0, err
|
||
}
|
||
if xtime.IsToday(lastTime) {
|
||
// 时间为今天，直接Inc
|
||
updated, err = op.rdb.HIncrBy(context.TODO(), rdbkey.UserBankruptStreakKey(), fmt.Sprintf("%d", uid), 1).Result()
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
|
||
} else {
|
||
err = op.rdb.HSet(context.TODO(), rdbkey.UserBankruptStreakKey(), fmt.Sprintf("%d", uid), 1).Err()
|
||
updated = 1
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
}
|
||
err = op.rdb.HSet(context.TODO(), rdbkey.UserBankruptStreakKey(), fmt.Sprintf("%d-timestamp", uid), xtime.Now().Timestamp()).Err()
|
||
return
|
||
}
|