From 64f96df9ebd67780923b953b6ca8064f510c03c2 Mon Sep 17 00:00:00 2001
From: Hammad Bashir <HammadB@users.noreply.github.com>
Date: Wed, 1 May 2024 11:45:20 -0700
Subject: [PATCH] [ENH] Allow rust worker to load config path from env var
 (#2109)

## Description of changes

*Summarize the changes made by this PR.*
 - Improvements & Bug fixes
	 - Load from evn path for
 - New functionality
	 - None

## Test plan
*How are these changes tested?*
- [x] Tests pass locally with `pytest` for python, `yarn test` for js,
`cargo test` for rust

## Documentation Changes
None
---
 rust/worker/src/lib.rs | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/rust/worker/src/lib.rs b/rust/worker/src/lib.rs
index 4357a59ea9d..ee06489601b 100644
--- a/rust/worker/src/lib.rs
+++ b/rust/worker/src/lib.rs
@@ -21,12 +21,19 @@ use memberlist::MemberlistProvider;
 use tokio::select;
 use tokio::signal::unix::{signal, SignalKind};
 
+const CONFIG_PATH_ENV_VAR: &str = "CONFIG_PATH";
+
 mod chroma_proto {
     tonic::include_proto!("chroma");
 }
 
 pub async fn query_service_entrypoint() {
-    let config = config::RootConfig::load();
+    // Check if the config path is set in the env var
+    let config = match std::env::var(CONFIG_PATH_ENV_VAR) {
+        Ok(config_path) => config::RootConfig::load_from_path(&config_path),
+        Err(_) => config::RootConfig::load(),
+    };
+
     let config = config.query_service;
     let system: system::System = system::System::new();
     let dispatcher =
@@ -80,7 +87,12 @@ pub async fn query_service_entrypoint() {
 }
 
 pub async fn compaction_service_entrypoint() {
-    let config = config::RootConfig::load();
+    // Check if the config path is set in the env var
+    let config = match std::env::var(CONFIG_PATH_ENV_VAR) {
+        Ok(config_path) => config::RootConfig::load_from_path(&config_path),
+        Err(_) => config::RootConfig::load(),
+    };
+
     let config = config.compaction_service;
     let system: system::System = system::System::new();