SpringBoot之Redis使用Bitmap实现登录、签到的学习

Bitmap 的底层数据结构用的是 String 类型的 SDS 数据结构来保存位数组，Redis 把每个字节数组的 8 个 bit 位利用起来，每个 bit 位 表示一个元素的二值状态（不是 0 就是 1）。

8 个 bit 组成一个 Byte，所以 Bitmap 会极大地节省存储空间。 这就是 Bitmap 的优势。

一、实现判断用户登录状态

1.引入redis的依赖

<!--redis-->
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-data-redis</artifactId>
</dependency>

Bitmap 提供了 GETBIT、SETBIT 操作，通过一个偏移值 offset 对 bit 数组的 offset 位置的 bit 位进行读写操作，需要注意的是 offset 从 0 开始。

只需要一个 key = login_status 表示存储用户登陆状态集合数据， 将用户 ID 作为 offset，在线就设置为 1，下线设置 0。通过 GETBIT判断对应的用户是否在线。

2.创建一个控制器测试

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

/**
 * @author qinxun
 * @date 2023-04-28
 * @Descripion: Bitmap学习
 */
@RestController
public class BitmapController {

    @Autowired
    private RedisTemplate redisTemplate;

    /**
     * 用户登录
     *
     * @param userId 用户ID
     */
    @GetMapping("/login")
    public String toSign(Long userId) {
        redisTemplate.opsForValue().setBit("login_status", userId, true);
        return "用户ID:" + userId + ",登录成功";
    }

    /**
     * 退出登录
     *
     * @param userId 用户ID
     */
    @GetMapping("/logout")
    public String toLogout(Long userId) {
        redisTemplate.opsForValue().setBit("login_status", userId, false);
        return "用户ID:" + userId + ",退出登录成功";
    }

    /**
     * 获取用户登录数据
     *
     * @param userId 用户ID
     */
    @GetMapping("/getLoginData")
    public String getSignData(Long userId) {
        Boolean status = redisTemplate.opsForValue().getBit("login_status", userId);
        return Boolean.TRUE.equals(status) ? "用户ID:" + userId + ",已登录" : "用户ID:" + userId + ",未登录";
    }

}

3.测试

对用户ID为1的用户登录操作

 

 

获取用户登录数据

用户ID为1 的用户退出登录

 

 再次获取用户登录数据

 发现用户ID为1的用户 已经退出登录了

 二.用户每个月的签到情况

1.创建一个时间处理工具类

/**
 * @author qinxun
 * @date 2023-04-28
 * @Descripion: 时间工具类
 */
public class DateUtil {

    /**
     * 获取年月数据
     *
     * @return 年和月数据
     */
    public static String getYearAndMonth() {
        SimpleDateFormat dateFormat = new SimpleDateFormat("yyyy-MM");
        return dateFormat.format(new Date());
    }

    /**
     * 获取当前数据天的数据
     *
     * @return 天
     */
    public static long getDay() {
        SimpleDateFormat dateFormat = new SimpleDateFormat("dd");
        return Long.parseLong(dateFormat.format(new Date()));
    }
}

2.创建签到控制层

/**
 * @author qinxun
 * @date 2023-04-28
 * @Descripion: 签到相关
 */
@RestController
public class SignController {

    @Autowired
    private StringRedisTemplate redisTemplate;

    /**
     * 用户每日签到
     *
     * @param userId 用户ID
     */
    @GetMapping("/sign")
    public String sign(Long userId) {
        // key:用户ID+年月数据
        redisTemplate.opsForValue().setBit(userId + ":" + DateUtil.getYearAndMonth(), DateUtil.getDay(), true);
        return "用户ID:" + userId + ",今日签到成功";
    }


    /**
     * 获取用户每日签到数据
     *
     * @param userId 用户ID
     */
    @GetMapping("/getSignData")
    public String getSignData(Long userId) {
        // key:用户ID+年月数据
        Boolean status = redisTemplate.opsForValue().getBit(userId + ":" + DateUtil.getYearAndMonth(), DateUtil.getDay());
        return Boolean.TRUE.equals(status) ? "用户ID:" + userId + ",今日已签到" : "用户ID:" + userId + ",今日未签到";
    }

    /**
     * 统计用户每个月的打卡次数
     *
     * @param userId 用户ID
     */
    @GetMapping("/getCalcData")
    public String getCalcData(Long userId) {
        String key = userId + ":" + DateUtil.getYearAndMonth();
        long nums = (long) redisTemplate.execute((RedisCallback) con -> con.bitCount(key.getBytes()));
        return "用户ID:" + userId + ",本月打卡次数为:" + nums;
    }

    /**
     * 获取用户本月首次打卡日期
     * @param userId 用户ID
     */
    @GetMapping("/getFirstSignData")
    public String getFirstSignData(Long userId){
        String key = userId + ":" + DateUtil.getYearAndMonth();
        Object date = redisTemplate.execute((RedisCallback<Object>) connection -> connection.bitPos(key.getBytes(), true));
        System.out.println(date);
        return "首次打卡日期为:"+date;
    }

}

签到操作

获取用户签到情况

 获取用户本月签到数据

 获取用户这个月首次打卡时间

 

 

小结:比如用户是否存在、 ip 是否是黑名单、以及签到打卡统计等场景就可以考虑使用 Bitmap。

只需要一个 bit 位就能表示 0 和 1。在统计海量数据的时候将大大减少内存占用。