diff --git a/src/main.rs b/src/main.rs index 8e5bed4..49137f3 100644 --- a/src/main.rs +++ b/src/main.rs @@ -58,14 +58,17 @@ async fn main() -> std::io::Result<()> { let site; let temp; let logs; + let db; if Cli::parse().dev { site = "./public"; logs = "/tmp/tfcsite"; temp = "/tmp/tfcsite"; + db = "sqlite://./data.db"; } else { site = "../public"; logs = "/storage/logs/tfcsite"; temp = "/tmp/tfcsite"; + db = "sqlite:///storage/data.db"; } std::fs::create_dir_all(&logs)?; @@ -107,7 +110,7 @@ async fn main() -> std::io::Result<()> { info!("starting HTTP server at http://localhost:4242"); - let conn = SqliteConnection::connect("sqlite://./data.db") + let conn = SqliteConnection::connect(db) .await .expect("Couldn't connect sqlite db"); let data = web::Data::new(conn);