AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-20 01:29:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: block call to flask_app

Browse Source
This commit is contained in:
Yeuoly 2024-12-26 22:58:34 +08:00
parent e77cd87842
commit c9eb9c14d7
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
api/services/plugin/plugin_migration.py
View File

@ -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]: