Try as_ref

This commit is contained in:
Billy Chan 2021-10-15 22:48:49 +08:00
parent 23155dced1
commit 351a78c784
No known key found for this signature in database
GPG Key ID: A2D690CAC7DF3CC7

View File

@ -60,8 +60,7 @@ impl ActiveModelBehavior for ActiveModel {
fn before_save(self, insert: bool) -> Result<Self, DbErr> { fn before_save(self, insert: bool) -> Result<Self, DbErr> {
use rust_decimal_macros::dec; use rust_decimal_macros::dec;
if let Some(price) = self.price.clone().take() { if self.price.as_ref() == &dec!(0) {
if price == dec!(0) {
Err(DbErr::Custom(format!( Err(DbErr::Custom(format!(
"[before_save] Invalid Price, insert: {}", "[before_save] Invalid Price, insert: {}",
insert insert
@ -69,18 +68,11 @@ impl ActiveModelBehavior for ActiveModel {
} else { } else {
Ok(self) Ok(self)
} }
} else {
Err(DbErr::Custom(format!(
"[before_save] Price must be set, insert: {}",
insert
)))
}
} }
fn after_save(self, insert: bool) -> Result<Self, DbErr> { fn after_save(self, insert: bool) -> Result<Self, DbErr> {
use rust_decimal_macros::dec; use rust_decimal_macros::dec;
if let Some(price) = dbg!(self.price.clone()).take() { if self.price.as_ref() < &dec!(0) {
if price < dec!(0) {
Err(DbErr::Custom(format!( Err(DbErr::Custom(format!(
"[after_save] Invalid Price, insert: {}", "[after_save] Invalid Price, insert: {}",
insert insert
@ -88,37 +80,23 @@ impl ActiveModelBehavior for ActiveModel {
} else { } else {
Ok(self) Ok(self)
} }
} else {
Err(DbErr::Custom(format!(
"[after_save] Price must be set, insert: {}",
insert
)))
}
} }
fn before_delete(self) -> Result<Self, DbErr> { fn before_delete(self) -> Result<Self, DbErr> {
if let Some(name) = self.name.clone().take() { if self.name.as_ref().contains("(err_on_before_delete)") {
if name.contains("(err_on_before_delete)") {
Err(DbErr::Custom( Err(DbErr::Custom(
"[before_delete] Cannot be deleted".to_owned(), "[before_delete] Cannot be deleted".to_owned(),
)) ))
} else { } else {
Ok(self) Ok(self)
} }
} else {
Err(DbErr::Custom("[before_delete] Name must be set".to_owned()))
}
} }
fn after_delete(self) -> Result<Self, DbErr> { fn after_delete(self) -> Result<Self, DbErr> {
if let Some(name) = self.name.clone().take() { if self.name.as_ref().contains("(err_on_after_delete)") {
if name.contains("(err_on_after_delete)") {
Err(DbErr::Custom("[after_delete] Cannot be deleted".to_owned())) Err(DbErr::Custom("[after_delete] Cannot be deleted".to_owned()))
} else { } else {
Ok(self) Ok(self)
} }
} else {
Err(DbErr::Custom("[after_delete] Name must be set".to_owned()))
}
} }
} }