From a2f88e29e6581c2e17ea1cfdb1ff75c76a7e570d Mon Sep 17 00:00:00 2001
From: Mark Veidemanis <m@zm.is>
Date: Fri, 23 Sep 2022 07:20:30 +0100
Subject: [PATCH] Implement uvloop

---
 docker/requirements.txt |  2 ++
 monolith.py             | 12 +++++++-----
 requirements.txt        |  2 ++
 sources/dis.py          |  1 +
 4 files changed, 12 insertions(+), 5 deletions(-)

diff --git a/docker/requirements.txt b/docker/requirements.txt
index 35412d3..ad4d775 100644
--- a/docker/requirements.txt
+++ b/docker/requirements.txt
@@ -19,3 +19,5 @@ nltk
 spacy
 python-Levenshtein
 orjson
+uvloop
+numba
diff --git a/monolith.py b/monolith.py
index 1eb559b..b80cd90 100644
--- a/monolith.py
+++ b/monolith.py
@@ -1,4 +1,7 @@
 import asyncio
+import sys
+import uvloop
+
 from os import getenv
 
 import util
@@ -6,6 +9,9 @@ from sources.ch4 import Chan4
 from sources.dis import DiscordClient
 from sources.ingest import Ingest
 
+# Use UVLoop
+asyncio.set_event_loop_policy(uvloop.EventLoopPolicy())
+
 log = util.get_logger("monolith")
 
 modules_enabled = getenv("MODULES_ENABLED", False)
@@ -16,22 +22,18 @@ if not token:
 
 
 async def main(loop):
-    log.info("Starting Discord handler.")
     client = DiscordClient()
     loop.create_task(client.start(token))
 
-    log.info("Starting 4chan handler.")
     chan = Chan4()
     loop.create_task(chan.run())
 
-    log.info("Starting ingest handler.")
     ingest = Ingest()
     loop.create_task(ingest.run())
 
-
 loop = asyncio.get_event_loop()
 loop.create_task(main(loop))
-# reactor.run()
+
 try:
     loop.run_forever()
 except KeyboardInterrupt:
diff --git a/requirements.txt b/requirements.txt
index d020596..ca35ea7 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -20,3 +20,5 @@ nltk
 spacy
 python-Levenshtein
 orjson
+uvloop
+numba
diff --git a/sources/dis.py b/sources/dis.py
index c187cf7..934b3b1 100644
--- a/sources/dis.py
+++ b/sources/dis.py
@@ -14,6 +14,7 @@ class DiscordClient(discord.Client):
         self.did_something = False
         name = self.__class__.__name__
         self.log = util.get_logger(name)
+        self.log.info("Starting Discord client")
 
         super().__init__(*args, **kwargs)