From 7eb3e12294976585683ce12f86d074751d1fbe45 Mon Sep 17 00:00:00 2001
From: Francis Le Roy <36069799+GrandChaman@users.noreply.github.com>
Date: Sun, 14 Jan 2024 16:51:51 +0100
Subject: [PATCH] Updated the 'QuerySelect' trait to expose
 'lock_with_behavior' (#1867)

---
 src/query/helper.rs | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

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;