使用Redis位图实现7天连续签到
需求背景
用户每日签到,以7天为一个周期。签到第一天领取10金币,连续签到两天领30金币,连续签到三天领40金币…期间如果断开则从签到第一天开始
实现思路
实现用户签到功能,我们需要知道用户今日是否签到,用户连续签到的天数,用户签到日历等信息。
对于用户签到数据,如果每条数据都用K/V的方式存储,当用户量大的时候内存开销是非常大的。而位图(BitMap)是由一组bit位组成的,每个bit位对应0和1两个状态,虽然内部还是采用String类型存储,但Redis提供了一些指令用于直接操作位图,可以把它看作是一个bit数组,数组的下标就是偏移量。它的优点是内存开销小、效率高且操作简单,很适合用于签到,用户登录这类场景。
考虑到每月初需要重置连续签到次数,最简单的方式是按用户每月存一条签到数据(也可以每年存一条数据)。Key的格式为u:sign:uid:yyyy-MM,Value则采用长度为4个字节(32位)的位图(最大月份只有31天)。位图的每一位代表一天的签到,1表示已签,0表示未签。例如u:sign:580:2021-08表示ID=580的用户在2021年8月的签到记录。
# 用户2月17号签到
SETBIT u:sign:1000:201902 16 1 # 偏移量是从0开始,所以要把17减1
# 检查2月17号是否签到
GETBIT u:sign:1000:201902 16 # 偏移量是从0开始,所以要把17减1
# 统计2月份的签到次数
BITCOUNT u:sign:1000:201902
# 获取2月份前28天的签到数据
BITFIELD u:sign:1000:201902 get u28 0
# 获取2月份首次签到的日期
BITPOS u:sign:1000:201902 1 # 返回的首次签到的偏移量,加上1即为当月的某一天
上面的两段引用自: https://www.cnblogs.com/liujiduo/p/10396020.html
实例代码
@Slf4j
@Service
public class SignServiceImpl implements SignService {
private final StringRedisTemplate stringRedisTemplate;
public SignServiceImpl(StringRedisTemplate stringRedisTemplate) {
this.stringRedisTemplate = stringRedisTemplate;
}
private int dayOfMonth() {
DateTime dateTime = new DateTime();
return dateTime.dayOfMonth().get();
}
/**
* 按照月份和用户ID生成用户签到标识 UserId:Sign:560:2021-08
*
* @param userId 用户id
* @return
*/
private String signKeyWitMouth(String userId) {
DateTime dateTime = new DateTime();
DateTimeFormatter fmt = DateTimeFormat.forPattern("yyyy-MM");
StringBuilder builder = new StringBuilder("UserId:Sign:");
builder.append(userId).append(":")
.append(dateTime.toString(fmt));
return builder.toString();
}
/**
* 设置标记位
* 标记是否签到
*
* @param key
* @param offset
* @param tag
* @return
*/
public Boolean mark(String key, long offset, boolean tag) {
return this.stringRedisTemplate.opsForValue().setBit(key, offset, tag);
}
/**
* 统计计数
*
* @param key 用户标识
* @return
*/
public long bitCount(String key) {
return stringRedisTemplate.execute((RedisCallback<Long>) redisConnection -> redisConnection.bitCount(key.getBytes()));