Skip to content

Commit fa483a4

Browse files
Rollup merge of #120900 - marcospb19:std-use-seek-stream-position, r=joshtriplett
std: use `stream_position` where applicable by replacing `seek(SeekFrom::Current(0))` calls
2 parents 29b1207 + af6840b commit fa483a4

File tree

3 files changed

+31
-29
lines changed

3 files changed

+31
-29
lines changed

library/std/src/fs.rs

+2-3
Original file line numberDiff line numberDiff line change
@@ -993,7 +993,7 @@ impl OpenOptions {
993993
/// If a file is opened with both read and append access, beware that after
994994
/// opening, and after every write, the position for reading may be set at the
995995
/// end of the file. So, before writing, save the current position (using
996-
/// <code>[seek]\([SeekFrom]::[Current]\(0))</code>), and restore it before the next read.
996+
/// <code>[Seek]::[stream_position]</code>), and restore it before the next read.
997997
///
998998
/// ## Note
999999
///
@@ -1002,8 +1002,7 @@ impl OpenOptions {
10021002
///
10031003
/// [`write()`]: Write::write "io::Write::write"
10041004
/// [`flush()`]: Write::flush "io::Write::flush"
1005-
/// [seek]: Seek::seek "io::Seek::seek"
1006-
/// [Current]: SeekFrom::Current "io::SeekFrom::Current"
1005+
/// [stream_position]: Seek::stream_position "io::Seek::stream_position"
10071006
///
10081007
/// # Examples
10091008
///

library/std/src/fs/tests.rs

+24-24
Original file line numberDiff line numberDiff line change
@@ -187,9 +187,9 @@ fn file_test_io_seek_and_tell_smoke_test() {
187187
{
188188
let mut read_stream = check!(File::open(filename));
189189
check!(read_stream.seek(SeekFrom::Start(set_cursor)));
190-
tell_pos_pre_read = check!(read_stream.seek(SeekFrom::Current(0)));
190+
tell_pos_pre_read = check!(read_stream.stream_position());
191191
check!(read_stream.read(&mut read_mem));
192-
tell_pos_post_read = check!(read_stream.seek(SeekFrom::Current(0)));
192+
tell_pos_post_read = check!(read_stream.stream_position());
193193
}
194194
check!(fs::remove_file(filename));
195195
let read_str = str::from_utf8(&read_mem).unwrap();
@@ -284,42 +284,42 @@ fn file_test_io_read_write_at() {
284284
let oo = OpenOptions::new().create_new(true).write(true).read(true).clone();
285285
let mut rw = check!(oo.open(&filename));
286286
assert_eq!(check!(rw.write_at(write1.as_bytes(), 5)), write1.len());
287-
assert_eq!(check!(rw.seek(SeekFrom::Current(0))), 0);
287+
assert_eq!(check!(rw.stream_position()), 0);
288288
assert_eq!(check!(rw.read_at(&mut buf, 5)), write1.len());
289289
assert_eq!(str::from_utf8(&buf[..write1.len()]), Ok(write1));
290-
assert_eq!(check!(rw.seek(SeekFrom::Current(0))), 0);
290+
assert_eq!(check!(rw.stream_position()), 0);
291291
assert_eq!(check!(rw.read_at(&mut buf[..write2.len()], 0)), write2.len());
292292
assert_eq!(str::from_utf8(&buf[..write2.len()]), Ok("\0\0\0\0\0"));
293-
assert_eq!(check!(rw.seek(SeekFrom::Current(0))), 0);
293+
assert_eq!(check!(rw.stream_position()), 0);
294294
assert_eq!(check!(rw.write(write2.as_bytes())), write2.len());
295-
assert_eq!(check!(rw.seek(SeekFrom::Current(0))), 5);
295+
assert_eq!(check!(rw.stream_position()), 5);
296296
assert_eq!(check!(rw.read(&mut buf)), write1.len());
297297
assert_eq!(str::from_utf8(&buf[..write1.len()]), Ok(write1));
298-
assert_eq!(check!(rw.seek(SeekFrom::Current(0))), 9);
298+
assert_eq!(check!(rw.stream_position()), 9);
299299
assert_eq!(check!(rw.read_at(&mut buf[..write2.len()], 0)), write2.len());
300300
assert_eq!(str::from_utf8(&buf[..write2.len()]), Ok(write2));
301-
assert_eq!(check!(rw.seek(SeekFrom::Current(0))), 9);
301+
assert_eq!(check!(rw.stream_position()), 9);
302302
assert_eq!(check!(rw.write_at(write3.as_bytes(), 9)), write3.len());
303-
assert_eq!(check!(rw.seek(SeekFrom::Current(0))), 9);
303+
assert_eq!(check!(rw.stream_position()), 9);
304304
}
305305
{
306306
let mut read = check!(File::open(&filename));
307307
assert_eq!(check!(read.read_at(&mut buf, 0)), content.len());
308308
assert_eq!(str::from_utf8(&buf[..content.len()]), Ok(content));
309-
assert_eq!(check!(read.seek(SeekFrom::Current(0))), 0);
309+
assert_eq!(check!(read.stream_position()), 0);
310310
assert_eq!(check!(read.seek(SeekFrom::End(-5))), 9);
311311
assert_eq!(check!(read.read_at(&mut buf, 0)), content.len());
312312
assert_eq!(str::from_utf8(&buf[..content.len()]), Ok(content));
313-
assert_eq!(check!(read.seek(SeekFrom::Current(0))), 9);
313+
assert_eq!(check!(read.stream_position()), 9);
314314
assert_eq!(check!(read.read(&mut buf)), write3.len());
315315
assert_eq!(str::from_utf8(&buf[..write3.len()]), Ok(write3));
316-
assert_eq!(check!(read.seek(SeekFrom::Current(0))), 14);
316+
assert_eq!(check!(read.stream_position()), 14);
317317
assert_eq!(check!(read.read_at(&mut buf, 0)), content.len());
318318
assert_eq!(str::from_utf8(&buf[..content.len()]), Ok(content));
319-
assert_eq!(check!(read.seek(SeekFrom::Current(0))), 14);
319+
assert_eq!(check!(read.stream_position()), 14);
320320
assert_eq!(check!(read.read_at(&mut buf, 14)), 0);
321321
assert_eq!(check!(read.read_at(&mut buf, 15)), 0);
322-
assert_eq!(check!(read.seek(SeekFrom::Current(0))), 14);
322+
assert_eq!(check!(read.stream_position()), 14);
323323
}
324324
check!(fs::remove_file(&filename));
325325
}
@@ -362,38 +362,38 @@ fn file_test_io_seek_read_write() {
362362
let oo = OpenOptions::new().create_new(true).write(true).read(true).clone();
363363
let mut rw = check!(oo.open(&filename));
364364
assert_eq!(check!(rw.seek_write(write1.as_bytes(), 5)), write1.len());
365-
assert_eq!(check!(rw.seek(SeekFrom::Current(0))), 9);
365+
assert_eq!(check!(rw.stream_position()), 9);
366366
assert_eq!(check!(rw.seek_read(&mut buf, 5)), write1.len());
367367
assert_eq!(str::from_utf8(&buf[..write1.len()]), Ok(write1));
368-
assert_eq!(check!(rw.seek(SeekFrom::Current(0))), 9);
368+
assert_eq!(check!(rw.stream_position()), 9);
369369
assert_eq!(check!(rw.seek(SeekFrom::Start(0))), 0);
370370
assert_eq!(check!(rw.write(write2.as_bytes())), write2.len());
371-
assert_eq!(check!(rw.seek(SeekFrom::Current(0))), 5);
371+
assert_eq!(check!(rw.stream_position()), 5);
372372
assert_eq!(check!(rw.read(&mut buf)), write1.len());
373373
assert_eq!(str::from_utf8(&buf[..write1.len()]), Ok(write1));
374-
assert_eq!(check!(rw.seek(SeekFrom::Current(0))), 9);
374+
assert_eq!(check!(rw.stream_position()), 9);
375375
assert_eq!(check!(rw.seek_read(&mut buf[..write2.len()], 0)), write2.len());
376376
assert_eq!(str::from_utf8(&buf[..write2.len()]), Ok(write2));
377-
assert_eq!(check!(rw.seek(SeekFrom::Current(0))), 5);
377+
assert_eq!(check!(rw.stream_position()), 5);
378378
assert_eq!(check!(rw.seek_write(write3.as_bytes(), 9)), write3.len());
379-
assert_eq!(check!(rw.seek(SeekFrom::Current(0))), 14);
379+
assert_eq!(check!(rw.stream_position()), 14);
380380
}
381381
{
382382
let mut read = check!(File::open(&filename));
383383
assert_eq!(check!(read.seek_read(&mut buf, 0)), content.len());
384384
assert_eq!(str::from_utf8(&buf[..content.len()]), Ok(content));
385-
assert_eq!(check!(read.seek(SeekFrom::Current(0))), 14);
385+
assert_eq!(check!(read.stream_position()), 14);
386386
assert_eq!(check!(read.seek(SeekFrom::End(-5))), 9);
387387
assert_eq!(check!(read.seek_read(&mut buf, 0)), content.len());
388388
assert_eq!(str::from_utf8(&buf[..content.len()]), Ok(content));
389-
assert_eq!(check!(read.seek(SeekFrom::Current(0))), 14);
389+
assert_eq!(check!(read.stream_position()), 14);
390390
assert_eq!(check!(read.seek(SeekFrom::End(-5))), 9);
391391
assert_eq!(check!(read.read(&mut buf)), write3.len());
392392
assert_eq!(str::from_utf8(&buf[..write3.len()]), Ok(write3));
393-
assert_eq!(check!(read.seek(SeekFrom::Current(0))), 14);
393+
assert_eq!(check!(read.stream_position()), 14);
394394
assert_eq!(check!(read.seek_read(&mut buf, 0)), content.len());
395395
assert_eq!(str::from_utf8(&buf[..content.len()]), Ok(content));
396-
assert_eq!(check!(read.seek(SeekFrom::Current(0))), 14);
396+
assert_eq!(check!(read.stream_position()), 14);
397397
assert_eq!(check!(read.seek_read(&mut buf, 14)), 0);
398398
assert_eq!(check!(read.seek_read(&mut buf, 15)), 0);
399399
}

library/std/src/io/buffered/tests.rs

+5-2
Original file line numberDiff line numberDiff line change
@@ -114,7 +114,7 @@ fn test_buffered_reader_seek() {
114114

115115
assert_eq!(reader.seek(SeekFrom::Start(3)).ok(), Some(3));
116116
assert_eq!(reader.fill_buf().ok(), Some(&[0, 1][..]));
117-
assert_eq!(reader.seek(SeekFrom::Current(0)).ok(), Some(3));
117+
assert_eq!(reader.stream_position().ok(), Some(3));
118118
assert_eq!(reader.fill_buf().ok(), Some(&[0, 1][..]));
119119
assert_eq!(reader.seek(SeekFrom::Current(1)).ok(), Some(4));
120120
assert_eq!(reader.fill_buf().ok(), Some(&[1, 2][..]));
@@ -230,6 +230,9 @@ fn test_buffered_reader_seek_underflow() {
230230
Ok(len)
231231
}
232232
}
233+
// note: this implementation of `Seek` is "broken" due to position
234+
// wrapping, so calling `reader.seek(Current(0))` is semantically different
235+
// than `reader.stream_position()`
233236
impl Seek for PositionReader {
234237
fn seek(&mut self, pos: SeekFrom) -> io::Result<u64> {
235238
match pos {
@@ -374,7 +377,7 @@ fn test_buffered_writer_seek() {
374377
let mut w = BufWriter::with_capacity(3, io::Cursor::new(Vec::new()));
375378
w.write_all(&[0, 1, 2, 3, 4, 5]).unwrap();
376379
w.write_all(&[6, 7]).unwrap();
377-
assert_eq!(w.seek(SeekFrom::Current(0)).ok(), Some(8));
380+
assert_eq!(w.stream_position().ok(), Some(8));
378381
assert_eq!(&w.get_ref().get_ref()[..], &[0, 1, 2, 3, 4, 5, 6, 7][..]);
379382
assert_eq!(w.seek(SeekFrom::Start(2)).ok(), Some(2));
380383
w.write_all(&[8, 9]).unwrap();

0 commit comments

Comments
 (0)