rust/hg-core/src/dirstate/dirs_multiset.rs

42607:8f7c3f43e3ac

//!
//! Used to counts the references to directories in a manifest or dirstate.
use std::collections::hash_map::{Entry, Iter};
use std::collections::HashMap;
use {DirsIterable, DirstateEntry, DirstateMapError};

#[derive(PartialEq, Debug)]
pub struct DirsMultiset {
    inner: HashMap<Vec<u8>, u32>,
}

        }

        multiset
    }

    /// Returns the slice up to the next directory name from right to left,
    /// without trailing slash
    fn find_dir(path: &[u8]) -> &[u8] {
        let mut path = path;
        loop {
            if let Some(new_pos) = path.len().checked_sub(1) {
                if path[new_pos] == b'/' {
                    break &path[..new_pos];
                }
                path = &path[..new_pos];
            } else {
                break &[];
            }
        }
    }

    /// Increases the count of deepest directory contained in the path.
    ///
    /// If the directory is not yet in the map, adds its parents.
    pub fn add_path(&mut self, path: &[u8]) {
        let mut pos = path.len();

        loop {
            let subpath = Self::find_dir(&path[..pos]);
            if let Some(val) = self.inner.get_mut(subpath) {
                *val += 1;
                break;
            }
            self.inner.insert(subpath.to_owned(), 1);

            pos = subpath.len();
            if pos == 0 {
                break;
            }
        }
    }

    /// Decreases the count of deepest directory contained in the path.
    ///

    /// If the directory is not in the map, something horrible has happened.
    pub fn delete_path(
        &mut self,
        path: &[u8],
    ) -> Result<(), DirstateMapError> {
        let mut pos = path.len();

        loop {
            let subpath = Self::find_dir(&path[..pos]);
            match self.inner.entry(subpath.to_owned()) {
                Entry::Occupied(mut entry) => {
                    let val = entry.get().clone();
                    if val > 1 {
                        entry.insert(val - 1);

                    return Err(DirstateMapError::PathNotFound(
                        path.to_owned(),
                    ))
                }
            };

            pos = subpath.len();
            if pos == 0 {
                break;
            }
        }

        Ok(())
    }

42608:717686c5e461

//!
//! Used to counts the references to directories in a manifest or dirstate.
use std::collections::hash_map::{Entry, Iter};
use std::collections::HashMap;
use {DirsIterable, DirstateEntry, DirstateMapError};
use utils::files;

#[derive(PartialEq, Debug)]
pub struct DirsMultiset {
    inner: HashMap<Vec<u8>, u32>,
}

        }

        multiset
    }

    /// Increases the count of deepest directory contained in the path.
    ///
    /// If the directory is not yet in the map, adds its parents.
    pub fn add_path(&mut self, path: &[u8]) {
        for subpath in files::find_dirs(path) {
            if let Some(val) = self.inner.get_mut(subpath) {
                *val += 1;
                break;
            }
            self.inner.insert(subpath.to_owned(), 1);
        }
    }

    /// Decreases the count of deepest directory contained in the path.
    ///

    /// If the directory is not in the map, something horrible has happened.
    pub fn delete_path(
        &mut self,
        path: &[u8],
    ) -> Result<(), DirstateMapError> {
        for subpath in files::find_dirs(path) {
            match self.inner.entry(subpath.to_owned()) {
                Entry::Occupied(mut entry) => {
                    let val = entry.get().clone();
                    if val > 1 {
                        entry.insert(val - 1);

                    return Err(DirstateMapError::PathNotFound(
                        path.to_owned(),
                    ))
                }
            };
        }

        Ok(())
    }