From c9eb9c14d7aa04e3edea9c0ffefcb13cf975c375 Mon Sep 17 00:00:00 2001
From: Yeuoly <admin@srmxy.cn>
Date: Thu, 26 Dec 2024 22:58:34 +0800
Subject: [PATCH] fix: block call to flask_app

---
 api/services/plugin/plugin_migration.py | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/api/services/plugin/plugin_migration.py b/api/services/plugin/plugin_migration.py
index 7ab7dcb986..6c3e1ef9f6 100644
--- a/api/services/plugin/plugin_migration.py
+++ b/api/services/plugin/plugin_migration.py
@@ -72,6 +72,8 @@ class PluginMigration:
                 except Exception:
                     logger.exception(f"Failed to process tenant {tenant_id}")
 
+        futures = []
+
         while current_time < ended_at:
             click.echo(click.style(f"Current time: {current_time}, Started at: {datetime.datetime.now()}", fg="white"))
             # Initial interval of 1 day, will be dynamically adjusted based on tenant count
@@ -129,13 +131,19 @@ class PluginMigration:
                         logger.exception(f"Failed to process tenant {tenant_id}")
                         continue
 
-            # Process batch with thread pool
-            thread_pool.map(lambda tenant_id: process_tenant(current_app, tenant_id), tenants)
+                    futures.append(
+                        thread_pool.submit(
+                            process_tenant,
+                            current_app._get_current_object(),  # type: ignore[attr-defined]
+                            tenant_id,
+                        )
+                    )
 
             current_time = batch_end
 
         # wait for all threads to finish
-        thread_pool.shutdown(wait=True)
+        for future in futures:
+            future.result()
 
     @classmethod
     def extract_installed_plugin_ids(cls, tenant_id: str) -> Sequence[str]: