Redis实现统计网站访问人数的功能

        现在想要统计某一网站的累积访问用户人数和日均活跃人数（连续多少天访问该网站的人数），可以通过Redis来实现类似功能。

        我使用的数据结构是Redis中的bitmap，其在大数据量下的空间占用量很小。大概思路就是每一位用户都是bitmap中的一位，为1就代表其访问了，为0就代表没访问。比如说现在有5位用户，第1、3位用户访问了，而2、4、5没访问，如果以索引位置作为其userId的话，那么bitmap存储的就是10100。

        累计用户的key设置为“totalKey”，其值为到今天为止所有用户访问的信息，为1就代表其访问过该网站，为0就代表该用户直到今天都没有访问过该网站；日均活跃人数的key设置为“activeKey:[当前的日期]”，比如说2019年5月31日的日均活跃人数key为“activeKey:20190531”，2019年5月30日的日均活跃人数key为“activeKey:20190530”，等等。所以如果要统计日均活跃人数的话，只要将这几个key做交集就可以了（因为只有都为1，相与后结果才为1，如果有一个为0，相与后结果就不是1），然后统计交集结果的1的个数，结果即为统计值。

        实现代码如下所示，在main函数中模拟了用户访问的情况。在2019年5月31日有userId为0到14一共15个人访问该网站，而在2019年5月30日有userId为6到14一共9个人访问过该网站：

package com.hys.redis;

import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;

import org.apache.commons.lang3.StringUtils;
import org.apache.commons.lang3.time.DateUtils;
import org.apache.commons.pool2.impl.GenericObjectPoolConfig;

import redis.clients.jedis.BitOP;
import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisPool;
import redis.clients.jedis.Pipeline;

/**
 * 统计累计和日均活跃用户人数
 * @author Robert Hou
 * @date 2019年5月31日
 */
public class Counter {

    /**
     * ip地址
     */
    private static final String IP_ADDRESS = "127.0.0.1";
    /**
     * 端口号
     */
    private static final int    PORT       = 6379;
    /**
     * jedis客户端
     */
    private Jedis               jedis;
    /**
     * 累计用户人数key
     */
    private static final String TOTAL_KEY  = "totalKey";
    /**
     * 日均活跃用户人数key
     */
    private static final String ACTIVE_KEY = "activeKey:";

    public Counter() {
        GenericObjectPoolConfig poolConfig = new GenericObjectPoolConfig();
        poolConfig.setMaxTotal(50);
        poolConfig.setMaxIdle(50);
        poolConfig.setMaxWaitMillis(1000);
        JedisPool jedisPool = new JedisPool(poolConfig, IP_ADDRESS, PORT);
        jedis = jedisPool.getResource();
    }

    /**
     * 更新累计和日均活跃用户人数
    * @param userId 用户id
    * @param time 当前日期
     */
    private void updateUser(long userId, String time) {
        if (StringUtils.isBlank(time)) {
            SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMdd");
            time = sdf.format(new Date());
        }
        Pipeline pipeline = jedis.pipelined();
        pipeline.setbit(TOTAL_KEY, userId, true);
        pipeline.setbit(ACTIVE_KEY + time, userId, true);
        pipeline.syncAndReturnAll();
    }

    /**
     * 获取累计用户人数
    * @return 累计用户人数
     */
    private Long getTotalUserCount() {
        Pipeline pipeline = jedis.pipelined();
        pipeline.bitcount(TOTAL_KEY);
        List<Object> totalKeyCountList = pipeline.syncAndReturnAll();
        return (Long) totalKeyCountList.get(0);
    }

    /**
     * 获取指定天数内的日均活跃人数
    * @param dayNum 指定天数
    * @return 日均活跃人数
     */
    private Long getActiveUserCount(int dayNum) {
        if (dayNum < 1) {
            return (long) 0;
        }
        List<String> pastDaysKey = new ArrayList<>();
        SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMdd");
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < dayNum; i++) {
            //保存距今dayNum天数的key的集合
            sb.append(ACTIVE_KEY).append(sdf.format(DateUtils.addDays(new Date(), -i)));
            pastDaysKey.add(sb.toString());
            sb.delete(0, sb.length());
        }
        if (pastDaysKey.isEmpty()) {
            return (long) 0;
        }
        String lastDaysKey = "last" + dayNum + "DaysActive";
        Pipeline pipeline = jedis.pipelined();
        pipeline.bitop(BitOP.AND, lastDaysKey, pastDaysKey.toArray(new String[pastDaysKey.size()]));
        pipeline.bitcount(lastDaysKey);
        //设置过期时间为5分钟
        pipeline.expire(lastDaysKey, 300);
        List<Object> activeKeyCountList = pipeline.syncAndReturnAll();
        return (Long) activeKeyCountList.get(1);
    }

    public static void main(String[] args) {
        Counter c = new Counter();
        //这里假设当前日期为2019年5月31日，测试的时候需要更改为当前日期的前几天
        for (int i = 0; i < 15; i++) {
            c.updateUser(i, "20190531");
        }        
        for (int i = 6; i < 15; i++) {
            c.updateUser(i, "20190530");
        }        
        System.out.println("累计用户数：" + c.getTotalUserCount());
        System.out.println("两天内的活跃人数：" + c.getActiveUserCount(2));
    }
}

        运行结果如下所示：

累计用户数：15
两天内的活跃人数：9

        上述使用的是Java方式连接的Redis，如果使用的是Spring Boot，可以使用spring-boot-starter-data-redis的方式进行连接。GitHub源码：https://github.com/ACoolMonkey/redis-counter