NgTech-LTD/HFCNotificator
2
0
Fork 0
mirror of https://github.com/GaMeNu/HFCNotificator.git synced 2024-11-16 15:24:51 +02:00
Code Issues Packages Projects Releases Wiki Activity

v2.2.6: fix: Added extra security layer for duped loops.

Browse Source
This commit is contained in:
GaMeNu 2024-11-06 18:28:03 +02:00
parent 0f51370c9f
commit b8d035627b
1 changed files with 19 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
cog_notificator.py
View File

@ -2,6 +2,7 @@ import datetime
import platform import platform
import re import re
import sys import sys
import time
import cpuinfo import cpuinfo
import discord import discord
@ -20,7 +21,8 @@ from markdown import md
load_dotenv() load_dotenv()
AUTHOR_ID = int(os.getenv('AUTHOR_ID')) AUTHOR_ID = int(os.getenv('AUTHOR_ID'))
EXPECTED_LOOP_DELTA_MIN = 0.8
EXPECTED_LOOP_DELTA_MAX = 1.2
# 2023-10-26 I have decided to start documenting the project. # 2023-10-26 I have decided to start documenting the project.
@ -124,7 +126,9 @@ class Notificator(commands.Cog):
self.active_districts = [] self.active_districts = []
self.district_timeouts = {} self.district_timeouts = {}
self.alert_reqs = AlertReqs() self.alert_reqs = AlertReqs()
self.loop_count_checker = 0 self.loop_count_checker = 0
self.last_loop_run_time = time.time() - 1 # Verify first iteration goes by smoothly
if not self.check_for_updates.is_running(): if not self.check_for_updates.is_running():
self.check_for_updates.start() self.check_for_updates.start()
@ -217,6 +221,20 @@ class Notificator(commands.Cog):
@tasks.loop(seconds=1, reconnect=False) @tasks.loop(seconds=1, reconnect=False)
async def check_for_updates(self): async def check_for_updates(self):
# Check if the loop is running multiple too fast or too slow
current_time = time.time()
delta = round(current_time - self.last_loop_run_time, 3)
print(delta)
if delta < EXPECTED_LOOP_DELTA_MIN:
self.log.warning(f'Loop is running too quickly! Expected delta > {EXPECTED_LOOP_DELTA_MIN}s, but got {delta}s. Restarting...')
self.check_for_updates.stop()
return
if delta > EXPECTED_LOOP_DELTA_MAX:
self.log.warning(f'Loop is running too slowly! Expected delta < {EXPECTED_LOOP_DELTA_MAX}s, but got {delta}s instead. Do you have enough resources?')
self.last_loop_run_time = current_time
# Check if the loop is running multiple times # Check if the loop is running multiple times
if self.loop_count_checker >= 1: if self.loop_count_checker >= 1: