package com.caspar.util; 
import java.util.concurrent.locks.ReentrantLock;  
import org.apache.log4j.Logger; 
import redis.clients.jedis.Jedis; 
import redis.clients.jedis.JedisPool; 
import redis.clients.jedis.JedisPoolConfig;  
/** 
 * Redis 工具類 
 * @author caspar 
 * 
 */ 
public class RedisUtil { 
   
  protected static ReentrantLock lockPool = new ReentrantLock(); 
  protected static ReentrantLock lockJedis = new ReentrantLock(); 
   
  protected static Logger logger = Logger.getLogger(RedisUtil.class); 
   
  //Redis服務(wù)器IP 
  private static String ADDR_ARRAY = FileUtil.getPropertyValue("/properties/redis.properties", "server"); 
   
  //Redis的端口號(hào) 
  private static int PORT = FileUtil.getPropertyValueInt("/properties/redis.properties", "port"); 
   
  //訪問密碼 
//  private static String AUTH = FileUtil.getPropertyValue("/properties/redis.properties", "auth"); 
   
  //可用連接實(shí)例的最大數(shù)目，默認(rèn)值為8； 
  //如果賦值為-1，則表示不限制；如果pool已經(jīng)分配了maxActive個(gè)jedis實(shí)例，則此時(shí)pool的狀態(tài)為exhausted(耗盡)。 
  private static int MAX_ACTIVE = FileUtil.getPropertyValueInt("/properties/redis.properties", "max_active");; 
   
  //控制一個(gè)pool最多有多少個(gè)狀態(tài)為idle(空閑的)的jedis實(shí)例，默認(rèn)值也是8。 
  private static int MAX_IDLE = FileUtil.getPropertyValueInt("/properties/redis.properties", "max_idle");; 
   
  //等待可用連接的最大時(shí)間，單位毫秒，默認(rèn)值為-1，表示永不超時(shí)。如果超過等待時(shí)間，則直接拋出JedisConnectionException； 
  private static int MAX_WAIT = FileUtil.getPropertyValueInt("/properties/redis.properties", "max_wait");; 
 
  //超時(shí)時(shí)間 
  private static int TIMEOUT = FileUtil.getPropertyValueInt("/properties/redis.properties", "timeout");; 
   
  //在borrow一個(gè)jedis實(shí)例時(shí)，是否提前進(jìn)行validate操作；如果為true，則得到的jedis實(shí)例均是可用的； 
  private static boolean TEST_ON_BORROW = FileUtil.getPropertyValueBoolean("/properties/redis.properties", "test_on_borrow");; 
   
  private static JedisPool jedisPool = null; 
   
  /** 
   * redis過期時(shí)間,以秒為單位 
   */ 
  public final static int EXRP_HOUR = 60*60;     //一小時(shí) 
  public final static int EXRP_DAY = 60*60*24;    //一天 
  public final static int EXRP_MONTH = 60*60*24*30;  //一個(gè)月 
   
  /** 
   * 初始化Redis連接池 
   */ 
  private static void initialPool(){ 
    try { 
      JedisPoolConfig config = new JedisPoolConfig(); 
      config.setMaxTotal(MAX_ACTIVE); 
      config.setMaxIdle(MAX_IDLE); 
      config.setMaxWaitMillis(MAX_WAIT); 
      config.setTestOnBorrow(TEST_ON_BORROW); 
      jedisPool = new JedisPool(config, ADDR_ARRAY.split(",")[0], PORT, TIMEOUT); 
    } catch (Exception e) { 
      logger.error("First create JedisPool error : "+e); 
      try{ 
        //如果第一個(gè)IP異常，則訪問第二個(gè)IP 
        JedisPoolConfig config = new JedisPoolConfig(); 
        config.setMaxTotal(MAX_ACTIVE); 
        config.setMaxIdle(MAX_IDLE); 
        config.setMaxWaitMillis(MAX_WAIT); 
        config.setTestOnBorrow(TEST_ON_BORROW); 
        jedisPool = new JedisPool(config, ADDR_ARRAY.split(",")[1], PORT, TIMEOUT); 
      }catch(Exception e2){ 
        logger.error("Second create JedisPool error : "+e2); 
      } 
    } 
  } 
   
   
  /** 
   * 在多線程環(huán)境同步初始化 
   */ 
  private static void poolInit() { 
    //斷言 ，當(dāng)前鎖是否已經(jīng)鎖住，如果鎖住了，就啥也不干，沒鎖的話就執(zhí)行下面步驟 
    assert ! lockPool.isHeldByCurrentThread();  
    lockPool.lock(); 
    try {  
      if (jedisPool == null) {  
        initialPool(); 
      } 
    }catch(Exception e){ 
      e.printStackTrace(); 
    } finally {  
      lockPool.unlock(); 
    } 
  } 
   
   
  public static Jedis getJedis() { 
    //斷言 ，當(dāng)前鎖是否已經(jīng)鎖住，如果鎖住了，就啥也不干，沒鎖的話就執(zhí)行下面步驟 
    assert ! lockJedis.isHeldByCurrentThread();  
    lockJedis.lock(); 
     
    if (jedisPool == null) {  
      poolInit(); 
    } 
    Jedis jedis = null; 
    try {  
      if (jedisPool != null) {  
        jedis = jedisPool.getResource();  
      } 
    } catch (Exception e) {  
      logger.error("Get jedis error : "+e); 
    }finally{ 
      returnResource(jedis); 
      lockJedis.unlock(); 
    } 
    return jedis; 
  }  
   
   
  /** 
   * 釋放jedis資源 
   * @param jedis 
   */ 
  public static void returnResource(final Jedis jedis) { 
    if (jedis != null && jedisPool !=null) { 
      jedisPool.returnResource(jedis); 
    } 
  } 
 
  /** 
   * 設(shè)置 String 
   * @param key 
   * @param value 
   */ 
  public synchronized static void setString(String key ,String value){ 
    try { 
      value = StringUtil.isEmpty(value) ? "" : value; 
      getJedis().set(key,value); 
    } catch (Exception e) { 
      logger.error("Set key error : "+e); 
    } 
  } 
   
  /** 
   * 設(shè)置 過期時(shí)間 
   * @param key 
   * @param seconds 以秒為單位 
   * @param value 
   */ 
  public synchronized static void setString(String key ,int seconds,String value){ 
    try { 
      value = StringUtil.isEmpty(value) ? "" : value; 
      getJedis().setex(key, seconds, value); 
    } catch (Exception e) { 
      logger.error("Set keyex error : "+e); 
    } 
  } 
   
  /** 
   * 獲取String值 
   * @param key 
   * @return value 
   */ 
  public synchronized static String getString(String key){ 
    if(getJedis() == null || !getJedis().exists(key)){ 
      return null; 
    } 
    return getJedis().get(key); 
  }    
}