Skip to content

Commit 5638de0

Browse files
committed
Update dependencies
1 parent f389333 commit 5638de0

5 files changed

Lines changed: 25 additions & 16 deletions

File tree

.vscode/settings.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
"chrono",
66
"codegen",
77
"dotenv",
8+
"dotenvy",
89
"openworkers",
910
"serde",
1011
"sqlx"

Cargo.lock

Lines changed: 2 additions & 8 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

Cargo.toml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
[package]
22
name = "openworkers-logs"
3-
version = "0.1.2"
3+
version = "0.1.3"
44
edition = "2024"
55

66
[dependencies]
@@ -14,11 +14,11 @@ serde = { version = "1.0.197", features = ["derive"] }
1414
serde_json = "1.0.137"
1515
log = "0.4.21"
1616
env_logger = "0.11.6"
17-
dotenv = "0.15.0"
1817
base64 = "0.22.1"
1918
chrono = { version = "0.4", features = ["serde"] }
2019
uuid = { version = "1.0", features = ["serde", "v4"] }
2120
async-stream = "0.3"
21+
dotenvy = "0.15.7"
2222

2323
[profile.release]
2424
strip = true

src/db.rs

Lines changed: 16 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
use chrono::{DateTime, Utc};
22
use serde::Serialize;
3-
use sqlx::{postgres::PgPoolOptions, PgPool};
3+
use sqlx::{PgPool, postgres::PgPoolOptions};
44
use std::str::FromStr;
55
use uuid::Uuid;
66

@@ -40,7 +40,21 @@ pub struct LogEntry {
4040
}
4141

4242
pub async fn create_pool() -> Result<PgPool, sqlx::Error> {
43-
let database_url = std::env::var("DATABASE_URL").expect("DATABASE_URL must be set");
43+
let database_url = match std::env::var("DATABASE_URL") {
44+
Ok(url) => url,
45+
Err(_) => {
46+
let host = std::env::var("POSTGRES_HOST").expect("POSTGRES_HOST must be set");
47+
let port = std::env::var("POSTGRES_PORT").expect("POSTGRES_PORT must be set");
48+
let user = std::env::var("POSTGRES_USER").expect("POSTGRES_USER must be set");
49+
let password =
50+
std::env::var("POSTGRES_PASSWORD").expect("POSTGRES_PASSWORD must be set");
51+
let database = std::env::var("POSTGRES_DB").expect("POSTGRES_DB must be set");
52+
53+
log::debug!("DATABASE_URL not set, using POSTGRES_* env vars");
54+
55+
format!("postgres://{user}:{password}@{host}:{port}/{database}")
56+
}
57+
};
4458

4559
log::info!("Connecting to database...");
4660

src/main.rs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,17 +2,17 @@ mod db;
22
mod nats;
33
mod routes;
44

5-
use actix_web::{middleware, web, App, HttpServer};
5+
use actix_web::{App, HttpServer, middleware, web};
66
use chrono::Utc;
77
use futures::StreamExt;
88
use uuid::Uuid;
99

10-
use db::{create_pool, insert_log, LogEntry, LogLevel};
11-
use routes::{health, stream_worker_logs, AppState};
10+
use db::{LogEntry, LogLevel, create_pool, insert_log};
11+
use routes::{AppState, health, stream_worker_logs};
1212

1313
#[actix_web::main]
1414
async fn main() -> std::io::Result<()> {
15-
dotenv::dotenv().ok();
15+
dotenvy::dotenv().ok();
1616
env_logger::init();
1717

1818
log::info!("Starting OpenWorkers Logs Service...");

0 commit comments

Comments
 (0)