beteran-server-service/src/repositories/member_bank_withdraw/repository.rs

//!
//!
use super::{models, schema::member_bank_withdraws};
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_bank_withdraws")
      .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_resource: &models::NewMemberBankWithdraw,
  ) -> Result<uuid::Uuid, Error> {
    use member_bank_withdraws::dsl;

    let inserted = diesel::insert_into(member_bank_withdraws::table)
      .values(new_resource)
      .returning(dsl::id)
      .get_result(conn)?;

    Ok(inserted)
  }

  ///
  pub fn select(
    &self,
    conn: &diesel::PgConnection,
    id: uuid::Uuid,
  ) -> Result<Option<models::MemberBankWithdraw>, Error> {
    match member_bank_withdraws::table
      .find(id as uuid::Uuid)
      .first::<models::MemberBankWithdraw>(conn)
    {
      Ok(m) => Ok(Some(m)),
      Err(e) => match e {
        diesel::result::Error::NotFound => Ok(None),
        _ => Err(e),
      },
    }
  }

  ///
  pub fn select_all_count(
    &self,
    conn: &diesel::PgConnection,
    find_all: &models::FindAll,
  ) -> Result<i64, Error> {
    let mut q = member_bank_withdraws::table.into_boxed();

    if let Some(s) = &find_all.search {
      if let Some(sp) = s.member_id {
        q = q.filter(member_bank_withdraws::dsl::member_id.eq(sp));
      }
      if let Some(sp) = &s.bank_name_like {
        q = q.filter(member_bank_withdraws::dsl::bank_name.like(sp));
      }
      if let Some(sp) = &s.name_like {
        q = q.filter(member_bank_withdraws::dsl::name.like(sp));
      }
      if let Some(sp) = &s.account_number_like {
        q = q.filter(member_bank_withdraws::dsl::account_number.like(sp));
      }
      if let Some(sp) = s.amount {
        q = q.filter(member_bank_withdraws::dsl::amount.eq(sp));
      }
      if let Some(sp) = &s.memo_like {
        q = q.filter(member_bank_withdraws::dsl::memo.like(sp));
      }
      if let Some(sp) = s.state {
        q = q.filter(member_bank_withdraws::dsl::state.eq(sp));
      }
    }

    q.count().get_result(conn)
  }

  ///
  pub fn select_all(
    &self,
    conn: &diesel::PgConnection,
    find_all: &models::FindAll,
  ) -> Result<Vec<models::MemberBankWithdraw>, Error> {
    let mut q = member_bank_withdraws::table.into_boxed();

    if let Some(s) = &find_all.search {
      if let Some(sp) = s.member_id {
        q = q.filter(member_bank_withdraws::dsl::member_id.eq(sp));
      }
      if let Some(sp) = &s.bank_name_like {
        q = q.filter(member_bank_withdraws::dsl::bank_name.like(sp));
      }
      if let Some(sp) = &s.name_like {
        q = q.filter(member_bank_withdraws::dsl::name.like(sp));
      }
      if let Some(sp) = &s.account_number_like {
        q = q.filter(member_bank_withdraws::dsl::account_number.like(sp));
      }
      if let Some(sp) = s.amount {
        q = q.filter(member_bank_withdraws::dsl::amount.eq(sp));
      }
      if let Some(sp) = &s.memo_like {
        q = q.filter(member_bank_withdraws::dsl::memo.like(sp));
      }
      if let Some(sp) = s.state {
        q = q.filter(member_bank_withdraws::dsl::state.eq(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::pagination::Sort::ASC(property) => match property.as_str() {
            "member_id" => {
              q = q.order_by(member_bank_withdraws::member_id.asc());
            }
            "bank_name" => {
              q = q.order_by(member_bank_withdraws::bank_name.asc());
            }
            "name" => {
              q = q.order_by(member_bank_withdraws::name.asc());
            }
            "account_number" => {
              q = q.order_by(member_bank_withdraws::account_number.asc());
            }
            "amount" => {
              q = q.order_by(member_bank_withdraws::amount.asc());
            }
            "state" => {
              q = q.order_by(member_bank_withdraws::state.asc());
            }
            "created_at" => {
              q = q.order_by(member_bank_withdraws::created_at.asc());
            }
            "updated_at" => {
              q = q.order_by(member_bank_withdraws::updated_at.asc());
            }
            _ => {}
          },
          bcr::pagination::Sort::DESC(property) => match property.as_str() {
            "member_id" => {
              q = q.order_by(member_bank_withdraws::member_id.desc());
            }
            "bank_name" => {
              q = q.order_by(member_bank_withdraws::bank_name.desc());
            }
            "name" => {
              q = q.order_by(member_bank_withdraws::name.desc());
            }
            "account_number" => {
              q = q.order_by(member_bank_withdraws::account_number.desc());
            }
            "amount" => {
              q = q.order_by(member_bank_withdraws::amount.desc());
            }
            "state" => {
              q = q.order_by(member_bank_withdraws::state.desc());
            }
            "created_at" => {
              q = q.order_by(member_bank_withdraws::created_at.desc());
            }
            "updated_at" => {
              q = q.order_by(member_bank_withdraws::updated_at.desc());
            }

            _ => {}
          },
        };
      }
    }

    q.load::<models::MemberBankWithdraw>(conn)
  }

  ///
  pub fn update(
    &self,
    conn: &diesel::PgConnection,
    id: uuid::Uuid,
    modify: &models::ModifyMemberBankWithdraw,
  ) -> Result<u64, Error> {
    use member_bank_withdraws::dsl;

    diesel::update(dsl::member_bank_withdraws.filter(dsl::id.eq(id)))
      .set(modify)
      .execute(conn)
      .map(|c| c as u64)
  }

  ///
  pub fn update_state(
    &self,
    conn: &diesel::PgConnection,
    id: uuid::Uuid,
    modify: &models::ModifyMemberBankWithdrawForState,
  ) -> Result<u64, Error> {
    use member_bank_withdraws::dsl;

    diesel::update(dsl::member_bank_withdraws.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_bank_withdraws::dsl;

    diesel::delete(member_bank_withdraws::table.filter(dsl::id.eq(id)))
      .execute(conn)
      .map(|c| c as u64)
  }
}