From 52019d85d93c2d0e2c856fdec6c6dcb31c3acaeb Mon Sep 17 00:00:00 2001
From: subframe7536 <1667077010@qq.com>
Date: Wed, 19 Feb 2025 19:51:35 +0800
Subject: [PATCH] use ThreadPoolExecutor

---
 build.py | 37 +++++++++++++++++++++++++++++++------
 1 file changed, 31 insertions(+), 6 deletions(-)

diff --git a/build.py b/build.py
index 2dea31f..c3a1e0a 100755
--- a/build.py
+++ b/build.py
@@ -1,12 +1,13 @@
 #!/usr/bin/env python3
 import argparse
+from concurrent.futures import ThreadPoolExecutor, as_completed
 import importlib.util
 import json
 import re
 import shutil
+import threading
 import time
 from functools import partial
-from multiprocessing import Pool
 from os import environ, getcwd, listdir, makedirs, path, remove, getenv
 from typing import Callable
 from fontTools.ttLib import TTFont, newTable
@@ -983,12 +984,36 @@ def build_cn(f: str, font_config: FontConfig, build_option: BuildOption):
 
 
 def run_build(pool_size: int, fn: Callable, dir: str):
-    if pool_size > 1:
-        with Pool(pool_size) as p:
-            p.map(fn, listdir(dir))
-    else:
-        for f in listdir(dir):
+    files = listdir(dir)
+
+    if pool_size <= 1:
+        for f in files:
             fn(f)
+        return
+
+    shutdown_event = threading.Event()
+
+    def wrapped_fn(filename):
+        if shutdown_event.is_set():
+            return
+        try:
+            return fn(filename)
+        except Exception as e:
+            shutdown_event.set()
+            raise e
+
+    with ThreadPoolExecutor(max_workers=pool_size) as executor:
+        futures = {executor.submit(wrapped_fn, f): f for f in files}
+
+        for future in as_completed(futures):
+            try:
+                future.result()
+            except Exception as e:
+                shutdown_event.set()
+                for f in futures:
+                    f.cancel()
+                executor.shutdown(wait=False, cancel_futures=True)
+                raise e
 
 
 def main():