yujunjun/datahub
2
0
Fork 0
mirror of https://github.com/datahub-project/datahub.git synced 2025-09-09 09:11:01 +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())
github_stats_future = asyncio.ensure_future(get_github_stats())
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):
done, pending = await asyncio.wait(tasks, timeout=0.1)
for t in done:
if t == client_version_stats_future:
client_version_stats = t.result()
elif t == github_stats_future:
(last_server_version, last_server_date) = t.result()
elif t == server_config_future:
(
current_server_type,
current_server_version,
current_server_release_date,
) = server_config_future.result()
tasks.remove(t)
client_version_stats = await client_version_stats_future
(last_server_version, last_server_date) = await github_stats_future
(
current_server_type,
current_server_version,
current_server_release_date,
) = await server_config_future
server_version_stats = None
if current_server_version: