package cn.ac.iscas.pebble.dc.redispool;
import java.io.File;
import java.io.FileOutputStream;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Set;
import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisPool;
import redis.clients.jedis.JedisPoolConfig;
import redis.clients.jedis.Pipeline;
import redis.clients.jedis.Response;
import cn.ac.iscas.pebble.dc.hdfs.AppendToFile;
import cn.ac.iscas.pebble.dc.test.P;
public class RedisPoolProxy {
private HashMap<String,JedisPool> poolList = new HashMap<String,JedisPool>();
public static Date yes = null;
private static RedisPoolProxy _proxy = null;
public static RedisPoolProxy getInstance(){
if(null == _proxy){
_proxy = new RedisPoolProxy();
}
return _proxy;
}
public JedisPool getPool(String ip,int port){
JedisPool pool = poolList.get(ip + ":" + port);
if(null == pool){
JedisPoolConfig config = new JedisPoolConfig();
config.setMaxTotal(P.maxTotal);//最大连接数
config.setMaxIdle(P.maxIdle);//最大空闲连接数
config.setMaxWaitMillis(P.maxWaitMillis);//获取连接时的最大等待毫秒数
config.setTestOnBorrow(true);//在获取连接的时候检查有效性, 默认false
pool = new JedisPool(config, ip, port,60*1000);
poolList.put(ip + ":" + port, pool);
}
return pool;
}
public static Integer getDataFromRedis(String ip, int port, int db, String localfilepath,boolean delete){
JedisPool pool = RedisPoolProxy.getInstance().getPool(ip,port);
int count = 0;
boolean flag = true;
if(null != pool){
Jedis redis = null;
try {
redis = pool.getResource();
Pipeline pipeline = redis.pipelined();
pipeline.select(db);
Response<Set<String>> response = pipeline.keys("*");
pipeline.sync();
Set<String> keys = response.get();
if(null != keys){
pipeline.get(key);
pipeline.del(key);
}
List<Object> result = pipeline.syncAndReturnAll();
for(Object value:result){
if(value instanceof String){
System.out.println((String)value+"\n");
count++;
}
}
}catch (Exception e) {
pool.returnBrokenResource(redis);
e.printStackTrace();
flag=false;
} finally {
if (null != redis && null != pool ) {
pool.returnResource(redis);
}
}
}
return count;
}
}