feat: add count and types to merge

launchbadge · Dec 22, 2024 · 31c6bf4 · 31c6bf4
1 parent 1c461ed
commit 31c6bf4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/sqlx-core/src/arguments.rs b/sqlx-core/src/arguments.rs
@@ -28,7 +28,7 @@ pub trait Arguments<'q>: Send + Sized + Default {
         writer.write_str("?")
     }
 
-    fn merge(&mut self, other: Self);
+    fn merge(mut self, other: Self);
 }
 
 pub trait IntoArguments<'q, DB: Database>: Sized + Send {

diff --git a/sqlx-postgres/src/arguments.rs b/sqlx-postgres/src/arguments.rs
@@ -148,6 +148,9 @@ impl<'q> Arguments<'q> for PgArguments {
     fn merge(&mut self, other: Self) {
         self.types.extend(other.types);
         self.buffer.extend_from_slice(&other.buffer);
+        self.buffer.count += other.buffer.count;
+        self.buffer.patches.extend(other.buffer.patches);
+        self.buffer.type_holes.extend(other.buffer.type_holes);
     }
 
     #[inline(always)]