Redis数据结构Bitmap实战之用户签到

前言

bitmap就是通过最小的单位bit来进行0或者1的设置，表示某个元素对应的值或者状态。
一个bit的值，或者是0，或者是1；也就是说一个bit能存储的最多信息是2。
换句话来说它的优势是占用空间小、处理速度快

业务分析

我们可以设置Redis的key为user:sign:customerId:yyyyMM 那么可推出如下命令
说明：签到天数从0开始，倒数第二位是偏移量代表天数，最后一位1代表已签到
第一天签到：setbit user:sign:1001:202203 0 1
第二天签到：setbit user:sign:1001:202203 1 1
第五天签到：setbit user:sign:1001:202203 4 1
此时Redis的值为11001000 ,其余地方自动补0，继续执行

第12天签到：setbit user:sign:1001:202203 11 1
此时Redis的值为 1100100000010000

签到代码实现

签到接口返回对象代码：

@Data
@ApiModel(value="签到对象", description="签到对象")
public class SignVO {

    @ApiModelProperty(value = "连续签到次数")
    private Integer continuous;

    @ApiModelProperty(value = "总签到次数")
    private Long count;

    @ApiModelProperty(value = "签到结果")
    private boolean flag;
}

统计当月连续签到次数和统计总签到次数在后面，签到具体代码如下：

/**
  * 用户签到
  * @return
  */
public SignVO doSign() {
	// 获取当前登录的用户id
    final String customerId = StpUtil.getLoginIdAsString();
    Map<String, Object> result = new HashMap<>();
    // 1.获取日期
    Date date = new Date();
    // 2.获取日期对应天数，多少号
    int day = DateUtil.dayOfMonth(date) - 1;
    // 3.构建Redis key
    String key = buildSignKey(customerId, date);
    // 4.查看今日是否签到
    boolean isSigned = redisTemplate.opsForValue().getBit(key,day);
    if (isSigned) throw new DefaultException("今日已签到");
    // 5.签到
    redisTemplate.opsForValue().setBit(key,day,true);
    // 6.TODO 统计当月连续签到次数
    // int continuous = getContinuousSignCount(customerId, date);
    // 7.TODO 统计总签到次数
    // long count = getSumSignCount(customerId, date);

	SignVO vo = new SignVO();
    vo.setFlag(true);
    return vo;
 }

 /**
  * 构建Redis key user:sign:customerId:yyyyMM
  * @param customerId 用户id
  * @param date 签到日期
  * @return key
  */
 private String buildSignKey(String customerId, Date date) {
     return String.format("user:sign:%s:%s",customerId,DateUtil.format(date,"yyyyMM"));
 }

统计总签到数和连续签到数

/**
  * 统计总签到次数
  * @param customerId 用户id
  * @param date       日期
  * @return 总共签到次数
  */
 public long getSumSignCount(String customerId, Date date) {
     String key = buildSignKey(customerId, date);
     return (Long) redisTemplate.execute((RedisCallback<Long>) connection ->
             connection.bitCount(key.getBytes())
     );
 }

 /**
  * 统计连续签到次数
  * @param customerId 用户id
  * @param date       日期
  * @return 连续签到次数
  */
 public int getContinuousSignCount(String customerId, Date date) {
     // 获取日期对应天数
     int dayOfMonth = DateUtil.dayOfMonth(date);
     // 构建 Redis Key
     String signKey = buildSignKey(customerId, date);
     // 命令例子：bitfield user:sign:1001:202203 get u20 0
     BitFieldSubCommands bitFieldSubCommands = BitFieldSubCommands.create();
     bitFieldSubCommands.get(BitFieldSubCommands.BitFieldType.unsigned(dayOfMonth)).valueAt(0);

     // 获取用户从当前日期开始到1号的签到状态
     List<Long> list = redisTemplate.opsForValue().bitField(signKey, bitFieldSubCommands);
     if (list == null || list.isEmpty()) {
         return 0;
     }
     // 连续签到计数器
     int signCount = 0;
     long v = list.get(0) == null ? 0 : list.get(0);
     // 位移运算连续签到次数
     for (int i = dayOfMonth; i > 0; i--) {
         // i表示位移操作的次数，右移再左移如果等于自己说明最低位是0，表示为签到
         if (v >> 1 << 1 == v) {
             // 用户可能当前还未签到，所以要排除是否是当天的可能性
             if (i != dayOfMonth) break;

         } else {
             // 右移再左移，如果不等于自己说明最低位是1，表示签到
             signCount++;
         }
         v >>= 1;
     }
     return 0;
 }