Skip to content

Commit

Permalink
[ENH] Allow rust worker to load config path from env var (#2109)
Browse files Browse the repository at this point in the history
## 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
  • Loading branch information
HammadB authored May 1, 2024
1 parent 1dd9674 commit 64f96df
Showing 1 changed file with 14 additions and 2 deletions.
16 changes: 14 additions & 2 deletions rust/worker/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -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 =
Expand Down Expand Up @@ -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();

Expand Down

0 comments on commit 64f96df

Please sign in to comment.