在Java中,Semaphore
和CountDownLatch
都是用于线程同步的工具类,但它们的应用场景不同。Semaphore
用于控制同时访问某些资源的线程数量,而CountDownLatch
用于等待一组线程完成他们的操作。
如果我们想要让一定数量的线程等待直到另外一定数量的线程完成他们的任务,我们可以将CountDownLatch
和Semaphore
组合使用。
Java 信号量(semaphore)搭配CountDownLatch 实现多线程处理循环内逻辑并限制创建线程数
- 使用场景
for循环中的处理逻辑较慢,这个时候可以在for循环中创建线程实现多线程处理数据,结合CountDownLatch 可以等待所有线程全部执行结束之后主线程再执行。
但是有的时候循环次数比较多,并不适合循环创建,因为线程会创建太多。
比如下面这段代码
ExecutorService threadPools = Executors.newFixedThreadPool(20);
CountDownLatch countDownLatch = new CountDownLatch(100);
for (int i = 0; i < 100; i++) {
final int token = i;
threadPools.execute(()->{
try {
System.out.println("线程" + Thread.currentThread().getName());
Thread.sleep(1000);
} catch (InterruptedException e) {
e.printStackTrace();
}finally {
countDownLatch.countDown();
}
});
}
try {
countDownLatch.await();
} catch (InterruptedException e) {
e.printStackTrace();
}
- 这时循环100次会执行100个任务,如果循环次数特别大,会直接导致系统内存占满死机。
这个时候可以使用信号量来限制任务执行的次视
代码
@Service
public class XxlService {
private final Semaphore semaphore = new Semaphore(10);
private final ExecutorService threadPools = Executors.newFixedThreadPool(20);
public void syncSaiWordData() {
// 2. 创建计数器
CountDownLatch countDownLatch = new CountDownLatch(needSyncDataList.size());
// 4. 开启多线程完善数据 最多10条线程
for (WordSaiwordLibraryPO wordSaiwordLibraryPO : needSyncDataList) {
threadPools.execute(() -> {
try {
// 获取许可
semaphore.acquire();
// 已经补全信息的无需再次查询
// 处理业务逻辑...
// ...
} catch (Exception e) {
e.printStackTrace();
} finally {
// 归还许可
semaphore.release();
//计数器 -1
countDownLatch.countDown();
}
});
}
try {
// 等待全部执行结束
countDownLatch.await();
//needSyncDataList.size()个数的线程全部执行完之后,继续处理下面逻辑
....
} catch (InterruptedException e) {
e.printStackTrace();
}
}