https://diuut.com/?p=1055 redistemple bitmap操作

import org.springframework.beans.factory.annotation.Autowired; import org.springframework.data.redis.connection.RedisStringCommands; import org.springframework.data.redis.core.RedisCallback; import org.springframework.data.redis.core.RedisTemplate; import org.springframework.stereotype.Component;

import java.util.concurrent.TimeUnit;

/**

@Author Diuut @Date 2020/6/9 11:18 */ @Component public class BitMapUtils { private static RedisTemplate<String,String> redisTemplate; @Autowired public void setRedisTemplate(RedisTemplate redisTemplate) { BitMapUtils.redisTemplate = redisTemplate; } /** 设置key字段第offset位bit数值 @param key 字段 @param offset 位置 @param value 数值 */ public static void setBit(String key, long offset, boolean value) { redisTemplate.execute((RedisCallback) con -> con.setBit(key.getBytes(), offset, value)); } /** 判断该key字段offset位否为1 @param key 字段 @param offset 位置 @return 结果 */ public static boolean getBit(String key, long offset) { return (boolean) redisTemplate.execute((RedisCallback) con -> con.getBit(key.getBytes(), offset)); } /** 统计key字段value为1的总数 @param key 字段 @return 总数 */ public static Long bitCount(String key) { return (Long) redisTemplate.execute((RedisCallback) con -> con.bitCount(key.getBytes())); } /** 统计key字段value为1的总数,从start开始到end结束 @param key 字段 @param start 起始 @param end 结束 @return 总数 */ public static Long bitCount(String key, Long start, Long end) { return (Long) redisTemplate.execute((RedisCallback) con -> con.bitCount(key.getBytes(), start, end)); } /** 取多个key并集并计算总数 @param key key @return 总数 */ public static Long OpOrCount(String… key) { byte[][] keys = new byte[key.length][]; for (int i = 0; i < key.length; i++) { keys[i] = key[i].getBytes(); } redisTemplate.execute((RedisCallback) con -> con.bitOp(RedisStringCommands.BitOperation.OR, (key[0] + “To” + key[key.length - 1]).getBytes(), keys)); redisTemplate.expire(key[0] + “To” + key[key.length - 1], 10, TimeUnit.SECONDS); return BitMapUtils.bitCount(key[0] + “To” + key[key.length - 1]); } /** 取多个key的交集并计算总数 @param key key @return 总数 */ public static Long OpAndCount(String… key) { byte[][] keys = new byte[key.length][]; for (int i = 0; i < key.length; i++) { keys[i] = key[i].getBytes(); } redisTemplate.execute((RedisCallback) con -> con.bitOp(RedisStringCommands.BitOperation.AND, (key[0] + “To” + key[key.length - 1]).getBytes(), keys)); redisTemplate.expire(key[0] + “To” + key[key.length - 1], 10, TimeUnit.SECONDS); return BitMapUtils.bitCount(key[0] + “To” + key[key.length - 1]); } /** 取多个key的补集并计算总数 @param key key @return 总数 */ public static Long OpXorCount(String… key) { byte[][] keys = new byte[key.length][]; for (int i = 0; i < key.length; i++) { keys[i] = key[i].getBytes(); } redisTemplate.execute((RedisCallback) con -> con.bitOp(RedisStringCommands.BitOperation.XOR, (key[0] + “To” + key[key.length - 1]).getBytes(), keys)); redisTemplate.expire(key[0] + “To” + key[key.length - 1], 10, TimeUnit.SECONDS); return BitMapUtils.bitCount(key[0] + “To” + key[key.length - 1]); } /** 取多个key的否集并计算总数 @param key key @return 总数 */ public static Long OpNotCount(String… key) { byte[][] keys = new byte[key.length][]; for (int i = 0; i < key.length; i++) { keys[i] = key[i].getBytes(); } redisTemplate.execute((RedisCallback) con -> con.bitOp(RedisStringCommands.BitOperation.NOT, (key[0] + “To” + key[key.length - 1]).getBytes(), keys)); redisTemplate.expire(key[0] + “To” + key[key.length - 1], 10, TimeUnit.SECONDS); return BitMapUtils.bitCount(key[0] + “To” + key[key.length - 1]); } }