问题描述
假设我想知道给定的 ExecutionContext
中有多少个线程。
所以我正在写一个这样的函数
def count(implicit ec: ExecutionContext): Int = {
val promise = Promise[Unit]
val counter = new AtomicInteger(0)
for (_ <- 0 to 1000) Future {
counter.getAndIncrement()
Await.ready(promise.future,Duration.Inf)
}
val result = counter.get()
promise.success(())
result
}
它至少对 ExecutionContext.global
不起作用,所以我改变了实现:
def count(implicit ec: ExecutionContext): Int = {
val barrier = new CyclicBarrier(1000)
var isReset = false
for (_ <- 0 to 1000) Future { if (!isReset) barrier.await() }
val result = barrier.getNumberWaiting
barrier.reset()
// make all futures complete and release all threads to allow JVM to exit
isReset = true
result
}
它有效,但我不知道
- 为什么第一个实现不起作用;
- 如何改进“障碍”实现(例如摆脱
isReset
); - 计算
ExecutionContext
中线程数的最佳方法是什么。
解决方法
例如尝试投射到特定的执行者
implicit val ec = scala.concurrent.ExecutionContext.Implicits.global
Future(42)
ec.asInstanceOf[java.util.concurrent.ForkJoinPool].getPoolSize // : Int = 1