Skip to content

Commit 94379d8

Browse files
authored
fix(postgres): fix buffer management in PgCopyIn.read_from (#2630)
1 parent f2bb464 commit 94379d8

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

sqlx-core/src/net/socket/buffered.rs

+1-1
Original file line numberDiff line numberDiff line change
@@ -151,7 +151,7 @@ impl WriteBuffer {
151151
self.buf.truncate(self.bytes_written);
152152
self.buf.extend_from_slice(slice);
153153
}
154-
154+
self.advance(slice.len());
155155
self.sanity_check();
156156
}
157157

sqlx-postgres/src/copy.rs

+1-1
Original file line numberDiff line numberDiff line change
@@ -229,7 +229,7 @@ impl<C: DerefMut<Target = PgConnection>> PgCopyIn<C> {
229229
let buf = conn.stream.write_buffer_mut();
230230

231231
// CopyData format code and reserved space for length
232-
buf.put_slice(b"d\0\0\0\0");
232+
buf.put_slice(b"d\0\0\0\x04");
233233

234234
let read = match () {
235235
// Tokio lets us read into the buffer without zeroing first

0 commit comments

Comments
 (0)