Skip to content

Commit

Permalink
Use env var as connection string
Browse files Browse the repository at this point in the history
  • Loading branch information
Lun4m committed Jul 12, 2024
1 parent 5702f39 commit 729a6b0
Showing 1 changed file with 6 additions and 12 deletions.
18 changes: 6 additions & 12 deletions ingestion/src/main.rs
Original file line number Diff line number Diff line change
Expand Up @@ -10,19 +10,13 @@ async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
// TODO: use clap for argument parsing
let args: Vec<String> = std::env::args().collect();

if args.len() < 5 {
panic!(concat!(
"not enough args passed in. At least the group for the kafka queue,",
"and host, user, dbname needed, optionally password, for postgres"
))
if args.len() != 2 {
panic!(
"USAGE: lard_ingestion <kafka_group>\nEnv vars LARD_STRING and STINFO_STRING are also needed"
// env var format: host={} user={} dbname={} ...
)
}

let mut connect_string = format!("host={} user={} dbname={}", &args[2], &args[3], &args[4]);
if args.len() > 5 {
connect_string.push_str(" password=");
connect_string.push_str(&args[5])
};

// Permit tables handling (needs connection to stinfosys database)
let permit_tables = Arc::new(RwLock::new(permissions::fetch_permits().await?));
let background_permit_tables = permit_tables.clone();
Expand All @@ -46,7 +40,7 @@ async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
});

// Set up postgres connection pool
let manager = PostgresConnectionManager::new_from_stringlike(connect_string, NoTls)?;
let manager = PostgresConnectionManager::new_from_stringlike("LARD_STRING", NoTls)?;
let db_pool = bb8::Pool::builder().build(manager).await?;

// Spawn kvkafka reader
Expand Down

0 comments on commit 729a6b0

Please sign in to comment.