com.plumelog.core.redis.RedisClient Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of plumelog-core Show documentation
There is a newer version: 3.5.3
package com.plumelog.core.redis;


import com.plumelog.core.AbstractClient;
import com.plumelog.core.exception.LogQueueConnectException;
import redis.clients.jedis.*;
import java.util.ArrayList;
import java.util.List;
/**
 * className：RedisClient
 * description：RedisClient instance
 * time：2020-05-11.16:17
 *
 * @author Frank.chen
 * @version 1.0.0
 */
public class RedisClient extends AbstractClient {
    private static  RedisClient instance;
    private  int MAX_ACTIVE = 30;
    private  int MAX_IDLE = 8;
    private  int MAX_WAIT = 1000;
    private  int TIMEOUT = 1000;
    private  boolean TEST_ON_BORROW = true;
    private JedisPool jedisPool = null;

    public static RedisClient getInstance(String host,int port,String pass) {
        if (instance == null) {
            synchronized (RedisClient.class) {
                if (instance == null) {
                    instance = new RedisClient(host, port, pass);
                }
            }
        }
        return instance;
    }
    private RedisClient(String host, int port, String pass){
        JedisPoolConfig config = new JedisPoolConfig();
        config.setMaxTotal(MAX_ACTIVE);
        config.setMaxIdle(MAX_IDLE);
        config.setMaxWaitMillis(MAX_WAIT);
        config.setTestOnBorrow(TEST_ON_BORROW);
        if(pass!=null&&!"".equals(pass)) {
            jedisPool = new JedisPool(config, host, port, TIMEOUT, pass);
        }else{
            jedisPool = new JedisPool(config, host, port, TIMEOUT);
        }
    }
    @Override
    public void pushMessage(String key, String strings) throws LogQueueConnectException {
        Jedis sj=null;
        try {
            sj=jedisPool.getResource();
            sj.rpush(key, strings);
        }catch (Exception e){
         throw new LogQueueConnectException("redis 写入失败！",e);
        }finally {
            if(sj!=null){
               sj.close();
            }
        }

    }
    public String getMessage(String key) {
        Jedis sj=jedisPool.getResource();
        String obj;
        try {
            obj=sj.lpop(key);
        }finally {
            sj.close();
        }
        return obj;
    }
    public void putMessageList(String key, List list){
        Jedis sj=jedisPool.getResource();
        try {
            Pipeline pl=sj.pipelined();
            list.forEach(str->{
                pl.rpush(key,str);
            });
            pl.sync();
        }finally {
            sj.close();
        }

    }
    public List getMessage(String key,int size) throws  LogQueueConnectException{
        Jedis sj=null;
        List list=new ArrayList<>();
        try {
        sj=jedisPool.getResource();
        List> listRes=new ArrayList<>();
            Pipeline pl=sj.pipelined();
            for(int i=0;i res=pl.lpop(key);
                if(res==null){
                    break;
                }
                listRes.add(res);
            }
            pl.sync();
            listRes.forEach(res->{
                String log=res.get();
                if(log!=null) {
                    list.add(log);
                }
            });
        }catch (Exception e){
            throw new LogQueueConnectException("redis 连接失败！",e);
        }finally {
            if(sj!=null) {
                sj.close();
            }
        }
        return list;
    }
}