修复parallelForEach的BUG

Gh0u1L5 · Nov 24, 2018 · 7fcb0d7 · 7fcb0d7
1 parent 1b0c1a8
commit 7fcb0d7
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/main/kotlin/com/gh0u1l5/wechatmagician/spellbook/util/ParallelUtil.kt b/src/main/kotlin/com/gh0u1l5/wechatmagician/spellbook/util/ParallelUtil.kt
@@ -1,6 +1,8 @@
 package com.gh0u1l5.wechatmagician.spellbook.util
 
+import com.gh0u1l5.wechatmagician.spellbook.util.BasicUtil.tryVerbosely
 import java.util.concurrent.Executors
+import java.util.concurrent.TimeUnit
 import kotlin.concurrent.thread
 
 /**
@@ -51,7 +53,11 @@ object ParallelUtil {
  val iterator = iterator()
  while (iterator.hasNext()) {
  val item = iterator.next()
- pool.execute { action(item) }
+ pool.execute {
+ tryVerbosely { action(item) } // 避免进程崩溃
+ }
  }
+ pool.shutdown()
+ pool.awaitTermination(Long.MAX_VALUE, TimeUnit.NANOSECONDS)
  }
 }