readonly/mirror-chhoto-url
1
0
Fork 0
mirror of https://github.com/minoplhy/chhoto-url.git synced 2024-11-22 09:16:46 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Use env variable for opening database

Browse Source
This commit is contained in:
SinTan1729 2023-04-03 18:58:36 -05:00
parent 71d7253248
commit a14dbf5cd2
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
actix/src/main.rs
View File

@ -22,10 +22,8 @@ struct AppState {
async fn add_link(req: String, data: web::Data<AppState>) -> HttpResponse {
let out = utils::add_link(req, &data.db);
if out.0 {
println!("ok{}", out.1);
HttpResponse::Ok().body(out.1)
} else {
println!("bad{}", out.1);
HttpResponse::BadRequest().body(out.1)
}
}
@ -74,7 +72,7 @@ async fn main() -> std::io::Result<()> {
HttpServer::new(|| {
App::new()
.app_data(web::Data::new(AppState {
db: database::open_db("./urls.sqlite".to_string()),
db: database::open_db(env::var("db_url").unwrap_or("./urls.sqlite".to_string())),
}))
.service(link_handler)
.service(error404)