bjoernager/rust
bjoernager/rust
1
Fork 0
Code Activity

define MmapMut and use it in Decodable impl

Browse source
This commit is contained in:
Yoshiki Matsuda 2022-05-08 12:36:12 +09:00
parent 47c36893a1
commit c57d778872
2 changed files with 74 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

25
compiler/rustc_metadata/src/rmeta/encoder.rs
View file

@ -4,7 +4,7 @@ use crate::rmeta::*;
use rustc_data_structures::fingerprint::Fingerprint;
use rustc_data_structures::fx::{FxHashMap, FxIndexSet};
use rustc_data_structures::memmap::Mmap;
use rustc_data_structures::memmap::{Mmap, MmapMut};
use rustc_data_structures::stable_hasher::{HashStable, StableHasher};
use rustc_data_structures::sync::{join, par_iter, Lrc, ParallelIterator};
use rustc_data_structures::temp_dir::MaybeTempDir;
@ -44,7 +44,6 @@ use std::io::{Read, Seek, Write};
use std::iter;
use std::num::NonZeroUsize;
use std::path::{Path, PathBuf};
use tempfile::Builder as TempFileBuilder;
use tracing::{debug, trace};
pub(super) struct EncodeContext<'a, 'tcx> {
@ -2179,19 +2178,19 @@ impl<S: Encoder> Encodable<S> for EncodedMetadata {
impl<D: Decoder> Decodable<D> for EncodedMetadata {
fn decode(d: &mut D) -> Self {
let temp_dir = TempFileBuilder::new().prefix("decoded").tempdir().unwrap();
let temp_dir = MaybeTempDir::new(temp_dir, false);
let filename = temp_dir.as_ref().join("decoded");
let file = std::fs::File::create(&filename).unwrap();
let mut file = std::io::BufWriter::new(file);
let len = d.read_usize();
for _ in 0..len {
file.write(&[d.read_u8()]).unwrap();
}
file.flush().unwrap();
let mmap = if len > 0 {
let mut mmap = MmapMut::map_anon(len).unwrap();
for _ in 0..len {
(&mut mmap[..]).write(&[d.read_u8()]).unwrap();
}
mmap.flush().unwrap();
Some(mmap.make_read_only().unwrap())
} else {
None
};
Self::from_path(filename, Some(temp_dir)).unwrap()
Self { mmap, _temp_dir: None }
}
}