Add monitoring

nonamenix · nonamenix · commit efb43edcaed6 · 2018-01-20T12:49:43.000+03:00
diff --git a/bot/bot.py b/bot/bot.py
@@ -1,24 +1,44 @@
 from typing import List
 
-from aiotg import Bot as BaseBot, API_TIMEOUT, Chat
+from aiotg import Bot as BaseBot, API_TIMEOUT, Chat, asyncio, aiohttp
 import re
 
 
 class Bot(BaseBot):
-    def __init__(self, api_token: str, moderators: List[int],
-                 api_timeout: int = API_TIMEOUT,
-                 botan_token=None,
-                 name=None):
+    healthcheckio_url = 'https://hchk.io/{token}'
+    healthcheckio_interval = 150  # seconds
+    healthcheckio_token = None
+
+    def __init__(
+            self,
+            api_token: str, moderators: List[int],
+            api_timeout: int = API_TIMEOUT,
+            botan_token: str = None,
+            healthcheckio_token: str = None,
+            name=None):
         super(Bot, self).__init__(
             api_token=api_token,
             api_timeout=api_timeout,
             botan_token=botan_token,
             name=name
         )
-
+        self.healthcheckio_token = healthcheckio_token
         self.moderators = moderators
         self._moderators_commands = []
 
+    async def still_alive(self):
+        async with aiohttp.ClientSession() as session:
+            async with session.get(self.healthcheckio_url.format(token=self.healthcheckio_token)):
+                pass
+        await asyncio.sleep(self.healthcheckio_interval)
+        await self.still_alive()
+
+    def run(self, debug=False, reload=None):
+        if self.healthcheckio_token:
+            loop = asyncio.get_event_loop()
+            loop.create_task(self.still_alive())
+        super(Bot, self).run(debug=debug, reload=reload)
+
     def add_moderators_command(self, regexp, fn):
         """
         Manually register regexp based command
diff --git a/bot/main.py b/bot/main.py
@@ -36,6 +36,7 @@ def get_moderators():
 
 bot = Bot(
     api_token=os.environ["BOT_TOKEN"],
+    healthcheckio_token=os.environ["HEALTHCHECKIO_TOKEN"],
     moderators=get_moderators()
 )
 
diff --git a/check.py b/check.py
@@ -0,0 +1,19 @@
+import asyncio
+
+
+async def primary_task(step=1):
+    print('primary task {step}'.format(step=step))
+    await asyncio.sleep(1)
+    await primary_task(step + 1)
+
+
+async def secondary_task(step=1):
+    print('secondary task {step}'.format(step=step))
+    await asyncio.sleep(1)
+    await secondary_task(step + 1)
+
+
+if __name__ == '__main__':
+    loop = asyncio.get_event_loop()
+    loop.create_task(secondary_task())
+    loop.run_until_complete(primary_task())

Original file line number	Diff line number	Diff line change
`@@ -36,6 +36,7 @@ def get_moderators():`
`36`	`36`
`37`	`37`	`bot = Bot(`
`38`	`38`	`api_token=os.environ["BOT_TOKEN"],`
	`39`	`+ healthcheckio_token=os.environ["HEALTHCHECKIO_TOKEN"],`
`39`	`40`	`moderators=get_moderators()`
`40`	`41`	`)`
`41`	`42`