feat: config huey to use redis (#424)

basedosdados · Oct 4, 2023 · 653aae6 · 653aae6
1 parent 970f7c9
commit 653aae6
Show file tree

Hide file tree

Showing 3 changed files with 42 additions and 3 deletions.
diff --git a/basedosdados_api/settings/base.py b/basedosdados_api/settings/base.py
@@ -409,14 +409,21 @@
 
 HUEY = {
     "name": "api.queue",
-    "huey_class": "huey.SqliteHuey",
+    "huey_class": "huey.RedisHuey",
     "results": True,
     "immediate": False,
-    "filename": path.join(BASE_DIR, "queue.sqlite3"),
+    "connection": {
+        "host": getenv("REDIS_URL", "localhost"),
+        "port": getenv("REDIS_PORT", 6379),
+        "db": getenv("REDIS_DB", 1),
+        "read_timeout": 1,
+        "connection_pool": None,
+    },
     "consumer": {
         "workers": 2,
         "worker_type": "thread",
         "check_worker_health": True,
         "periodic": True,
     },
 }
+HUEY = None
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -32,6 +32,7 @@ google-cloud-bigquery = "^3.11.3"
 faker = "^19.6.1"
 loguru = "^0.7.2"
 huey = "^2.5.0"
+redis = "^5.0.1"
 
 [tool.poetry.group.dev.dependencies]
 black = "^22.3.0"