stef_buwalda/dashboard_test
1
0
Fork 0
mirror of https://github.com/StefBuwalda/dashboard_test.git synced 2025-10-30 03:09:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

Creates logs when services are pinged

Browse Source
This commit is contained in:
Stef
2025-09-03 17:11:02 +02:00
parent dea5278731
commit 5a7aa44d6c
3 changed files with 11 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
app.py
View File

@@ -49,7 +49,6 @@ def favicon():
# Only run if directly running file # Only run if directly running file
if __name__ == "__main__": if __name__ == "__main__":
t = threading.Thread(target=start_async_loop, daemon=True) t = threading.Thread(target=start_async_loop, daemon=True)
t.start() t.start()

6
models.py
View File

@@ -5,7 +5,7 @@ from typing import Any, Optional
class log(db.Model): class log(db.Model):
id: int = db.Column(db.Integer, primary_key=True) id: int = db.Column(db.Integer, primary_key=True) # TODO: Switch to UUID
dateCreated: datetime = db.Column(db.DateTime, nullable=False, index=True) dateCreated: datetime = db.Column(db.DateTime, nullable=False, index=True)
service_id: int = db.Column( service_id: int = db.Column(
db.Integer, db.Integer,
@@ -14,8 +14,10 @@ class log(db.Model):
) )
ping: Optional[int] = db.Column(db.Integer, nullable=True) ping: Optional[int] = db.Column(db.Integer, nullable=True)
def __init__(self): def __init__(self, service_id: int, ping: Optional[int]):
super().__init__() super().__init__()
self.service_id = service_id
self.ping = ping
self.dateCreated = datetime.now(timezone.utc) self.dateCreated = datetime.now(timezone.utc)

16
poll_services.py
View File

@@ -31,7 +31,10 @@ async def check_service(client: aiohttp.ClientSession, s: service) -> log:
s.set_error(None) s.set_error(None)
s.set_online(r.status == 200) s.set_online(r.status == 200)
s.set_status(r.status) s.set_status(r.status)
s.set_ping(int((after - before) * 1000)) if r.status != 200:
s.set_ping(None)
else:
s.set_ping(int((after - before) * 1000))
except aiohttp.ConnectionTimeoutError: except aiohttp.ConnectionTimeoutError:
s.set_error("Connection Timeout") s.set_error("Connection Timeout")
s.set_online(False) s.set_online(False)
@@ -43,7 +46,7 @@ async def check_service(client: aiohttp.ClientSession, s: service) -> log:
s.set_online(False) s.set_online(False)
s.set_status(None) s.set_status(None)
s.set_ping(None) s.set_ping(None)
return log() return log(service_id=s.id, ping=s.ping)
def start_async_loop(): def start_async_loop():
@@ -53,11 +56,6 @@ def start_async_loop():
loop.run_forever() loop.run_forever()
async def sleepTask():
await asyncio.sleep(5)
return log()
async def update_services(loop: asyncio.AbstractEventLoop): async def update_services(loop: asyncio.AbstractEventLoop):
print("Starting service updates...") print("Starting service updates...")
with app.app_context(): with app.app_context():
@@ -65,10 +63,10 @@ async def update_services(loop: asyncio.AbstractEventLoop):
client = aiohttp.ClientSession() client = aiohttp.ClientSession()
while True: while True:
session = WorkerSession() session = WorkerSession()
sleeptask = asyncio.create_task(asyncio.sleep(5))
tasks = [check_service(client=client, s=s) for s in services] tasks = [check_service(client=client, s=s) for s in services]
tasks.append(sleepTask())
logs = await asyncio.gather(*tasks) logs = await asyncio.gather(*tasks)
logs = logs[:-1] await sleeptask
try: try:
session.add_all(logs) session.add_all(logs)
session.commit() session.commit()