Include simultanious testing in status
All checks were successful
Build and Publish Jimmy.2 / build_and_publish (push) Successful in 7s
All checks were successful
Build and Publish Jimmy.2 / build_and_publish (push) Successful in 7s
This commit is contained in:
parent
899e9c84dd
commit
b70016326a
1 changed files with 16 additions and 10 deletions
|
@ -978,19 +978,25 @@ class Ollama(commands.Cog):
|
|||
|
||||
msg = await ctx.reply(embed=embed)
|
||||
last_edit = time.time()
|
||||
|
||||
tasks = {}
|
||||
for server in self.servers.keys():
|
||||
if self.servers[server].locked():
|
||||
continue
|
||||
try:
|
||||
ok = await self.check_server(CONFIG["ollama"][server]["base_url"])
|
||||
except Exception as e:
|
||||
ok = False
|
||||
tasks[server] = asyncio.create_task(self.check_server(CONFIG["ollama"][server]["base_url"]))
|
||||
|
||||
await asyncio.gather(*tasks.values())
|
||||
for server, task in tasks.items():
|
||||
if not task.done:
|
||||
await task
|
||||
if e := task.exception():
|
||||
self.log.error("Error while checking server %r: %r", server, e, exc_info=True)
|
||||
else:
|
||||
ok = task.result()
|
||||
for field in embed.fields:
|
||||
if field.name == server:
|
||||
field.value = "\U00002705 Online" if ok else "\U0000274c Offline"
|
||||
break
|
||||
await asyncio.sleep(1 - (time.time() - last_edit))
|
||||
await msg.edit(embed=embed)
|
||||
|
||||
|
||||
|
|
Loading…
Reference in a new issue