diff options
Diffstat (limited to 'src/db.rs')
-rw-r--r-- | src/db.rs | 62 |
1 files changed, 62 insertions, 0 deletions
@@ -16,6 +16,13 @@ along with this program. If not, see <https://www.gnu.org/licenses/>. */ +use crate::{ + models, + schema::*, +}; + +use diesel::prelude::*; + use rocket_contrib::{ database, databases::diesel, @@ -23,3 +30,58 @@ use rocket_contrib::{ #[database("main")] pub struct Connection(diesel::PgConnection); + +impl Connection { + pub fn create_screening( + &self, + room: &models::room::Room, + film: &models::film::Film, + date: chrono::NaiveDate, + start_time: chrono::NaiveTime, + end_time: chrono::NaiveTime) -> QueryResult<usize> + { + let s = models::screening::NewScreening { + room_id: room.id, + film_id: film.id, + date: date, + start_time: start_time, + end_time: end_time, + }; + + diesel::insert_into(screenings::table) + .values(&s) + .execute(&**self) + } + + pub fn get_screenings(&self) -> QueryResult<Vec<(models::screening::Screening, Option<models::film::Film>, Option<models::room::Room>)>> { + use crate::schema::screenings::dsl::*; + screenings + .left_join(films::table) + .left_join(rooms::table) + .get_results(&**self) + } + + pub fn get_rooms(&self) -> QueryResult<Vec<models::room::Room>> { + use crate::schema::rooms::dsl::*; + rooms.load(&**self) + } + + pub fn get_room_by_name(&self, room_name: &str) -> QueryResult<models::room::Room> { + use crate::schema::rooms::dsl::*; + rooms + .filter(name.eq(&room_name)) + .get_result(&**self) + } + + pub fn get_films(&self) -> QueryResult<Vec<models::film::Film>> { + use crate::schema::films::dsl::*; + films.load(&**self) + } + + pub fn get_film_by_title(&self, film_title: &str) -> QueryResult<models::film::Film> { + use crate::schema::films::dsl::*; + films + .filter(title.eq(&film_title)) + .get_result(&**self) + } +} |