add keepalive

Author: Divkix

WebStreamer/__main__.py

@@ -1,4 +1,4 @@
-from asyncio import get_event_loop
+from asyncio import create_task, get_event_loop
 from glob import glob
 
 from aiohttp import web
@@ -8,6 +8,7 @@
 from WebStreamer.bot.client import initialize_clients
 from WebStreamer.logger import LOGGER
 from WebStreamer.server import web_server
+from WebStreamer.utils import ping_server
 from WebStreamer.vars import Vars
 
 files = glob("WebStreamer/bot/plugins/*.py")
@@ -27,6 +28,8 @@ async def start_services():
     if bot_info.dc_id:
         LOGGER.info(f"DC ID =>> {str(bot_info.dc_id)}")
     await initialize_clients()
+    if Vars.KEEP_ALIVE:
+        create_task(ping_server())
     await server.setup()
     await web.TCPSite(server, Vars.BIND_ADDRESS, Vars.PORT).start()
     LOGGER.info("Service Started")

WebStreamer/utils/keep_alive.py

@@ -0,0 +1,21 @@
+from asyncio import sleep
+
+from aiohttp import ClientSession, ClientTimeout
+
+from WebStreamer.logger import LOGGER
+from WebStreamer.vars import Vars
+
+
+async def ping_server():
+    sleep_time = Vars.PING_INTERVAL
+    LOGGER.info(f"Started with {sleep_time}s interval between pings")
+    while True:
+        await sleep(sleep_time)
+        try:
+            async with ClientSession(timeout=ClientTimeout(total=10)) as session:
+                async with session.get(Vars.URL) as resp:
+                    LOGGER.info(f"Pinged server with response: {resp.status}")
+        except TimeoutError:
+            LOGGER.warning("Couldn't connect to the site URL..")
+        except Exception:
+            LOGGER.error("Unexpected error: ", exc_info=True)

WebStreamer/vars.py

@@ -17,8 +17,8 @@ class Vars:
 
     # Bot tokens
     BOT_TOKEN = str(config("BOT_TOKEN", default=None))
-    # Bot tokens end
-
+    PING_INTERVAL = int(config("PING_INTERVAL", default=0))  # should be in seconds
+    KEEP_ALIVE = PING_INTERVAL != 0
     SLEEP_THRESHOLD = int(config("SLEEP_THRESHOLD", default=60))
     WORKERS = int(config("WORKERS", default=8))
     LOG_CHANNEL = int(config("LOG_CHANNEL", default=-100))