diff options
| author | Matt Brubeck <mbrubeck@limpet.net> | 2018-01-17 08:37:03 -0800 |
|---|---|---|
| committer | Matt Brubeck <mbrubeck@limpet.net> | 2018-01-17 08:37:03 -0800 |
| commit | e9fdee8818f5c089013b9c8604e183b1cbd816ad (patch) | |
| tree | 1557cc7555761a5cad5110a0cea26d4f315d2b69 /src/libstd | |
| parent | 44912bf77bf1fd2137e5b7d1d31b2e59e2819136 (diff) | |
| download | rust-e9fdee8818f5c089013b9c8604e183b1cbd816ad.tar.gz rust-e9fdee8818f5c089013b9c8604e183b1cbd816ad.zip | |
Use File::metadata instead of fs::metadata to choose buffer size
This replaces a `stat` syscall with `fstat` or similar, which can be faster. Fixes #47519.
Diffstat (limited to 'src/libstd')
| -rw-r--r-- | src/libstd/fs.rs | 16 |
1 files changed, 9 insertions, 7 deletions
diff --git a/src/libstd/fs.rs b/src/libstd/fs.rs index 51cb9609120..1325ca464c4 100644 --- a/src/libstd/fs.rs +++ b/src/libstd/fs.rs @@ -211,12 +211,12 @@ pub struct DirBuilder { recursive: bool, } -/// How large a buffer to pre-allocate before reading the entire file at `path`. -fn initial_buffer_size<P: AsRef<Path>>(path: P) -> usize { +/// How large a buffer to pre-allocate before reading the entire file. +fn initial_buffer_size(file: &File) -> usize { // Allocate one extra byte so the buffer doesn't need to grow before the // final `read` call at the end of the file. Don't worry about `usize` // overflow because reading will fail regardless in that case. - metadata(path).map(|m| m.len() as usize + 1).unwrap_or(0) + file.metadata().map(|m| m.len() as usize + 1).unwrap_or(0) } /// Read the entire contents of a file into a bytes vector. @@ -254,8 +254,9 @@ fn initial_buffer_size<P: AsRef<Path>>(path: P) -> usize { /// ``` #[unstable(feature = "fs_read_write", issue = "46588")] pub fn read<P: AsRef<Path>>(path: P) -> io::Result<Vec<u8>> { - let mut bytes = Vec::with_capacity(initial_buffer_size(&path)); - File::open(path)?.read_to_end(&mut bytes)?; + let mut file = File::open(path)?; + let mut bytes = Vec::with_capacity(initial_buffer_size(&file)); + file.read_to_end(&mut bytes)?; Ok(bytes) } @@ -295,8 +296,9 @@ pub fn read<P: AsRef<Path>>(path: P) -> io::Result<Vec<u8>> { /// ``` #[unstable(feature = "fs_read_write", issue = "46588")] pub fn read_string<P: AsRef<Path>>(path: P) -> io::Result<String> { - let mut string = String::with_capacity(initial_buffer_size(&path)); - File::open(path)?.read_to_string(&mut string)?; + let mut file = File::open(path)?; + let mut string = String::with_capacity(initial_buffer_size(&file)); + file.read_to_string(&mut string)?; Ok(string) } |
