Add serde renames and make structs use proper rust variable names

This commit is contained in:
Sebastian H. Gabrielli 2023-12-25 15:58:18 +01:00
parent 56f1a1375e
commit cde27b56d9
2 changed files with 47 additions and 34 deletions

View File

@ -1,20 +1,22 @@
use rocket::serde::{Serialize, Deserialize}; use rocket::serde::{Serialize, Deserialize};
use sea_orm::*; use sea_orm::*;
use super::entities::{prelude::*, *}; use super::entities::{*};
#[derive(Serialize, Deserialize)] #[derive(Serialize, Deserialize)]
pub struct RfidCard { pub struct RfidCard {
pub cardId: String, #[serde(rename = "cardId")]
pub cardComment: String pub card_id: String,
#[serde(rename = "cardComment")]
pub card_comment: String
} }
// Implement `.into()` to auto map values from the database model to the local struct // Implement `.into()` to auto map values from the database model to the local struct
impl From<rfid_cards::Model> for RfidCard { impl From<rfid_cards::Model> for RfidCard {
fn from(rfid_card: rfid_cards::Model) -> RfidCard { fn from(rfid_card: rfid_cards::Model) -> RfidCard {
RfidCard { RfidCard {
cardId: rfid_card.card_id, card_id: rfid_card.card_id,
cardComment: rfid_card.card_comment, card_comment: rfid_card.card_comment,
} }
} }
} }
@ -22,13 +24,18 @@ impl From<rfid_cards::Model> for RfidCard {
#[derive(Serialize, Deserialize)] #[derive(Serialize, Deserialize)]
pub struct Member { pub struct Member {
pub id: i32, pub id: i32,
pub ntnuUsername: String, #[serde(rename = "ntnuUsername")]
pub firstName: String, pub ntnu_username: String,
pub lastName: String, #[serde(rename = "firstName")]
pub first_name: String,
#[serde(rename = "lastName")]
pub last_name: String,
pub email: String, pub email: String,
pub balance: i32, pub balance: i32,
pub imagePreference: String, #[serde(rename = "imagePreference")]
pub rfidCards: Vec<RfidCard> pub image_preference: String,
#[serde(rename = "lastName")]
pub rfid_cards: Vec<RfidCard>
} }
// Create the `.into()` functionality to auto map values from the database model to the local struct // Create the `.into()` functionality to auto map values from the database model to the local struct
@ -37,13 +44,13 @@ impl From<members::Model> for Member {
fn from(member: members::Model) -> Member { fn from(member: members::Model) -> Member {
Member { Member {
id: member.id, id: member.id,
ntnuUsername: member.ntnu_username, ntnu_username: member.ntnu_username,
firstName: member.first_name, first_name: member.first_name,
lastName: member.last_name, last_name: member.last_name,
email: member.email, email: member.email,
balance: member.balance, balance: member.balance,
imagePreference: member.image_preference, image_preference: member.image_preference,
rfidCards: Vec::new(), rfid_cards: Vec::new(),
} }
} }
} }
@ -53,9 +60,9 @@ impl From<members::Model> for MinimalMember {
fn from(member: members::Model) -> MinimalMember { fn from(member: members::Model) -> MinimalMember {
MinimalMember { MinimalMember {
id: member.id, id: member.id,
ntnuUsername: member.ntnu_username, ntnu_username: member.ntnu_username,
firstName: member.first_name, first_name: member.first_name,
lastName: member.last_name, last_name: member.last_name,
email: member.email, email: member.email,
} }
} }
@ -65,9 +72,9 @@ impl From<members::Model> for MinimalMember {
impl Into<members::ActiveModel> for rocket::serde::json::Json<MinimalMemberWithoutId> { impl Into<members::ActiveModel> for rocket::serde::json::Json<MinimalMemberWithoutId> {
fn into(self) -> members::ActiveModel { fn into(self) -> members::ActiveModel {
members::ActiveModel { members::ActiveModel {
ntnu_username: ActiveValue::Set(self.ntnuUsername.to_owned()), ntnu_username: ActiveValue::Set(self.ntnu_username.to_owned()),
first_name: ActiveValue::Set(self.firstName.to_owned()), first_name: ActiveValue::Set(self.first_name.to_owned()),
last_name: ActiveValue::Set(self.lastName.to_owned()), last_name: ActiveValue::Set(self.last_name.to_owned()),
email: ActiveValue::Set(self.email.to_owned()), email: ActiveValue::Set(self.email.to_owned()),
balance: ActiveValue::Set(0), balance: ActiveValue::Set(0),
image_preference: ActiveValue::Set("".to_string()), image_preference: ActiveValue::Set("".to_string()),
@ -79,17 +86,23 @@ impl Into<members::ActiveModel> for rocket::serde::json::Json<MinimalMemberWitho
#[derive(Serialize, Deserialize)] #[derive(Serialize, Deserialize)]
pub struct MinimalMember { pub struct MinimalMember {
pub id: i32, pub id: i32,
pub ntnuUsername: String, #[serde(rename = "ntnuUsername")]
pub firstName: String, pub ntnu_username: String,
pub lastName: String, #[serde(rename = "firstName")]
pub first_name: String,
#[serde(rename = "lastName")]
pub last_name: String,
pub email: String, pub email: String,
} }
#[derive(Serialize, Deserialize)] #[derive(Serialize, Deserialize)]
pub struct MinimalMemberWithoutId { pub struct MinimalMemberWithoutId {
pub ntnuUsername: String, #[serde(rename = "ntnuUsername")]
pub firstName: String, pub ntnu_username: String,
pub lastName: String, #[serde(rename = "firstName")]
pub first_name: String,
#[serde(rename = "lastName")]
pub last_name: String,
pub email: String, pub email: String,
} }

View File

@ -57,7 +57,7 @@ pub async fn get_members(db: &State<DatabaseConnection>) -> Result<Json<Multiple
let rfid_cards: Vec<RfidCard> = rfid_cards.into_iter().map(RfidCard::from).collect(); let rfid_cards: Vec<RfidCard> = rfid_cards.into_iter().map(RfidCard::from).collect();
// Add these cards to the member's rfidCards field // Add these cards to the member's rfidCards field
member.rfidCards = rfid_cards; member.rfid_cards = rfid_cards;
} }
@ -75,7 +75,7 @@ pub async fn add_member(db: &State<DatabaseConnection>, minimal_member_without_i
// Check if a member with the same NTNU username already exists // Check if a member with the same NTNU username already exists
let matching_member: Option<members::Model> = Members::find() // Find a member in the "Members" table let matching_member: Option<members::Model> = Members::find() // Find a member in the "Members" table
.filter(members::Column::NtnuUsername.eq(minimal_member_without_id.ntnuUsername.to_owned())) // Filter by the provided username in the NtnuUsername column .filter(members::Column::NtnuUsername.eq(minimal_member_without_id.ntnu_username.to_owned())) // Filter by the provided username in the NtnuUsername column
.one(db) // We only care about one result .one(db) // We only care about one result
.await?; // Wait for the result .await?; // Wait for the result
@ -113,19 +113,19 @@ pub async fn add_member(db: &State<DatabaseConnection>, minimal_member_without_i
Ok(Json(member)) Ok(Json(member))
} }
#[get("/member/<memberId>")] #[get("/member/<member_id>")]
pub async fn get_member_by_id(db: &State<DatabaseConnection>, memberId: i32) -> Result<Json<Member>, ErrorResponder> { pub async fn get_member_by_id(db: &State<DatabaseConnection>, member_id: i32) -> Result<Json<Member>, ErrorResponder> {
let db = db as &DatabaseConnection; let db = db as &DatabaseConnection;
// Create an empty variable to store the resulting member info // Create an empty variable to store the resulting member info
let database_member; let database_member;
// Search the database for the member based on the ID // Search the database for the member based on the ID
match Members::find_by_id(memberId).one(db).await? { match Members::find_by_id(member_id).one(db).await? {
Some(model) => database_member = model, // If the member is found, add the info to the previously created variable Some(model) => database_member = model, // If the member is found, add the info to the previously created variable
None => { None => {
return Err( return Err(
ErrorResponder { ErrorResponder {
message: format!("Failed to fetch member by ID. memberId: {}", memberId) // If it is not found return with this error message: format!("Failed to fetch member by ID. memberId: {}", member_id) // If it is not found return with this error
} }
); );
}, },
@ -137,7 +137,7 @@ pub async fn get_member_by_id(db: &State<DatabaseConnection>, memberId: i32) ->
// Create a member from the database model and append the RFID cards // Create a member from the database model and append the RFID cards
let mut member: Member = database_member.into(); let mut member: Member = database_member.into();
member.rfidCards = rfids; member.rfid_cards = rfids;
Ok(Json(member)) Ok(Json(member))
} }