Mercurial > hg
view rust/hg-core/src/operations/debugdata.rs @ 46431:645ee7225fab
rust: Make NodePrefix allocation-free and Copy, remove NodePrefixRef
The `*Ref` struct only existed to avoid allocating `Vec`s
when cloning `NodePrefix`, but we can avoid having `Vec`
in the first place by using an inline array instead.
This makes `NodePrefix` 21 bytes (with 1 for the length)
which is smaller than before as `Vec` alone is 24 bytes.
Differential Revision: https://phab.mercurial-scm.org/D9863
author | Simon Sapin <simon.sapin@octobus.net> |
---|---|
date | Mon, 25 Jan 2021 11:48:47 +0100 |
parents | 8a4914397d02 |
children | 4b381dbbf8b7 |
line wrap: on
line source
// debugdata.rs // // Copyright 2020 Antoine Cezar <antoine.cezar@octobus.net> // // This software may be used and distributed according to the terms of the // GNU General Public License version 2 or any later version. use crate::repo::Repo; use crate::revlog::revlog::{Revlog, RevlogError}; use crate::revlog::NodePrefix; use crate::revlog::Revision; /// Kind of data to debug #[derive(Debug, Copy, Clone)] pub enum DebugDataKind { Changelog, Manifest, } /// Kind of error encountered by DebugData #[derive(Debug)] pub enum DebugDataErrorKind { /// Error when reading a `revlog` file. IoError(std::io::Error), /// The revision has not been found. InvalidRevision, /// Found more than one revision whose ID match the requested prefix AmbiguousPrefix, /// A `revlog` file is corrupted. CorruptedRevlog, /// The `revlog` format version is not supported. UnsuportedRevlogVersion(u16), /// The `revlog` data format is not supported. UnknowRevlogDataFormat(u8), } /// A DebugData error #[derive(Debug)] pub struct DebugDataError { /// Kind of error encountered by DebugData pub kind: DebugDataErrorKind, } impl From<DebugDataErrorKind> for DebugDataError { fn from(kind: DebugDataErrorKind) -> Self { DebugDataError { kind } } } impl From<std::io::Error> for DebugDataError { fn from(err: std::io::Error) -> Self { let kind = DebugDataErrorKind::IoError(err); DebugDataError { kind } } } impl From<RevlogError> for DebugDataError { fn from(err: RevlogError) -> Self { match err { RevlogError::IoError(err) => DebugDataErrorKind::IoError(err), RevlogError::UnsuportedVersion(version) => { DebugDataErrorKind::UnsuportedRevlogVersion(version) } RevlogError::InvalidRevision => { DebugDataErrorKind::InvalidRevision } RevlogError::AmbiguousPrefix => { DebugDataErrorKind::AmbiguousPrefix } RevlogError::Corrupted => DebugDataErrorKind::CorruptedRevlog, RevlogError::UnknowDataFormat(format) => { DebugDataErrorKind::UnknowRevlogDataFormat(format) } } .into() } } /// Dump the contents data of a revision. pub fn debug_data( repo: &Repo, rev: &str, kind: DebugDataKind, ) -> Result<Vec<u8>, DebugDataError> { let index_file = match kind { DebugDataKind::Changelog => "00changelog.i", DebugDataKind::Manifest => "00manifest.i", }; let revlog = Revlog::open(repo, index_file, None)?; let data = match rev.parse::<Revision>() { Ok(rev) => revlog.get_rev_data(rev)?, _ => { let node = NodePrefix::from_hex(&rev) .map_err(|_| DebugDataErrorKind::InvalidRevision)?; let rev = revlog.get_node_rev(node)?; revlog.get_rev_data(rev)? } }; Ok(data) }