|
@@ -116,7 +116,7 @@ class IterationNode(BaseNode[IterationNodeData]):
|
|
|
variable_pool.add([self.node_id, "item"], iterator_list_value[0])
|
|
|
|
|
|
# init graph engine
|
|
|
- from core.workflow.graph_engine.graph_engine import GraphEngine
|
|
|
+ from core.workflow.graph_engine.graph_engine import GraphEngine, GraphEngineThreadPool
|
|
|
|
|
|
graph_engine = GraphEngine(
|
|
|
tenant_id=self.tenant_id,
|
|
@@ -162,8 +162,7 @@ class IterationNode(BaseNode[IterationNodeData]):
|
|
|
if self.node_data.is_parallel:
|
|
|
futures: list[Future] = []
|
|
|
q = Queue()
|
|
|
- thread_pool = graph_engine.workflow_thread_pool_mapping[graph_engine.thread_pool_id]
|
|
|
- thread_pool._max_workers = self.node_data.parallel_nums
|
|
|
+ thread_pool = GraphEngineThreadPool(max_workers=self.node_data.parallel_nums, max_submit_count=100)
|
|
|
for index, item in enumerate(iterator_list_value):
|
|
|
future: Future = thread_pool.submit(
|
|
|
self._run_single_iter_parallel,
|