diff --git a/src/query/helper.rs b/src/query/helper.rs index 976220c1..c5511cd6 100644 --- a/src/query/helper.rs +++ b/src/query/helper.rs @@ -3,8 +3,8 @@ use crate::{ PrimaryKeyToColumn, RelationDef, }; use sea_query::{ - Alias, ConditionType, Expr, Iden, IntoCondition, IntoIden, LockType, SeaRc, SelectExpr, - SelectStatement, SimpleExpr, TableRef, + Alias, ConditionType, Expr, Iden, IntoCondition, IntoIden, LockBehavior, LockType, SeaRc, + SelectExpr, SelectStatement, SimpleExpr, TableRef, }; pub use sea_query::{Condition, ConditionalStatement, DynIden, JoinType, Order, OrderedStatement}; @@ -447,6 +447,14 @@ pub trait QuerySelect: Sized { self } + /// Row locking with behavior (if supported). + /// + /// See [`SelectStatement::lock_with_behavior`]. + fn lock_with_behavior(mut self, r#type: LockType, behavior: LockBehavior) -> Self { + self.query().lock_with_behavior(r#type, behavior); + self + } + /// Add an expression to the select expression list. /// ``` /// use sea_orm::sea_query::Expr;