封装 分布式工具，使用炒鸡方便

背景 有时候我们希望 ：

1. 获取锁执行 其他业务等待 获取锁在执行
2. 获取到锁执行，获取不到就不执行

故而简单封装，用起来还是很挺爽的。

@Component
public class DistributedLockService {

	private static final String DISTRIBUTED_LOCK_SERVICE = "distributed_lock_service:";

	@Autowired
	private RedissonClient redissonClient;

	/**
	 * 等待执行 函数
	 * @param lockKey
	 * @param obj 函数参数
	 * @param function
	 * @return
	 */
	public Object lockFunction(String lockKey,Object obj,Function function){
		RLock lock = redissonClient.getLock(DISTRIBUTED_LOCK_SERVICE+lockKey);
		try {
			lock.lock();
			Object result = function.apply(obj);
			return result;
		}finally {
			lock.unlock();
		}
	}

	/**
	 * 尝试执行 函数
	 * @param lockKey
	 * @param obj 函数参数
	 * @param function
	 * @return
	 */
	public Object tryLockFunction(String lockKey, Object obj, Function function) {
		RLock lock = redissonClient.getLock(DISTRIBUTED_LOCK_SERVICE + lockKey);
		boolean tryLock = lock.tryLock();
		if (tryLock) {
			try {
				Object result = function.apply(obj);
				return result;
			} finally {
				lock.unlock();
			}
		}
		return null;
	}

	/**
	 * 等待执行 函数
	 * @param lockKey
	 * @param supplier
	 * @return
	 */
	public Object lockSupplier(String lockKey,  Supplier supplier){
		RLock lock = redissonClient.getLock(DISTRIBUTED_LOCK_SERVICE+lockKey);
		try {
			lock.lock();
			Object result = supplier.get();
			return result;
		}finally {
			lock.unlock();
		}
	}


	/**
	 * 尝试执行 函数
	 * @param lockKey
	 * @param supplier
	 * @return
	 */
	public Object tryLockSupplier(String lockKey, Supplier supplier) {
		RLock lock = redissonClient.getLock(DISTRIBUTED_LOCK_SERVICE + lockKey);
		boolean tryLock = lock.tryLock();
		if (tryLock) {
			try {
				Object result = supplier.get();
				return result;
			} finally {
				lock.unlock();
			}
		}
		return null;
	}
}

使用 注入 直接用，就是这么 朴实无华

 
 @Test
    public void testMultiThread() throws InterruptedException {
        CountDownLatch countDownLatch = new CountDownLatch(10);
        ExecutorService executorService = Executors.newFixedThreadPool(10);
        CyclicBarrier cyclicBarrier = new CyclicBarrier(10);
        for (int i = 0; i < 10; i++) {
            executorService.execute(new Runnable() {
                @Override
                public void run() {
                    try {
                        Thread.sleep(new Random().nextInt(60)*1000);
                        cyclicBarrier.await();
                        // 业务方法
                        distributedLockService.lockSupplier("test",()->{
                            System.out.println("尝试执行多次");
                            return null;
                        });
                        log.info("开始执行 {},{}",Thread.currentThread().getName());
                        countDownLatch.countDown();
                    } catch (InterruptedException e) {
                        e.printStackTrace();
                    } catch (BrokenBarrierException e) {
                        e.printStackTrace();
                    }
                }
            });
        }
        countDownLatch.await();
        executorService.shutdown();
    }

 @Test
    public void testMultiThread() throws InterruptedException {
        CountDownLatch countDownLatch = new CountDownLatch(10);
        ExecutorService executorService = Executors.newFixedThreadPool(10);
        CyclicBarrier cyclicBarrier = new CyclicBarrier(10);
        for (int i = 0; i < 10; i++) {
            executorService.execute(new Runnable() {
                @Override
                public void run() {
                    try {
                        Thread.sleep(new Random().nextInt(60)*1000);
                        cyclicBarrier.await();
                        // 业务方法
                        distributedLockService.tryLockFunction("test","小米",(e)->{
                            System.out.println("尝试执行一次"+e);
                            return null;
                        });
                        log.info("开始执行 {},{}",Thread.currentThread().getName());
                        countDownLatch.countDown();
                    } catch (InterruptedException e) {
                        e.printStackTrace();
                    } catch (BrokenBarrierException e) {
                        e.printStackTrace();
                    }
                }
            });
        }