BaseRedis

import com.iba.frame.redis.PojoSerializable;
import com.iba.frame.util.CommonUtil;
import org.apache.log4j.Logger;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.dao.DataAccessException;
import org.springframework.data.redis.connection.RedisConnection;
import org.springframework.data.redis.core.RedisCallback;
import org.springframework.data.redis.core.RedisConnectionUtils;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.serializer.Jackson2JsonRedisSerializer;
import org.springframework.data.redis.serializer.RedisSerializer;
import org.springframework.stereotype.Repository;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import java.util.Set;

@Repository("baseRedis")
public abstract class BaseRedis {

    // 加载数据的时候每页加载1000条,免得数据库超时
    protected final Integer LOAD_PAGE_SIZE = 1000;
    @Autowired
    protected RedisTemplate redisTemplate;
    @Autowired
    protected RedisTemplate transRedisTemplate;
    private Logger logger = Logger.getLogger(BaseRedis.class);

    // @Autowired
    // protected Jedis jedis;

    /////////////////////////////////////////////////////////////////
    /////////////////////////// 普通通用接口////////////////////////////
    /////////////////////////////////////////////////////////////////

    /**
     * 设置一个结构数据
     *
     * @param key
     * @param value
     * @param seconds
     * @return
     */
    protected Long incr(final String key) {
        Long result = redisTemplate.execute(new RedisCallback() {
            @Override
            public Long doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                return connection.incr(byteKey);
            }
        });
        return result;
    }

    /**
     * 设置一个结构数据
     *
     * @param key
     * @param value
     * @param seconds
     * @return
     */
    protected Long decr(final String key) {
        Long result = redisTemplate.execute(new RedisCallback() {
            @Override
            public Long doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                return connection.decr(byteKey);
            }
        });
        return result;
    }

    /**
     * 设置一个结构数据
     *
     * @param key
     * @param value
     * @param seconds
     * @return
     */
    protected boolean set(final String key, final String value, final long seconds) {
        boolean result = transRedisTemplate.execute(new RedisCallback() {
            @Override
            public Boolean doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                byte[] byteValue = serializer.serialize(value);
                if (seconds > 0) {
                    connection.setEx(byteKey, seconds, byteValue);
                } else {
                    connection.set(byteKey, byteValue);
                }
                return true;
            }
        });
        return result;
    }

    /**
     * @param key
     * @param value
     * @param seconds
     * @param suppressTran 是否压制事务,true,表示在事务环境中立即执行(不进redis队列)
     * @return
     */
    protected boolean set(final String key, final String value, final long seconds, final boolean suppressTran) {
        RedisTemplate realTemp = suppressTran ? redisTemplate : transRedisTemplate;
        boolean result = realTemp.execute(new RedisCallback() {
            @Override
            public Boolean doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                byte[] byteValue = serializer.serialize(value);
                if (seconds > 0) {
                    connection.setEx(byteKey, seconds, byteValue);
                } else {
                    connection.set(byteKey, byteValue);
                }
                return true;
            }
        });
        return result;
    }

    /**
     * 设置一个结构数据(主要用于保存bean对象,初始化incr值请使用String类型的方法)
     *
     * @param key
     * @param value
     * @param seconds
     * @return
     */
    protected boolean set(final String key, final Serializable value, final long seconds) {
        boolean result = transRedisTemplate.execute((RedisCallback) connection -> {
            if(transRedisTemplate.getValueSerializer() instanceof  Jackson2JsonRedisSerializer){
                Jackson2JsonRedisSerializer valueSerializer = (Jackson2JsonRedisSerializer)transRedisTemplate.getValueSerializer();
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                byte[] byteValue = valueSerializer.serialize(value);
                if (seconds > 0) {
                    connection.setEx(byteKey, seconds, byteValue);
                } else {
                    connection.set(byteKey, byteValue);
                }
            }else{
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                byte[] byteValue = CommonUtil.transObj2ByteArray(value);
                if (seconds > 0) {
                    connection.setEx(byteKey, seconds, byteValue);
                } else {
                    connection.set(byteKey, byteValue);
                }
            }

            return true;
        });
        return result;
    }

    /**
     * 根据Key获取对象
     *
     * @param key
     * @return
     */
    protected Object get(final String key) {
        Object result = redisTemplate.execute((RedisCallback) connection -> {
            RedisSerializer serializer = getRedisSerializer();
            byte[] byteKey = serializer.serialize(key);
            byte[] value = connection.get(byteKey);
            if (value == null) {
                return null;
            }
            Object dataObj = serializer.deserialize(value);
            return dataObj;
        });
        return result;
    }

    /**
     * 根据Key获取对象
     *
     * @param key
     * @return
     */
    protected  T get(final String key, final Class clazz) {
        T result = redisTemplate.execute((RedisCallback) connection -> {
            RedisSerializer serializer = getRedisSerializer();
            byte[] byteKey = serializer.serialize(key);
            byte[] value = connection.get(byteKey);
            if (value == null) {
                return null;
            }
            if(transRedisTemplate.getValueSerializer() instanceof  Jackson2JsonRedisSerializer){
                Jackson2JsonRedisSerializer valueSerializer = (Jackson2JsonRedisSerializer)transRedisTemplate.getValueSerializer();
                Object dataObj = valueSerializer.deserialize(value);
                System.out.println("dataObj1::"+dataObj);
                return (T)dataObj;
            }else{
                T dataObj = CommonUtil.transByteArray2Obj(value, clazz);
                System.out.println("dataObj1::"+dataObj);
                return dataObj;
            }

        });
        return result;
    }

    protected Object mget(final List keys) {
        Object result = redisTemplate.execute(new RedisCallback() {
            @Override
            public List doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                List listByteKeys = new ArrayList();
                for (String key : keys) {
                    byte[] byteKey = serializer.serialize(key);
                    listByteKeys.add(byteKey);
                }

                List listByteValues = connection.mGet(listByteKeys.toArray(new byte[listByteKeys.size()][]));
                List listValue = new ArrayList();
                for (byte[] value : listByteValues) {
                    listValue.add(serializer.deserialize(value));
                }
                return listValue;
            }
        });
        return result;
    }

    /**
     * @param argKey
     * @return
     */
    protected Object keys(final String argKey) {
        Object result = redisTemplate.execute(new RedisCallback() {
            @Override
            public List doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(argKey);
                Set setBytes = connection.keys(byteKey);
                List listValue = new ArrayList();
                for (byte[] value : setBytes) {
                    listValue.add(serializer.deserialize(value));
                }
                return listValue;
            }
        });
        return result;
    }

    protected  List mget(final List keys, final Class clazz) {
        List result = redisTemplate.execute(new RedisCallback>() {
            @Override
            public List doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                List listByteKeys = new ArrayList();
                for (String key : keys) {
                    byte[] byteKey = serializer.serialize(key);
                    listByteKeys.add(byteKey);
                }

                List listByteValues = connection.mGet(listByteKeys.toArray(new byte[listByteKeys.size()][]));
                List listValue = new ArrayList();
                for (byte[] value : listByteValues) {
                    listValue.add(CommonUtil.transByteArray2Obj(value, clazz));
                }
                return listValue;
            }
        });
        return result;
    }

    protected boolean exists(final String key) {
        boolean result = redisTemplate.execute(new RedisCallback() {
            @Override
            public Boolean doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                return connection.exists(byteKey);
            }
        });
        return result;
    }

    protected Long ttl(final String key) {
        Long result = redisTemplate.execute(new RedisCallback() {
            @Override
            public Long doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);

                return connection.ttl(byteKey);
            }
        });
        return result;
    }

    /////////////////////////////////////////////////////////////////
    /////////////////////////// Set 操作接口 //////////////////////////
    /////////////////////////////////////////////////////////////////
    protected boolean sadd(final String key, final String value) {
        boolean result = transRedisTemplate.execute(new RedisCallback() {
            @Override
            public Boolean doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                byte[] byteValue = serializer.serialize(value);
                connection.sAdd(byteKey, byteValue);
                return true;
            }
        });
        return result;
    }

    /**
     * 仅限于基本对象使用,Bean对象禁止使用
     *
     * @param key
     * @param value
     * @return
     */
    protected boolean sadd(final String key, final Serializable value) {
        boolean result = transRedisTemplate.execute(new RedisCallback() {
            @Override
            public Boolean doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                byte[] byteValue = CommonUtil.transObj2ByteArray(value);
                connection.sAdd(byteKey, byteValue);
                return true;
            }
        });
        return result;
    }

    protected boolean sadd(final String key, final List values) {
        boolean result = transRedisTemplate.execute(new RedisCallback() {
            @Override
            public Boolean doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                List byteValues = new ArrayList(values.size());
                for (String v : values) {
                    byteValues.add(serializer.serialize(v));
                }
                connection.sAdd(byteKey, byteValues.toArray(new byte[values.size()][]));

                return true;
            }
        });
        return result;
    }

    protected boolean sadd(final String key, final Set values) {
        boolean result = transRedisTemplate.execute(new RedisCallback() {
            @Override
            public Boolean doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                byte[][] byteValues = new byte[values.size()][];
                int i = 0;
                for (Serializable v : values) {
                    byteValues[i++] = CommonUtil.transObj2ByteArray(v);
                }
                connection.sAdd(byteKey, byteValues);
                return true;
            }
        });
        return result;
    }

    /**
     * 增加sadd类型
     *
     * @param key
     * @param values
     * @return
     */
    protected boolean sadd(final String key, final Collection values) {
        boolean result = transRedisTemplate.execute(new RedisCallback() {
            @Override
            public Boolean doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                byte[][] byteValues = new byte[values.size()][];
                int i = 0;
                for (Serializable v : values) {
                    byteValues[i++] = CommonUtil.transObj2ByteArray(v);
                }
                connection.sAdd(byteKey, byteValues);
                return true;
            }
        });
        return result;
    }

    protected Long scard(final String key) {
        Object result = redisTemplate.execute(new RedisCallback() {
            @Override
            public Long doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                Long count = connection.sCard(byteKey);
                return count;
            }
        });
        return Long.parseLong(String.valueOf(result));
    }

    @SuppressWarnings("rawtypes")
    protected Object smembers(final String key) {
        Object result = redisTemplate.execute(new RedisCallback() {
            @Override
            public Object doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);

                Set setValue = connection.sMembers(byteKey);

                Iterator iter = setValue.iterator();

                List listObj = new ArrayList();
                while (iter.hasNext()) {
                    Object obj = serializer.deserialize((byte[]) iter.next());
                    listObj.add(obj);
                }
                return listObj;
            }
        });
        return result;
    }

    protected  Set smembers(final String key, final Class clazz) {
        Set result = redisTemplate.execute(new RedisCallback>() {
            @Override
            public Set doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                Set setValue = connection.sMembers(byteKey);
                Iterator iter = setValue.iterator();
                Set result = new HashSet();
                while (iter.hasNext()) {
                    T tmp = CommonUtil.transByteArray2Obj(iter.next(), clazz);
                    result.add(tmp);
                }
                if (result.size() == 0) {
                    return null;
                }
                return result;
            }
        });
        return result;
    }

    protected Set smembersString(final String key) {
        Set result = redisTemplate.execute(new RedisCallback>() {
            @Override
            public Set doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                Set setValue = connection.sMembers(byteKey);
                Iterator iter = setValue.iterator();
                Set setStr = new HashSet();
                while (iter.hasNext()) {
                    String obj = serializer.deserialize(iter.next());
                    setStr.add(obj);
                }
                return setStr;
            }
        });
        return result;
    }

    /**
     * 移出set中指定元素,在事务中无法返回结果,若业务需要判断移出是否成功,请使用smember判断
     *
     * @param key
     * @param value
     * @return
     */
    protected boolean srem(final String key, final String value) {
        boolean result = transRedisTemplate.execute(new RedisCallback() {
            @Override
            public Boolean doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                byte[] byteValue = serializer.serialize(value);
                Long index = connection.sRem(byteKey, byteValue);
                return index != null && index > 0L;
            }
        });
        return result;
    }

    protected boolean srem(final String key, final Serializable value) {
        boolean result = transRedisTemplate.execute(new RedisCallback() {
            @Override
            public Boolean doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                byte[] byteValue = CommonUtil.transObj2ByteArray(value);
                Long index = connection.sRem(byteKey, byteValue);
                return index != null && index > 0L;
            }
        });
        return result;

    }

    protected boolean srem(final String key, final List values) {
        boolean result = transRedisTemplate.execute(new RedisCallback() {
            @Override
            public Boolean doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                byte[][] byteValues = new byte[values.size()][];
                for (int i = 0; i < values.size(); i++) {
                    byteValues[i] = serializer.serialize(values.get(i));
                }
                Long index = connection.sRem(byteKey, byteValues);
                return true;
            }
        });
        return result;

    }

    /////////////////////////////////////////////////////////////////
    /////////////////////////// ZSet接口//////////////////////////////
    /////////////////////////////////////////////////////////////////
    protected boolean zadd(final String key, final Double score, final String value) {
        boolean result = transRedisTemplate.execute(new RedisCallback() {
            @Override
            public Boolean doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                byte[] byteValue = serializer.serialize(value);
                connection.zAdd(byteKey, score, byteValue);
                return true;
            }
        });
        return result;
    }

    protected boolean zrem(final String key, final String value) {
        boolean result = transRedisTemplate.execute(new RedisCallback() {
            @Override
            public Boolean doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                byte[] byteValue = serializer.serialize(value);
                connection.zRem(byteKey, byteValue);
                return true;
            }
        });
        return result;
    }

    protected Object zrange(final String key, final Long begin, final Long end) {
        Object result = redisTemplate.execute(new RedisCallback() {
            @Override
            public Object doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);

                Set setValue = connection.zRange(byteKey, begin, end);
                Iterator iter = setValue.iterator();
                List listObj = new ArrayList();
                while (iter.hasNext()) {
                    Object obj = serializer.deserialize(iter.next());
                    listObj.add(obj);
                }
                return listObj;
            }
        });
        return result;
    }

    @SuppressWarnings("rawtypes")
    protected Object zrangebyscore(final String key, final Double begin, final Double end) {
        Object result = redisTemplate.execute(new RedisCallback() {
            @Override
            public Object doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);

                Set setValue = connection.zRangeByScore(byteKey, begin, end);
                Iterator iter = setValue.iterator();
                List listObj = new ArrayList();
                while (iter.hasNext()) {
                    Object obj = serializer.deserialize((byte[]) iter.next());
                    listObj.add(obj);
                }
                return listObj;
            }
        });
        return result;
    }

    /////////////////////////////////////////////////////////////////
    /////////////////////////// HSet接口//////////////////////////////
    /////////////////////////////////////////////////////////////////
    protected boolean hset(final String key, final String field, final String value, final Long seconds) {
        boolean result = transRedisTemplate.execute(new RedisCallback() {
            @Override
            public Boolean doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                byte[] byteField = serializer.serialize(field);
                byte[] byteValue = serializer.serialize(value);
                connection.hSet(byteKey, byteField, byteValue);
                if (seconds.equals(0L)) {
                    connection.persist(byteKey);
                } else {
                    connection.expire(byteKey, seconds);// 设置过期时间
                }
                return true;
            }
        });
        return result;
    }

    protected boolean hset(final String key, final String field, final Serializable value, final Long seconds) {
        boolean result = transRedisTemplate.execute(new RedisCallback() {
            @Override
            public Boolean doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                byte[] byteField = serializer.serialize(field);
                byte[] byteValue = CommonUtil.transObj2ByteArray(value);
                connection.hSet(byteKey, byteField, byteValue);
                if (seconds.equals(0L)) {
                    connection.persist(byteKey);
                } else {
                    connection.expire(byteKey, seconds);// 设置过期时间
                }
                return true;
            }
        });
        return result;
    }

    protected Object hget(final String key, final String field) {
        Object result = redisTemplate.execute(new RedisCallback() {
            @Override
            public Object doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                byte[] byteField = serializer.serialize(field);
                byte[] value = connection.hGet(byteKey, byteField);
                if (value == null) {
                    logger.error("未找到redis中保存的数据" + new String(byteKey));
                    return null;
                }
                Object dataObj = serializer.deserialize(value);
                return dataObj;
            }
        });
        return result;
    }

    protected  T hget(final String key, final String field, final Class clazz) {
        T result = redisTemplate.execute(new RedisCallback() {
            @Override
            public T doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                byte[] byteField = serializer.serialize(field);
                byte[] value = connection.hGet(byteKey, byteField);
                if (value == null) {

                    return null;
                }
                T dataObj = CommonUtil.transByteArray2Obj(value, clazz);
                return dataObj;
            }
        });
        return result;
    }

    protected Long hincrBy(final String key, final String field, Long incrBy) {
        Long result = redisTemplate.execute(new RedisCallback() {
            @Override
            public Long doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                byte[] byteField = serializer.serialize(field);
                Long value = connection.hIncrBy(byteKey, byteField, incrBy);
                if (value == null) {
                    return 0L;
                }
                return value;
            }
        });
        return result;
    }

    protected Double hincrFloatBy(final String key, final String field, Double incrBy) {
        Double result = redisTemplate.execute(new RedisCallback() {
            @Override
            public Double doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                byte[] byteField = serializer.serialize(field);
                Double value = connection.hIncrBy(byteKey, byteField, incrBy);
                if (value == null) {
                    return 0.0;
                }
                return value;
            }
        });
        return result;
    }

    @SuppressWarnings("rawtypes")
    protected Object hkeys(final String key) {
        Object result = redisTemplate.execute(new RedisCallback() {
            @Override
            public Object doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);

                Set setValue = connection.hKeys(byteKey);

                Iterator iter = setValue.iterator();

                List listObj = new ArrayList();
                while (iter.hasNext()) {
                    Object obj = serializer.deserialize((byte[]) iter.next());
                    listObj.add(obj);
                }
                return listObj;
            }
        });
        return result;
    }

    protected Set hkeysSerial(final String key) {
        Set result = redisTemplate.execute(new RedisCallback>() {
            @Override
            public Set doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);

                Set setValue = connection.hKeys(byteKey);

                Iterator iter = setValue.iterator();

                Set keys = new HashSet();
                while (iter.hasNext()) {
                    String obj = serializer.deserialize(iter.next());
                    keys.add(obj);
                }
                return keys;
            }
        });
        return result;
    }

    protected boolean hdel(final String key, final String field) {
        boolean result = transRedisTemplate.execute(new RedisCallback() {
            @Override
            public Boolean doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                byte[] byteField = serializer.serialize(field);
                connection.hDel(byteKey, byteField);
                return true;
            }
        });
        return result;
    }

    /**
     * 设置redis键过期时间
     *
     * @param key
     * @param seconds
     */
    protected boolean expire(final String key, final Long seconds) {
        boolean result = transRedisTemplate.execute(new RedisCallback() {
            @Override
            public Boolean doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                connection.expire(byteKey, seconds);// 设置过期时间
                return true;
            }
        });
        return result;
    }

    /**
     * @param keys
     */
    public void delete(List keys) {
        transRedisTemplate.delete(keys);
    }

    /**
     * 根据条件删除分页查询结果。
     */
    protected void deleteHashSetByPage(K hsetKey) {
        List listKeys = new ArrayList(1);
        listKeys.add(hsetKey);
        delete(listKeys);
    }

    /**
     * @param key
     * @param suppressTran 是否压制事务,true,表示在立即执行到redis(不需要等事务提交,同时也不会因事务失败而回滚)
     */
    public void delete(K key, final boolean suppressTran) {
        RedisTemplate realTemp = suppressTran ? redisTemplate : transRedisTemplate;
        realTemp.delete(key);
    }

    /**
     * 将 key 的值设为 value ,当且仅当 key 不存在, 若给定的 key 已经存在,则 SETNX 不做任何动作。
     *
     * @param key
     * @param value
     * @param seconds
     * @param suppressTran 是否压制事务,true,表示在立即执行到redis(不需要等事务提交,同时也不会因事务失败而回滚)
     * @return
     */
    protected Boolean setNx(final String key, final Serializable value, final long seconds, final boolean suppressTran) {
        RedisTemplate realTemp = suppressTran ? redisTemplate : transRedisTemplate;
        boolean result = realTemp.execute(new RedisCallback() {
            @Override
            public Boolean doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                byte[] byteValue = CommonUtil.transObj2ByteArray(value);
                Boolean result = connection.setNX(byteKey, byteValue);
                if (seconds > 0 && (null == result && !suppressTran) || (result && suppressTran)) {
                    connection.expire(byteKey, seconds);
                }
                return result;
            }
        });
        return result;
    }

    /**
     * @param keys
     */
    public Long dbSize() {
        Long result = transRedisTemplate.execute(new RedisCallback() {
            @Override
            public Long doInRedis(RedisConnection connection) throws DataAccessException {
                return connection.dbSize();
            }
        });
        return result;
    }

    // public Set keys(K pattern) {
    // return redisTemplate.keys(pattern);
    // }

    public void clean(K pattern) {
        Set keySet = transRedisTemplate.keys(pattern);
        if (keySet != null && keySet.size() > 0) {
            List keyList = new ArrayList(keySet);
            delete(keyList);
        }
    }

    /**
     * pipeline方式获取数据
     *
     * @param keys
     * @return
     */
    protected List pipelineGet(final List keys) {
        return redisTemplate.executePipelined(new RedisCallback() {

            @Override
            public Object doInRedis(RedisConnection connection) throws DataAccessException {
                // 直接执行所有需要执行的命令,返回值由jedis通过底层connection.closePipeline()实现
                RedisSerializer serializer = getRedisSerializer();
                for (String key : keys) {
                    byte[] byteKey = serializer.serialize(key);
                    connection.get(byteKey);
                }
                // jedis api限制 只能返回null 否则会抛异常
                return null;
            }

        }, getRedisSerializer());

    }

    protected  List pipelineGet(final Collection keys, final Class clazz) {
        List tmps = redisTemplate.executePipelined(new RedisCallback() {

            @Override
            public Object doInRedis(RedisConnection connection) throws DataAccessException {
                // 直接执行所有需要执行的命令,返回值由jedis通过底层connection.closePipeline()实现
                RedisSerializer serializer = getRedisSerializer();
                for (String key : keys) {
                    byte[] byteKey = serializer.serialize(key);
                    connection.get(byteKey);
                }
                // jedis api限制 只能返回null 否则会抛异常
                return null;
            }

        }, getPojoSerializer(clazz));
        List result = new ArrayList(tmps.size());
        for (Object obj : tmps) {
            result.add(clazz.cast(obj));
        }
        return result;
    }

    /////////////////////////////////////////////////////////////////
    /////////////////////////// List 操作接口//////////////////////////
    /////////////////////////////////////////////////////////////////

    /**
     * 使用redis自身事物,向一个定长的列表加入元素,每次都要传入list的长度。即外部可以不用放入事务中。
     *
     * @param key
     * @param length
     * @param value
     * @param seconds
     * @return
     */
    protected boolean lpushWithTrim(final String key, final long length, final Serializable value, final Long seconds) {
        boolean result = redisTemplate.execute(new RedisCallback() {
            @Override
            public Boolean doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                byte[] byteValue = CommonUtil.transObj2ByteArray(value);
                if (length > 0L) {
                    connection.multi();
                    connection.lPush(byteKey, byteValue);
                    connection.lTrim(byteKey, 0, length - 1);
                    connection.exec();
                } else {
                    connection.lPush(byteKey, byteValue);
                }
                if (seconds.equals(0L)) {
                    connection.persist(byteKey);
                } else {
                    connection.expire(byteKey, seconds);// 设置过期时间
                }
                return true;
            }
        });
        return result;
    }

    /**
     * @param key
     * @param value
     * @param seconds
     * @return
     */
    protected boolean lpush(final String key, final Serializable value, final Long seconds) {
        boolean result = transRedisTemplate.execute(new RedisCallback() {
            @Override
            public Boolean doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                byte[] byteValue = CommonUtil.transObj2ByteArray(value);
                connection.lPush(byteKey, byteValue);
                if (seconds.equals(0L)) {
                    connection.persist(byteKey);
                } else {
                    connection.expire(byteKey, seconds);// 设置过期时间
                }
                return true;
            }
        });
        return result;
    }

    /**
     * @param key
     * @param begin
     * @param end
     * @return
     */
    protected boolean ltrim(final String key, long begin, long end) {
        boolean result = transRedisTemplate.execute(new RedisCallback() {
            @Override
            public Boolean doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                connection.multi();
                connection.lTrim(byteKey, begin, end);
                connection.exec();
                return true;
            }
        });
        return result;
    }

    /**
     * @param key
     * @param begin
     * @param end
     * @return
     */
    protected  List lrange(final String key, long begin, long end, final Class clazz) {
        List result = redisTemplate.execute(new RedisCallback>() {
            @Override
            public List doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                List listByteValues = connection.lRange(byteKey, begin, end);
                List listValue = new ArrayList(listByteValues.size());
                for (byte[] value : listByteValues) {
                    listValue.add(CommonUtil.transByteArray2Obj(value, clazz));
                }
                return listValue;
            }
        });
        return result;
    }

    /////////////////////////////////////////////////////////////
    /////////////////////// Pipeline 方式 /////////////////////////
    /////////////////////////////////////////////////////////////
    // 用pipeline的方式实现get多个数据
    protected List pget(final List keys) {

        @SuppressWarnings("unchecked")
        List result = (List) redisTemplate.execute(new RedisCallback() {
            @Override
            public List doInRedis(RedisConnection connection) throws DataAccessException {
                connection.openPipeline();

                RedisSerializer serializer = getRedisSerializer();
                for (String key : keys) {
                    byte[] byteKey = serializer.serialize(key);
                    connection.get(byteKey);
                }
                List pipeResult = connection.closePipeline();
                List strObjs = new ArrayList();
                for (Object tmp : pipeResult) {
                    byte[] bytes = (byte[]) tmp;
                    String tmpObj = serializer.deserialize(bytes);
                    strObjs.add(tmpObj);
                }
                return strObjs;
            }
        });

        return result;
    }

    /**
     * 判断 value 元素是否集合(set) key 的成员
     *
     * @param key
     * @param value
     * @return
     */
    protected boolean sIsMember(final String key, final String value) {
        boolean result = redisTemplate.execute(new RedisCallback() {
            @Override
            public Boolean doInRedis(RedisConnection connection) throws DataAccessException {
                RedisSerializer serializer = getRedisSerializer();
                byte[] byteKey = serializer.serialize(key);
                byte[] byteVal = serializer.serialize(value);
                return connection.sIsMember(byteKey, byteVal);
            }
        });
        return result;
    }

    /**
     * 设置RedisTemplate 可以注入,不需要显示调用
     *
     * @return
     */
    protected RedisTemplate getRedisTemplate() {
        return transRedisTemplate;
    }

    /**
     * @param redisTemplate
     */
    public void setRedisTemplate(RedisTemplate redisTemplate) {
        this.transRedisTemplate = redisTemplate;
    }

    /**
     * @return
     */
    protected RedisSerializer getRedisSerializer() {

        return transRedisTemplate.getStringSerializer();
    }

    protected  RedisSerializer getPojoSerializer(Class clazz) {
        return new PojoSerializable(clazz);
    }

    public void closeConn() {
        RedisConnection conn = RedisConnectionUtils.getConnection(transRedisTemplate.getConnectionFactory());
        conn.close();
    }

}







你可能感兴趣的:(BaseRedis)