利用Redis BitMap 统计用户活跃指标

    bitMap原理 :

    以下： index 从 0 到 9 ，依次对应到一个bit位上，若是index 表明用户id,bit位上的0 1分表 表明用户是否登陆；

1

0

1

1

0

1

1

0

1

1

0

1

2

3

4

5

6

7

8

9

   

   redis数据结构中 string 类型，包含了对bitmap的实现;在redis-cli中，能够经过setbit getbit 来对bit进行操做；本文经过jedis来对redis进行操做； 

BitSet工具类：实现对经过jedis.get(key)取出的byte[]值与BitSet的转换

public class BitSetUtils {

    /**
     * 将BitSet对象转化为ByteArray
     * @param bitSet
     * @return
     */
    public static byte[] bitSet2ByteArray(BitSet bitSet) {
        byte[] bytes = new byte[bitSet.size() / 8];
        for (int i = 0; i < bitSet.size(); i++) {
            int index = i / 8;
            int offset = 7 - i % 8;
            bytes[index] |= (bitSet.get(i) ? 1 : 0) << offset;
        }
        return bytes;
    }

    /**
     * 
     * @param bytes
     * @return
     */
    public static BitSet byteArray2BitSet(byte[] bytes) {
        BitSet bitSet = new BitSet(bytes.length * 8);
        int index = 0;
        for (int i = 0; i < bytes.length; i++) {
            for (int j = 7; j >= 0; j--) {
                bitSet.set(index++, (bytes[i] & (1 << j)) >> j == 1 ? true
                        : false);
            }
        }
        return bitSet;
    }

}

    具体对java中BitSet操做，见http://my.oschina.net/cloudcoder/blog/294810 ，该篇对bitSet用法介绍很详细；

redis工具类：

public class RedisUtil {
    static {
        initPool();
    }
    private static volatile JedisPool jedisPool;
    private static ResourceBundle resourceBundle;
    public static Jedis getResource() {
        return jedisPool.getResource();
    }
    public static void returnResource(Jedis jedis) {
        jedisPool.returnResource(jedis);
    }
    public static void initPool() {
            if(jedisPool != null){
                return;
            }
            loadProperties();
            String host = resourceBundle.getString("redis.host");
            String passwd= resourceBundle.getString("redis.passwd");
            int port = Integer.parseInt(resourceBundle.getString("redis.port"));
            JedisPoolConfig config = config();
            jedisPool = new JedisPool(config,host,port,60,passwd);
    }
    private static void loadProperties() {
        resourceBundle = ResourceBundle.getBundle("config/redis-config");
    }
    private static JedisPoolConfig config() {
        JedisPoolConfig config = new JedisPoolConfig();
        return config;
    }
    public static void main(String[]args){
       Jedis jedis= RedisUtil.getResource();
       RedisUtil.returnResource(jedis);
    }
}

 1 统计系统中某天用户登陆的状况：以当天日期作为key ,好比 ‘20150410’ ，对应的 bitMap 的 index 用userId来标示，UserId这里用  long 型表示，若是id不是以0开头，能够加上相应的偏移量就OK了；若是该天用户登陆，调用activeUser方法，来更改bitMap相应index上的标示；

public void activeUser(long userId, String dateKey) {
    Jedis jedis= RedisUtil.getResource();
    try{
        jedis.setbit(dateKey,userId,true);
    }finally {
        RedisUtil.returnResource(jedis);
    }
}

若是咱们想统计该天用户登陆的数量，及登陆的用户id,能够经过以下方法实现：

//该天用户总数
public long totalCount(String dateKey) {
    Jedis jedis= RedisUtil.getResource();
    try{
        return jedis.bitcount(dateKey);
    }finally {
        RedisUtil.returnResource(jedis);
    }
}

//该天登陆全部的用户id
public List<Long> activeUserIds(String  dateKey) {
    Jedis jedis= RedisUtil.getResource();
    try{
        
        if(jedis.get(key)==null){
                 return null;
         }
        BitSet set= BitSetUtils.byteArray2BitSet(jedis.get(key).getBytes());
      
        List<Long>list=new ArrayList<Long>();
        for (long i=0;i<set.size();i++){
            if(set.get(i)){
                list.add(i);
            }
        }
        return list;
    }finally {
        RedisUtil.returnResource(jedis);
    }
}

   2 若是咱们想统计n天，连续登陆的用户数，及UserId:

public List<Long> continueActiveUserCount(String... dateKeys) {
    Jedis jedis= RedisUtil.getResource();
    try{
        BitSet all = null;
        for (String key:dateKeys){
             if(jedis.get(key)==null){
                 continue;
             }
             BitSet set= BitSetUtils.byteArray2BitSet(jedis.get(key).getBytes());
             if(all==null){
                 all=set;
             }
             System.out.println(set.size());
             all.and(set);
        }
        List<Long>list=new ArrayList<Long>();
        for (long i=0;i<all.size();i++){
            if(all.get(i)){
                list.add(i);
            }
        }
        return list;
    }finally {
        RedisUtil.returnResource(jedis);
    }
}

 

转载请注明来源：http://my.oschina.net/robinyao/blog/398808