From e22a3dcac76f66d64eef49f32dc6367e67147c51 Mon Sep 17 00:00:00 2001
From: MohamedBassem <me@mbassem.com>
Date: Wed, 27 Mar 2024 12:06:03 +0000
Subject: feature: Allow specifying a specific redis db to use. Fixes #13

---
 packages/shared/config.ts | 2 ++
 packages/shared/queues.ts | 1 +
 2 files changed, 3 insertions(+)

(limited to 'packages/shared')

diff --git a/packages/shared/config.ts b/packages/shared/config.ts
index ff077147..edc96627 100644
--- a/packages/shared/config.ts
+++ b/packages/shared/config.ts
@@ -17,6 +17,7 @@ const allEnv = z.object({
   INFERENCE_IMAGE_MODEL: z.string().default("gpt-4-vision-preview"),
   REDIS_HOST: z.string().default("localhost"),
   REDIS_PORT: z.coerce.number().default(6379),
+  REDIS_DB_IDX: z.coerce.number().optional(),
   CRAWLER_HEADLESS_BROWSER: stringBool("true"),
   BROWSER_EXECUTABLE_PATH: z.string().optional(), // If not set, the system's browser will be used
   BROWSER_USER_DATA_DIR: z.string().optional(),
@@ -46,6 +47,7 @@ const serverConfigSchema = allEnv.transform((val) => {
     bullMQ: {
       redisHost: val.REDIS_HOST,
       redisPort: val.REDIS_PORT,
+      redisDBIdx: val.REDIS_DB_IDX,
     },
     crawler: {
       headlessBrowser: val.CRAWLER_HEADLESS_BROWSER,
diff --git a/packages/shared/queues.ts b/packages/shared/queues.ts
index a2cbeceb..b264e2c4 100644
--- a/packages/shared/queues.ts
+++ b/packages/shared/queues.ts
@@ -5,6 +5,7 @@ import serverConfig from "./config";
 export const queueConnectionDetails = {
   host: serverConfig.bullMQ.redisHost,
   port: serverConfig.bullMQ.redisPort,
+  db: serverConfig.bullMQ.redisDBIdx,
 };
 
 // Link Crawler
-- 
cgit v1.2.3-70-g09d2