actions/cargo-binstall
1
0
Fork 0
mirror of https://github.com/cargo-bins/cargo-binstall.git synced 2025-04-21 13:08:42 +00:00
Code Issues Projects Releases Packages Wiki Activity

Fix Records::overwrite: Update file len

Browse source 
Signed-off-by: Jiahao XU <Jiahao_XU@outlook.com>
This commit is contained in:
Jiahao XU 2022-07-27 23:13:41 +10:00
parent c9139fec68
commit d796424199
No known key found for this signature in database
GPG key ID: 591C0B03040416D6
1 changed files with 15 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
src/metafiles/binstall_v1.rs
View file

@ -2,7 +2,7 @@ use std::{
borrow, cmp, borrow, cmp,
collections::{btree_set, BTreeSet}, collections::{btree_set, BTreeSet},
fs, hash, fs, hash,
io::{self, Write}, io::{self, Seek, Write},
iter::{IntoIterator, Iterator}, iter::{IntoIterator, Iterator},
path::{Path, PathBuf}, path::{Path, PathBuf},
}; };
@ -95,17 +95,20 @@ pub fn append_to_path<Iter>(path: impl AsRef<Path>, iter: Iter) -> Result<(), Er
where where
Iter: IntoIterator<Item = MetaData>, Iter: IntoIterator<Item = MetaData>,
{ {
let file = FileLock::new_exclusive( let mut file = FileLock::new_exclusive(
fs::OpenOptions::new() fs::OpenOptions::new()
.create(true) .create(true)
.append(true) .append(true)
.open(path)?, .open(path)?,
)?; )?;
write_to(file, &mut iter.into_iter()) write_to(&mut file, &mut iter.into_iter())
} }
pub fn write_to(file: FileLock, iter: &mut dyn Iterator<Item = MetaData>) -> Result<(), Error> { pub fn write_to(
file: &mut FileLock,
iter: &mut dyn Iterator<Item = MetaData>,
) -> Result<(), Error> {
let writer = io::BufWriter::with_capacity(512, file); let writer = io::BufWriter::with_capacity(512, file);
let mut ser = serde_json::Serializer::new(writer); let mut ser = serde_json::Serializer::new(writer);
@ -158,8 +161,14 @@ impl Records {
} }
/// **Warning: This will overwrite all existing records!** /// **Warning: This will overwrite all existing records!**
pub fn overwrite(self) -> Result<(), Error> { pub fn overwrite(mut self) -> Result<(), Error> {
write_to(self.file, &mut self.data.into_iter()) self.file.rewind()?;
write_to(&mut self.file, &mut self.data.into_iter())?;
let len = self.file.stream_position()?;
self.file.set_len(len)?;
Ok(())
} }
pub fn get(&self, value: impl AsRef<str>) -> Option<&MetaData> { pub fn get(&self, value: impl AsRef<str>) -> Option<&MetaData> {