yujunjun/datahub
2
0
Fork 0
mirror of https://github.com/datahub-project/datahub.git synced 2025-09-09 17:20:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(ingest): simplify upgrade version stats (#5588)

Browse Source
This commit is contained in:
Harshal Sheth 2022-08-09 16:37:34 +00:00 committed by GitHub
parent b46a6cd63f
commit c7b3413b11
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
metadata-ingestion/src/datahub/upgrade/upgrade.py
View File

@ -163,22 +163,14 @@ async def retrieve_version_stats(
client_version_stats_future = asyncio.ensure_future(get_client_version_stats()) client_version_stats_future = asyncio.ensure_future(get_client_version_stats())
github_stats_future = asyncio.ensure_future(get_github_stats()) github_stats_future = asyncio.ensure_future(get_github_stats())
server_config_future = asyncio.ensure_future(get_server_version_stats(server)) server_config_future = asyncio.ensure_future(get_server_version_stats(server))
tasks = [client_version_stats_future, github_stats_future, server_config_future]
while len(tasks): client_version_stats = await client_version_stats_future
done, pending = await asyncio.wait(tasks, timeout=0.1) (last_server_version, last_server_date) = await github_stats_future
for t in done: (
if t == client_version_stats_future: current_server_type,
client_version_stats = t.result() current_server_version,
elif t == github_stats_future: current_server_release_date,
(last_server_version, last_server_date) = t.result() ) = await server_config_future
elif t == server_config_future:
(
current_server_type,
current_server_version,
current_server_release_date,
) = server_config_future.result()
tasks.remove(t)
server_version_stats = None server_version_stats = None
if current_server_version: if current_server_version: