78 lines
2.0 KiB
Rust
78 lines
2.0 KiB
Rust
use diesel::prelude::*;
|
|
use rocket::{http::Status, serde::json::Json, State};
|
|
|
|
use kairo_common::{postgres, schema::antennas, Antenna};
|
|
|
|
#[rocket::get("/id/<id>")]
|
|
pub fn get_by_id(db_pool: &State<postgres::DbPool>, id: String) -> Option<Json<Antenna>> {
|
|
let mut db = db_pool.get().unwrap();
|
|
|
|
let res = antennas::table
|
|
.select(antennas::all_columns)
|
|
.find(id)
|
|
.get_result::<Antenna>(&mut db);
|
|
|
|
match res {
|
|
Ok(v) => Some(rocket::serde::json::Json(v)),
|
|
_ => None,
|
|
}
|
|
}
|
|
|
|
#[rocket::get("/")]
|
|
pub fn get_list(db_pool: &State<postgres::DbPool>) -> Json<Vec<Antenna>> {
|
|
let mut db = db_pool.get().unwrap();
|
|
|
|
let res = antennas::table
|
|
.select(antennas::all_columns)
|
|
.load::<Antenna>(&mut db);
|
|
|
|
match res {
|
|
Ok(v) => rocket::serde::json::Json(v),
|
|
_ => rocket::serde::json::Json(vec![]),
|
|
}
|
|
}
|
|
|
|
#[rocket::post("/new", format = "json", data = "<antenna>")]
|
|
pub fn new(db_pool: &State<postgres::DbPool>, antenna: Json<Antenna>) -> Status {
|
|
let mut db = db_pool.get().unwrap();
|
|
|
|
let res = diesel::insert_into(antennas::table)
|
|
.values(antenna.0)
|
|
.execute(&mut db);
|
|
|
|
match res {
|
|
Ok(_) => Status::Ok,
|
|
_ => Status::NotAcceptable,
|
|
}
|
|
}
|
|
|
|
#[rocket::patch("/update", format = "json", data = "<antenna>")]
|
|
pub fn update(db_pool: &State<postgres::DbPool>, antenna: Json<Antenna>) -> Status {
|
|
let mut db = db_pool.get().unwrap();
|
|
|
|
let res = diesel::update(antennas::table)
|
|
.filter(antennas::id.eq(antenna.id.clone()))
|
|
.set(antenna.0)
|
|
.execute(&mut db);
|
|
|
|
match res {
|
|
Ok(0) => Status::NotModified,
|
|
Ok(1) => Status::Ok,
|
|
_ => Status::BadRequest,
|
|
}
|
|
}
|
|
|
|
#[rocket::delete("/delete/<id>")]
|
|
pub fn delete(db_pool: &State<postgres::DbPool>, id: String) -> Status {
|
|
let mut db = db_pool.get().unwrap();
|
|
|
|
let res = diesel::delete(antennas::table)
|
|
.filter( antennas::id.eq(id) )
|
|
.execute(&mut db);
|
|
|
|
match res {
|
|
Ok(1) => Status::Ok,
|
|
_ => Status::BadRequest,
|
|
}
|
|
}
|