很多应用上都有用户签到的功能,尤其是配合积分系统一起使用。现在有以下需求:
- 签到1天得1积分,连续签到2天得2积分,3天得3积分,3天以上均得3积分等。
- 如果连续签到中断,则重置计数,每月重置计数。
- 显示用户某月的签到次数和首次签到时间。
- 在日历控件上展示用户每月签到,可以切换年月显示。
- …
功能分析
对于用户签到数据,如果直接采用数据库存储,当出现高并发访问时,对数据库压力会很大,例如双十一签到活动。这时候应该采用缓存,以减轻数据库的压力,Redis是高性能的内存数据库,适用于这样的场景。
如果采用String类型保存,当用户数量大时,内存开销就非常大。
如果采用集合类型保存,例如Set、Hash,查询用户某个范围的数据时,查询效率又不高。
Redis提供的数据类型BitMap(位图),每个bit位对应0和1两个状态。虽然内部还是采用String类型存储,但Redis提供了一些指令用于直接操作BitMap,可以把它看作一个bit数组,数组的下标就是偏移量。
它的优点是内存开销小,效率高且操作简单,很适合用于签到这类场景。缺点在于位计算和位表示数值的局限。如果要用位来做业务数据记录,就不要在意value的值。
Redis提供了以下几个指令用于操作BitMap:
命令 | 说明 | 可用版本 | 时间复杂度 |
---|---|---|---|
SETBIT | 对 key 所储存的字符串值,设置或清除指定偏移量上的位(bit)。 |
>= 2.2.0 | O(1) |
GETBIT | 对 key 所储存的字符串值,获取指定偏移量上的位(bit)。 |
>= 2.2.0 | O(1) |
BITCOUNT | 计算给定字符串中,被设置为 1 的比特位的数量。 | >= 2.6.0 | O(N) |
BITPOS | 返回位图中第一个值为 bit 的二进制位的位置。 | >= 2.8.7 | O(N) |
BITOP | 对一个或多个保存二进制位的字符串 key 进行位元操作。 |
>= 2.6.0 | O(N) |
BITFIELD | BITFIELD 命令可以在一次调用中同时对多个位范围进行操作。 |
>= 3.2.0 | O(1) |
考虑到每月要重置连续签到次数,最简单的方式是按用户每月存一条签到数据。Key的格式为 u:sign:{uid}:{yyyMM}
,而Value则采用长度为4个字节的(32位)的BitMap(最大月份只有31天)。BitMap的每一位代表一天的签到,1表示已签,0表示未签。
例如 u:sign:1225:202101
表示ID=1225的用户在2021年1月的签到记录
# 用户1月6号签到
SETBIT u:sign:1225:202101 5 1 # 偏移量是从0开始,所以要把6减1
# 检查1月6号是否签到
GETBIT u:sign:1225:202101 5 # 偏移量是从0开始,所以要把6减1
# 统计1月份的签到次数
BITCOUNT u:sign:1225:202101
# 获取1月份前31天的签到数据
BITFIELD u:sign:1225:202101 get u31 0
# 获取1月份首次签到的日期
BITPOS u:sign:1225:202101 1 # 返回的首次签到的偏移量,加上1即为当月的某一天
示例代码
using StackExchange.Redis;
using System;
using System.Collections.Generic;
using System.Linq;
/**
* 基于Redis Bitmap的用户签到功能实现类
*
* 实现功能:
* 1\. 用户签到
* 2\. 检查用户是否签到
* 3\. 获取当月签到次数
* 4\. 获取当月连续签到次数
* 5\. 获取当月首次签到日期
* 6\. 获取当月签到情况
*/
public class UserSignDemo
{
private IDatabase _db;
public UserSignDemo(IDatabase db)
{
_db = db;
}
/**
* 用户签到
*
* @param uid 用户ID
* @param date 日期
* @return 之前的签到状态
*/
public bool DoSign(int uid, DateTime date)
{
int offset = date.Day - 1;
return _db.StringSetBit(BuildSignKey(uid, date), offset, true);
}
/**
* 检查用户是否签到
*
* @param uid 用户ID
* @param date 日期
* @return 当前的签到状态
*/
public bool CheckSign(int uid, DateTime date)
{
int offset = date.Day - 1;
return _db.Strin