23.Redission分布式锁工具类

23.Redission分布式锁工具类

import org.redisson.api.RLock;
import org.redisson.api.RedissonClient;

import java.util.concurrent.TimeUnit;


public class RedisDistributedLocker {

    private static final long DEFAULT_WAIT_TIME = 30;
    private static final long DEFAULT_TIMEOUT = 5;
    private static final TimeUnit DEFAULT_TIME_UNIT = TimeUnit.SECONDS;

    private static RedissonClient redisson;

    public void setRedissonClient(RedissonClient locker) {
        redisson = locker;
    }

    
    public static RLock lock(String lockName, boolean fairLock) {
        return lock(lockName, DEFAULT_TIMEOUT, DEFAULT_TIME_UNIT, fairLock);
    }

    
    public static RLock lock(String lockName, long leaseTime, TimeUnit timeUnit, boolean fairLock) {
        RLock lock = getLock(lockName, fairLock);
        lock.lock(leaseTime, timeUnit);
        return lock;
    }

    
    public static void unlock(String lockName) {
        RLock lock = redisson.getLock(lockName);
        if (lock != null && lock.isHeldByCurrentThread()) {
            lock.unlock();
        }
    }

    
    public static void unlock(RLock lock) {
        if (lock != null && lock.isHeldByCurrentThread()) {
            lock.unlock();
        }
    }

    
    public static boolean tryLock(String lockName, boolean fairLock) {
        return tryLock(lockName, DEFAULT_WAIT_TIME, DEFAULT_TIMEOUT, DEFAULT_TIME_UNIT, fairLock);
    }

    
    public static boolean tryLock(String lockName,
                                  long waitTime,
                                  long leaseTime,
                                  TimeUnit timeUnit,
                                  boolean fairLock) {
        RLock lock = getLock(lockName, fairLock);
        try {
            return lock.tryLock(waitTime, leaseTime, timeUnit);
        } catch (InterruptedException e) {
            System.out.println("获取锁失败");
            return false;
        }
    }

    public static RLock getLock(String lockName, boolean fairLock) {
        RLock lock;
        if (fairLock) {
            lock = redisson.getFairLock(lockName);
        } else {
            lock = redisson.getLock(lockName);
        }
        return lock;
    }
}