package com.mic.synchrolock.util;
import java.util.ArrayList;
import java.util.List;
import java.util.UUID;
import javax.annotation.PostConstruct;
import javax.annotation.PreDestroy;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.springframework.beans.factory.annotation.Autowired;
import com.mic.constants.Constants;
import com.mic.constants.InvestType;
/**
 * 分布式同步鎖工具類
 * @author Administrator
 *
 */
public class SynchrolockUtil {
 private final Log logger = LogFactory.getLog(getClass());
 @Autowired
 private RedisClientTemplate redisClientTemplate;
 public final String RETRYTYPE_WAIT = "1";  //加鎖方法當(dāng)對(duì)象已加鎖時(shí)，設(shè)置為等待并輪詢
 public final String RETRYTYPE_NOWAIT = "0";  //加鎖方法當(dāng)對(duì)象已加鎖時(shí)，設(shè)置為直接返回
 private String requestTimeOutName = "";  //投資同步鎖請(qǐng)求超時(shí)時(shí)間
 private String retryIntervalName = "";   //投資同步鎖輪詢間隔
 private String keyTimeoutName = "";  //緩存中key的失效時(shí)間
 private String investProductSn = "";   //產(chǎn)品Sn
 private String uuid;    //對(duì)象唯一標(biāo)識(shí)
 private Long startTime = System.currentTimeMillis(); //首次調(diào)用時(shí)間
 public Long getStartTime() {
  return startTime;
 }
 List<String> keyList = new ArrayList<String>(); //緩存key的保存集合
 public List<String> getKeyList() {
  return keyList;
 }
 public void setKeyList(List<String> keyList) {
  this.keyList = keyList;
 }
 @PostConstruct
 public void init() {
  uuid = UUID.randomUUID().toString();
 }
 @PreDestroy
 public void destroy() {
  this.unlock();
 }
 /**
  * 根據(jù)傳入key值，判斷緩存中是否存在該key
  * 存在-已上鎖：判斷retryType，輪詢超時(shí)，或直接返回，返回ture
  * 不存在-未上鎖：將該放入緩存，返回false
  * @param key
  * @param retryType 當(dāng)遇到上鎖情況時(shí) 1：輪詢；0：直接返回
  * @return
  */
 public boolean islocked(String key,String retryType){
  boolean flag = true;
  logger.info("====投資同步鎖設(shè)置輪詢間隔、請(qǐng)求超時(shí)時(shí)長(zhǎng)、緩存key失效時(shí)長(zhǎng)====");
  //投資同步鎖輪詢間隔 毫秒
  Long retryInterval = Long.parseLong(Constants.getProperty(retryIntervalName));
  //投資同步鎖請(qǐng)求超時(shí)時(shí)間 毫秒
  Long requestTimeOut = Long.parseLong(Constants.getProperty(requestTimeOutName));
  //緩存中key的失效時(shí)間 秒
  Integer keyTimeout = Integer.parseInt(Constants.getProperty(keyTimeoutName));
  //調(diào)用緩存獲取當(dāng)前產(chǎn)品鎖
  logger.info("====當(dāng)前產(chǎn)品key為："+key+"====");
  if(isLockedInRedis(key,keyTimeout)){
   if("1".equals(retryType)){
    //采用輪詢方式等待
    while (true) {
     logger.info("====產(chǎn)品已被占用，開(kāi)始輪詢====");
     try {
      Thread.sleep(retryInterval);
     } catch (InterruptedException e) {
      logger.error("線程睡眠異常:"+e.getMessage(), e);
      return flag;
     }
     logger.info("====判斷請(qǐng)求是否超時(shí)====");
     Long currentTime = System.currentTimeMillis(); //當(dāng)前調(diào)用時(shí)間
     long Interval = currentTime - startTime;
     if (Interval > requestTimeOut) {
      logger.info("====請(qǐng)求超時(shí)====");
      return flag;
     }
     if(!isLockedInRedis(key,keyTimeout)){
      logger.info("====輪詢結(jié)束，添加同步鎖====");
      flag = false;
      keyList.add(key);
      break;
     }
    }
   }else{
    //不等待，直接返回
    logger.info("====產(chǎn)品已被占用，直接返回====");
    return flag;
   }
  }else{
   logger.info("====產(chǎn)品未被占用，添加同步鎖====");
   flag = false;
   keyList.add(key);
  }
  return flag;
 }
 /**
  * 在緩存中查詢key是否存在
  * 若存在則返回true;
  * 若不存在則將key放入緩存，設(shè)置過(guò)期時(shí)間，返回false
  * @param key
  * @param keyTimeout key超時(shí)時(shí)間單位是秒
  * @return
  */
 boolean isLockedInRedis(String key,int keyTimeout){
  logger.info("====在緩存中查詢key是否存在====");
  boolean isExist = false;
  //與redis交互，查詢對(duì)象是否上鎖
  Long result = this.redisClientTemplate.setnx(key, uuid);
  logger.info("====上鎖 result = "+result+"====");
  if(null != result && 1 == Integer.parseInt(result.toString())){
   logger.info("====設(shè)置緩存失效時(shí)長(zhǎng) = "+keyTimeout+"秒====");
   this.redisClientTemplate.expire(key, keyTimeout);
   logger.info("====上鎖成功====");
   isExist = false;
  }else{
   logger.info("====上鎖失敗====");
   isExist = true;
  }
  return isExist;
 }
 /**
  * 根據(jù)傳入key，對(duì)該產(chǎn)品進(jìn)行解鎖
  * @param key
  * @return
  */
 public void unlock(){
  //與redis交互，對(duì)產(chǎn)品解鎖
  if(keyList.size()>0){
   for(String key : this.keyList){
    String value = this.redisClientTemplate.get(key);
    if(null != value && !"".equals(value)){
     if(uuid.equals(value)){
      logger.info("====解鎖key:"+key+" value="+value+"====");
      this.redisClientTemplate.del(key);
     }else{
      logger.info("====待解鎖集合中key:"+key+" value="+value+"與uuid不匹配====");
     }
    }else{
     logger.info("====待解鎖集合中key="+key+"的value為空====");
    }
   }
  }else{
   logger.info("====待解鎖集合為空====");
  }
 }
}

//獲取同步鎖工具類
　　SynchrolockUtil synchrolockUtil = SpringUtils.getBean("synchrolockUtil");
　　//獲取需上鎖資源的KEY
　　String key = "abc";
　　//查詢是否上鎖，上鎖輪詢，未上鎖加鎖
　　boolean isLocked = synchrolockUtil.islocked(key,synchrolockUtil.RETRYTYPE_WAIT);
　　//判斷上鎖結(jié)果
　　if(isLocked){
 　　logger.error("同步鎖請(qǐng)求超時(shí)并返回 key ="+key);
　　}else{
 　　logger.info("====同步鎖加鎖陳功====");
　　}
　　try {
 　　//執(zhí)行業(yè)務(wù)處理
　　} catch (Exception e) {
 　　logger.error("業(yè)務(wù)異常:"+e.getMessage(), e);
　　}finally{
 　　//解鎖
 　　 synchrolockUtil.unlock();
　　}