基于redis實(shí)現(xiàn)分布式鎖的原理與方法

/**
 * @author 張新華
 * @version V1.0
 * @Package com.ay.framework.order.redis.product
 * @date 2020年03月26日, 0026 10:29
 * @Copyright © 2020 安元科技有限公司
 */
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Inherited
public @interface StockLock {

  /**
   * @author zxhtom
   * @Description 鎖key的前綴
   * @Date 15:25 2020年03月25日, 0025
   * @Param []
   * @return java.lang.String
   */
  String prefix() default "";
  /**
   * @author zxhtom
   * @Description key的分隔符
   * @Date 15:27 2020年03月25日, 0025
   * @Param []
   * @return java.lang.String
   */
  String delimiter() default ":";
}

/**
 * @author 張新華
 * @version V1.0
 * @Package com.ay.framework.order.redis.product
 * @date 2020年03月26日, 0026 11:09
 * @Copyright © 2020 安元科技有限公司
 */
@Target({ElementType.PARAMETER , ElementType.METHOD , ElementType.FIELD})
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Inherited
public @interface StockParam {
  /**
  * @author zxhtom
  * @Description 組成key
  * @Date 11:11 2020年03月26日, 0026
  * @Param []
  * @return java.lang.String[]
  */
  String[] names() default {""};
}

@Around("execution(public * *(..))  @annotation(com.ay.framework.order.redis.product.StockLock)")

RedisAtomicLong entityIdCounter = new RedisAtomicLong(lockKey, redisTemplate.getConnectionFactory());
  if (redisTemplate.hasKey(CoreConstants.UPDATEPRODUCTREDISLOCKKEY + lockKey)) {
    //表示lockKey的庫(kù)存信息有變動(dòng)。此時(shí)無(wú)法進(jìn)行交易
    throw new BusinessException("庫(kù)存變動(dòng)。暫無(wú)法交易");
  }
  Long increment = entityIdCounter.decrementAndGet();
  if (increment >= 0) {
    try {
      Object proceed = pjp.proceed();
    } catch (Throwable throwable) {
      //所占資源需要釋放回資源池
      while (!redisLock.tryGetLock(CoreConstants.UPDATEPRODUCTREDISLOCKKEY + lockKey, "")) {

      }
      //表示lockKey的庫(kù)存信息有變動(dòng)。此時(shí)無(wú)法進(jìn)行交易
      long l = entityIdCounter.incrementAndGet();
      if (l  1) {
        redisTemplate.opsForValue().set(lockKey,1);
      }
      redisLock.unLock(CoreConstants.UPDATEPRODUCTREDISLOCKKEY + lockKey);
      throwable.printStackTrace();
    }
  } else {
    redisTemplate.opsForValue().set(lockKey,0);
    throw new BusinessException("庫(kù)存不足！無(wú)法操作");
  }

public Boolean tryGetLock(String key , String value) {
  return tryGetLock(key, value, -1, TimeUnit.DAYS);
}
public Boolean tryGetLock(String key , String value, Integer expire) {
  return tryGetLock(key, value, expire, TimeUnit.SECONDS);
}
public Boolean tryGetLock(String key , String value, Integer expire , TimeUnit timeUnit) {
  ValueOperations operations = redisTemplate.opsForValue();
  if (operations.setIfAbsent(key, value)) {
    //說(shuō)明 redis沒(méi)有該key , 換言之 加鎖成功 設(shè)置過(guò)期時(shí)間防止死鎖
    if (expire > 0) {
      redisTemplate.expire(key, expire, timeUnit);
    }
    return true;
  }
  return false;
}

public Boolean unLock(String key) {
  return redisTemplate.delete(key);
}

@Component()
@Primary
public class StockKeyGenerator implements CacheKeyGenerator {
  @Override
  public String getLockKey(ProceedingJoinPoint pjp) {
    //獲取方法簽名
    MethodSignature signature = (MethodSignature) pjp.getSignature();
    Method method = signature.getMethod();
    //獲取方法cacheLock注解
    StockLock stockLock = method.getAnnotation(StockLock.class);
    //獲取方法參數(shù)
    Object[] args = pjp.getArgs();
    Parameter[] parameters = method.getParameters();
    StringBuilder builder = new StringBuilder();
    for (int i = 0; i  parameters.length; i++) {
      StockParam stockParam = parameters[i].getAnnotation(StockParam.class);
      Object arg = args[i];
      if (arg instanceof Map) {
        MapString, Object> temArgMap = (MapString, Object>) arg;
        String[] names = stockParam.names();
        for (String name : names) {
          if (builder.length() > 0) {
            builder.append(stockLock.delimiter());
          }
          builder.append(temArgMap.get(name));
        }
      }

    }
    return builder.toString();
  }
}