Redis位图实现周签到,补签,查看签到及定时提醒签到(goland)

Redis位图实现周签到,补签,查看签到及定时提醒签到

场景需求

1.用户签到送积分:

签到1天送2积分,连续2天签到送4积分,3天送6积分…(周签到);

如果连续签到中断,则重置计数,每周初重置计数;

思路及步骤:

1.使用redis位图(BitMap)实现签到功能

2.连接redis

// 创建 Redis 客户端
func InitRedis() *redis.Client {
    rdb := redis.NewClient(&redis.Options{
       Addr: "localhost:6379", // Redis 地址
       DB:   4,                // 使用的数据库索引（这里是 4）
    })
    return rdb
}

3.检查用户今天是否已签到

ctx = context.Background()              // 创建上下文，用于 Redis 操作
today := time.Now().Format("2006-01-02") // 获取当前日期，格式为 YYYY-MM-DD
key := fmt.Sprintf("sign_in:%s", today)  // 生成 Redis 键，用于标识今天的签到状态

// 检查用户今天是否已经签到
signedIn, err := rdb.GetBit(ctx, key, int64(userID)).Result()
if err != nil {
    return err // 如果发生错误，返回
}
if signedIn == 1 {
    return nil // 如果已签到，直接返回
}

4.将用户的签到设置为1,表示已签到

// 将用户的签到位设置为 1，表示已签到
if _, err = rdb.SetBit(ctx, key, int64(userID), 1).Result(); err != nil {
    return err // 如果发生错误，返回
}

5.计算连续签到的天数

// 计算连续签到天数
consecutiveDays := 0     // 初始化连续签到天数计数器
for i := 0; i < 7; i++ { // 检查过去7天的签到记录
    date := time.Now().AddDate(0, 0, -i).Format("2006-01-02") // 获取过去 i 天的日期
    if signedIn, _ := rdb.GetBit(ctx, fmt.Sprintf("sign_in:%s", date), int64(userID)).Result(); signedIn == 1 {
       if i== consecutiveDays { // 如果当前天数与计数器相同，表示连续签到
          consecutiveDays++
       } else {
          break // 如果发现中断，则停止计数
       }
    } else {
       break // 如果某一天没有签到，也停止计数
    }
}

6.每周初重置计数

// 每周初重置计数
if time.Now().Weekday() == time.Sunday {
    consecutiveDays = 0 // 如果是周日，重置连续签到天数计数
}

7.根据连续签到天数计算奖励积分

// 根据连续签到天数计算奖励积分
rewardPoints := (consecutiveDays + 1) * 2                      // 奖励积分，第一天加2，第二天加4，以此类推

8.添加或更新用户积分

userPoints := UserPoints{UserID: userID, Points: rewardPoints} // 创建用户积分对象

// 使用 GORM 的 Save 方法，插入或更新用户积分
if err = db.Where(UserPoints{UserID: userID}).Attrs(UserPoints{Points: rewardPoints}).FirstOrCreate(&userPoints).Error; err != nil {
    return err // 如果发生错误，返回
}

// 更新积分，增加新的奖励积分
if err = db.Model(&userPoints).Update("points", gorm.Expr("points + ?", rewardPoints)).Error; err != nil {
    return err // 如果发生错误，返回
}

完整示例:

// SignIn 函数处理用户签到并更新积分
func SignIn(db *gorm.DB, rdb *redis.Client, userID int) error {
    ctx = context.Background()               // 创建上下文，用于 Redis 操作
    today := time.Now().Format("2006-01-02") // 获取当前日期，格式为 YYYY-MM-DD
    key := fmt.Sprintf("sign_in:%s", today)  // 生成 Redis 键，用于标识今天的签到状态

    // 检查用户今天是否已经签到
    signedIn, err := rdb.GetBit(ctx, key, int64(userID)).Result()
    if err != nil {
       return err // 如果发生错误，返回
    }
    if signedIn == 1 {
       return nil // 如果已签到，直接返回
    }

    // 将用户的签到位设置为 1，表示已签到
    if _, err = rdb.SetBit(ctx, key, int64(userID), 1).Result(); err != nil {
       return err // 如果发生错误，返回
    }

    // 计算连续签到天数
    consecutiveDays := 0     // 初始化连续签到天数计数器
    for i := 0; i < 7; i++ { // 检查过去7天的签到记录
       date := time.Now().AddDate(0, 0, -i).Format("2006-01-02") // 获取过去 i 天的日期