🔨 Refactors into packages models and dtos
This commit is contained in:
17
src/main.rs
17
src/main.rs
@@ -1,12 +1,10 @@
|
||||
mod api;
|
||||
mod dtos;
|
||||
mod models;
|
||||
mod schema;
|
||||
mod dtos;
|
||||
|
||||
use crate::api::tracker;
|
||||
use crate::models::Hit;
|
||||
use crate::schema::hits::dsl::*;
|
||||
use diesel::{Connection, PgConnection, QueryDsl, RunQueryDsl, SelectableHelper};
|
||||
use diesel::{Connection, PgConnection};
|
||||
use std::env;
|
||||
|
||||
#[macro_use]
|
||||
@@ -22,19 +20,12 @@ fn rocket() -> _ {
|
||||
dotenv::dotenv().ok();
|
||||
|
||||
let database_url = env::var("DATABASE_URL").expect("DATABASE_URL must be set");
|
||||
let mut db = PgConnection::establish(&database_url)
|
||||
let db = PgConnection::establish(&database_url)
|
||||
.expect(&format!("Error connecting to {}", database_url));
|
||||
|
||||
let results = hits
|
||||
.limit(5)
|
||||
.select(Hit::as_select())
|
||||
.load(&mut db)
|
||||
.expect("Error loading hits");
|
||||
println!("results: {}", results.len());
|
||||
|
||||
let app_data = models::AppState::new(db);
|
||||
rocket::build()
|
||||
.manage(app_data)
|
||||
.mount("/", routes![index])
|
||||
.mount("/tracker", routes![tracker::index])
|
||||
.mount("/tracker", routes![tracker::index, tracker::get, tracker::create])
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user