love.keeping.starter.web.components.redis.locker.RedisLockBuilder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of web-starter Show documentation
The newest version!
package love.keeping.starter.web.components.redis.locker;

import love.keeping.starter.common.locker.LockBuilder;
import love.keeping.starter.common.locker.Locker;
import love.keeping.starter.web.utils.IdUtil;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeUnit;
import javax.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.springframework.dao.DataAccessException;
import org.springframework.data.redis.connection.RedisConnection;
import org.springframework.data.redis.connection.RedisStringCommands;
import org.springframework.data.redis.connection.ReturnType;
import org.springframework.data.redis.core.RedisCallback;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.types.Expiration;

public class RedisLockBuilder implements LockBuilder {

  private static final ThreadLocal> LOCKERS;

  static {
    LOCKERS = new ThreadLocal<>();
  }

  @Resource(name = "redisTemplate")
  private RedisTemplate redisTemplate;

  @Override
  public Locker buildLocker(String lockName, long expireTime, long waitTime) {

    if (LOCKERS.get() == null) {
      LOCKERS.set(new HashMap<>());
    }

    Locker locker = LOCKERS.get().get(lockName);
    if (locker != null) {
      return locker;
    }

    locker = new RedisLocker(redisTemplate, lockName, expireTime, waitTime);
    LOCKERS.get().put(lockName, locker);

    return locker;
  }

  @Slf4j
  public static class RedisLocker implements Locker {


    private static final String SCRIPT = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";

    private static final Long INTERVAL_TIME = 100L;

    private final RedisTemplate redisTemplate;

    private final byte[] key;

    private final byte[] requestId = IdUtil.getUUID().getBytes();

    private final long expireTime;

    private boolean lock;

    private long waitTime;

    private int lockCount = 0;

    public RedisLocker(RedisTemplate redisTemplate, String key, long expireTime,
        long waitTime) {

      this.redisTemplate = redisTemplate;

      this.key = key.getBytes();

      this.expireTime = expireTime;

      this.waitTime = waitTime;
    }

    @Override
    public boolean lock() {

      if (this.lock) {
        addLockCount();
        return true;
      }

      while (!(this.lock = this.redisTemplate.execute(new TryLockRedisCallBack()))
          && this.waitTime > 0L) {
        try {
          Thread.sleep(INTERVAL_TIME);
          this.waitTime -= INTERVAL_TIME;
        } catch (InterruptedException e) {
          log.error("获取lock中断", e);
          return false;
        }
      }

      if (log.isDebugEnabled()) {
        log.debug("获取lock完成，result={}, key={}, requestId={}", this.lock, new String(this.key),
            this.requestId);
      }

      if (this.lock) {
        addLockCount();
      }
      return this.lock;
    }

    private void addLockCount() {

      this.lockCount++;
      if (log.isDebugEnabled()) {
        log.debug("重入次数：{}", this.lockCount);
      }
    }

    private void reduceLockCount() {

      this.lockCount--;
      if (log.isDebugEnabled()) {
        log.debug("重入次数：{}", this.lockCount);
      }
    }

    @Override
    public boolean unLock() {

      if (this.lock) {
        if (this.lockCount <= 1) {
          boolean release = this.redisTemplate.execute(new ReleaseRedisCallBack());
          if (log.isDebugEnabled()) {
            log.debug("释放lock完成，result={}, key={}, requestId={}", this.lock, new String(this.key),
                new String(this.requestId));
          }

          this.lock = !release;

          reduceLockCount();

          RedisLockBuilder.LOCKERS.get().remove(this.key);

          return release;
        } else {

          reduceLockCount();
          return true;
        }
      }

      return true;
    }

    @Override
    public void close() {

      unLock();
    }

    private class TryLockRedisCallBack implements RedisCallback {

      @Override
      public Boolean doInRedis(RedisConnection connection) throws DataAccessException {

        if (log.isDebugEnabled()) {
          log.debug("key={}, requestId={}", new String(key), new String(requestId));
        }

        return connection.set(key, requestId, Expiration.from(expireTime, TimeUnit.MILLISECONDS),
            RedisStringCommands.SetOption.ifAbsent());
      }
    }

    private class ReleaseRedisCallBack implements RedisCallback {

      @Override
      public Boolean doInRedis(RedisConnection connection) throws DataAccessException {

        return connection.eval(SCRIPT.getBytes(), ReturnType.BOOLEAN, 1, key, requestId);
      }
    }
  }
}