caisin/sea-orm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Handle "None of the database rows are affected" for Postgres

Browse Source
This commit is contained in:
Billy Chan 2021-11-08 15:12:47 +08:00
parent 50605c731b
commit 623873678b
No known key found for this signature in database
GPG Key ID: A2D690CAC7DF3CC7
1 changed files with 22 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
src/executor/update.rs
View File

@ -91,33 +91,43 @@ where
C: ConnectionTrait<'a>, C: ConnectionTrait<'a>,
{ {
let db_backend = db.get_database_backend(); let db_backend = db.get_database_backend();
let found = match db_backend { match db_backend {
DbBackend::Postgres => { DbBackend::Postgres => {
query.returning(Returning::Columns( query.returning(Returning::Columns(
<A::Entity as EntityTrait>::Column::iter() <A::Entity as EntityTrait>::Column::iter()
.map(|c| c.into_column_ref()) .map(|c| c.into_column_ref())
.collect(), .collect(),
)); ));
SelectorRaw::<SelectModel<<A::Entity as EntityTrait>::Model>>::from_statement( let found: Option<<A::Entity as EntityTrait>::Model> =
db_backend.build(&query), SelectorRaw::<SelectModel<<A::Entity as EntityTrait>::Model>>::from_statement(
) db_backend.build(&query),
.one(db) )
.await? .one(db)
.await?;
// If we got `None` then we are updating a row that does not exist.
match found {
Some(model) => Ok(model.into_active_model()),
None => Err(DbErr::RecordNotFound(
"None of the database rows are affected".to_owned(),
)),
}
} }
_ => { _ => {
// If we updating a row that does not exist, error will be thrown here.
Updater::new(query).check_record_exists().exec(db).await?; Updater::new(query).check_record_exists().exec(db).await?;
let primary_key_value = match model.get_primary_key_value() { let primary_key_value = match model.get_primary_key_value() {
Some(val) => FromValueTuple::from_value_tuple(val), Some(val) => FromValueTuple::from_value_tuple(val),
None => return Err(DbErr::Exec("Fail to get primary key from model".to_owned())), None => return Err(DbErr::Exec("Fail to get primary key from model".to_owned())),
}; };
<A::Entity as EntityTrait>::find_by_id(primary_key_value) let found = <A::Entity as EntityTrait>::find_by_id(primary_key_value)
.one(db) .one(db)
.await? .await?;
// If we cannot select the updated row from db by the cached primary key
match found {
Some(model) => Ok(model.into_active_model()),
None => Err(DbErr::Exec("Failed to find inserted item".to_owned())),
}
} }
};
match found {
Some(model) => Ok(model.into_active_model()),
None => Err(DbErr::Exec("Failed to find inserted item".to_owned())),
} }
} }