148 lines
3.4 KiB
Rust
148 lines
3.4 KiB
Rust
//!
|
|
//!
|
|
use super::{models, schema::member_classes};
|
|
use beteran_common_rust as bcr;
|
|
use diesel::prelude::*;
|
|
use diesel::result::Error;
|
|
|
|
///
|
|
pub struct Repository {}
|
|
|
|
impl std::fmt::Debug for Repository {
|
|
fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
|
|
f.debug_struct("Repository of member_classes").finish()
|
|
}
|
|
}
|
|
|
|
impl Default for Repository {
|
|
fn default() -> Self {
|
|
Self::new()
|
|
}
|
|
}
|
|
|
|
impl Repository {
|
|
///
|
|
pub fn new() -> Repository {
|
|
Repository {}
|
|
}
|
|
|
|
///
|
|
pub fn insert(
|
|
&self,
|
|
conn: &diesel::PgConnection,
|
|
new_member_class: &models::NewMemberClass,
|
|
) -> Result<models::MemberClass, Error> {
|
|
let inserted = diesel::insert_into(member_classes::table)
|
|
.values(new_member_class)
|
|
.get_result::<models::MemberClass>(conn)?;
|
|
|
|
Ok(inserted)
|
|
}
|
|
|
|
///
|
|
pub fn select(
|
|
&self,
|
|
conn: &diesel::PgConnection,
|
|
id: uuid::Uuid,
|
|
) -> Result<models::MemberClass, Error> {
|
|
member_classes::table
|
|
.find(id as uuid::Uuid)
|
|
.first::<models::MemberClass>(conn)
|
|
}
|
|
|
|
///
|
|
pub fn select_all_count(
|
|
&self,
|
|
conn: &diesel::PgConnection,
|
|
find_all: &models::FindAll,
|
|
) -> Result<i64, Error> {
|
|
let mut q = member_classes::table.into_boxed();
|
|
|
|
if let Some(sp) = &find_all.name_like {
|
|
q = q.filter(member_classes::dsl::name.like(sp));
|
|
}
|
|
|
|
q.count().get_result(conn)
|
|
}
|
|
|
|
///
|
|
pub fn select_all(
|
|
&self,
|
|
conn: &diesel::PgConnection,
|
|
find_all: &models::FindAll,
|
|
) -> Result<Vec<models::MemberClass>, Error> {
|
|
let mut q = member_classes::table.into_boxed();
|
|
|
|
if let Some(sp) = &find_all.name_like {
|
|
q = q.filter(member_classes::dsl::name.like(sp));
|
|
}
|
|
|
|
if let Some(p) = &find_all.pagination {
|
|
let page = p.page.unwrap_or(1);
|
|
|
|
if let Some(page_size) = p.page_size {
|
|
q = q.offset(((page - 1) * page_size) as i64);
|
|
q = q.limit(page_size as i64);
|
|
}
|
|
}
|
|
if let Some(orderbys) = &find_all.sorts {
|
|
for s in orderbys {
|
|
match s {
|
|
bcr::models::pagination::Sort::ASC(property) => match property.as_str() {
|
|
"name" => {
|
|
q = q.order_by(member_classes::name.asc());
|
|
}
|
|
"created_at" => {
|
|
q = q.order_by(member_classes::created_at.asc());
|
|
}
|
|
"updated_at" => {
|
|
q = q.order_by(member_classes::updated_at.asc());
|
|
}
|
|
|
|
_ => {}
|
|
},
|
|
bcr::models::pagination::Sort::DESC(property) => match property.as_str() {
|
|
"name" => {
|
|
q = q.order_by(member_classes::name.desc());
|
|
}
|
|
"created_at" => {
|
|
q = q.order_by(member_classes::created_at.desc());
|
|
}
|
|
"updated_at" => {
|
|
q = q.order_by(member_classes::updated_at.desc());
|
|
}
|
|
|
|
_ => {}
|
|
},
|
|
};
|
|
}
|
|
}
|
|
|
|
q.load::<models::MemberClass>(conn)
|
|
}
|
|
|
|
///
|
|
pub fn update(
|
|
&self,
|
|
conn: &diesel::PgConnection,
|
|
id: uuid::Uuid,
|
|
modify: &models::ModifyMemberClass,
|
|
) -> Result<u64, Error> {
|
|
use member_classes::dsl;
|
|
|
|
diesel::update(dsl::member_classes.filter(dsl::id.eq(id)))
|
|
.set(modify)
|
|
.execute(conn)
|
|
.map(|c| c as u64)
|
|
}
|
|
|
|
///
|
|
pub fn delete(&self, conn: &diesel::PgConnection, id: uuid::Uuid) -> Result<u64, Error> {
|
|
use member_classes::dsl;
|
|
|
|
diesel::delete(member_classes::table.filter(dsl::id.eq(id)))
|
|
.execute(conn)
|
|
.map(|c| c as u64)
|
|
}
|
|
}
|