优化版
针对以上死循环的情况，我们可以改进的思路有以下两种：
- 添加最大次数限制：如果经过了 n 次尝试获取锁之后，还未获取到锁，则认为获取锁失败，执行失败策略之后终止轮询（失败策略可以是记录日志或其他操作）；
- 添加最大时长限制：如果经过了 n 秒尝试获取锁之后，还未获取到锁，则认为获取锁失败，执行失败策略之后终止轮询。
以上策略任选其一就可以解决死循环的问题，出于实现成本的考虑，我们可以采用轮询最大次数的方式来改进轮询锁，具体实现代码如下：
import java.util.concurrent.locks.Lock; import java.util.concurrent.locks.ReentrantLock; publicclass SolveDeadLockExample { public static void main(String[] args) { Lock lockA = new ReentrantLock(); // 创建锁 A Lock lockB = new ReentrantLock(); // 创建锁 B // 创建线程 1(使用轮询锁) Thread t1 = new Thread(new Runnable() { @Override public void run() { // 调用轮询锁 pollingLock(lockA, lockB, 3); } }); t1.start(); // 运行线程 // 创建线程 2 Thread t2 = new Thread(new Runnable() { @Override public void run() { lockB.lock(); // 加锁 System.out.println("线程 2:获取到锁 B!"); try { Thread.sleep(1000); System.out.println("线程 2:等待获取 A..."); lockA.lock(); // 加锁 try { System.out.println("线程 2:获取到锁 A!"); } finally { lockA.unlock(); // 释放锁 } } catch (InterruptedException e) { e.printStackTrace(); } finally { // 线程 2 忘记释放锁资源 // lockB.unlock(); // 释放锁 } } }); t2.start(); // 运行线程 } /** * 轮询锁 * * maxCount：最大轮询次数 */ public static void pollingLock(Lock lockA, Lock lockB, int maxCount) { // 轮询次数计数器 int count = 0; while (true) { if (lockA.tryLock()) { // 尝试获取锁 System.out.println("线程 1:获取到锁 A!"); try { Thread.sleep(1000); System.out.println("线程 1:等待获取 B..."); if (lockB.tryLock()) { // 尝试获取锁 try { System.out.println("线程 1:获取到锁 B!"); } finally { lockB.unlock(); // 释放锁 System.out.println("线程 1:释放锁 B."); break; } } } catch (InterruptedException e) { e.printStackTrace(); } finally { lockA.unlock(); // 释放锁 System.out.println("线程 1:释放锁 A."); } } // 判断是否已经超过最大次数限制 if (count++ > maxCount) { // 终止循环 System.out.println("轮询锁获取失败,记录日志或执行其他失败策略"); return; } // 等待一秒再继续尝试获取锁 try { Thread.sleep(1000); } catch (InterruptedException e) { e.printStackTrace(); } } } }
以上代码的执行结果如下：
从以上结果可以看出，当我们改进之后，轮询锁就不会出现死循环的问题了，它会尝试一定次数之后终止执行。
