servo/components/devtools/id.rs

/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/. */

use base::id::{BrowsingContextId, PipelineId, WebViewId};
use malloc_size_of_derive::MallocSizeOf;
use rustc_hash::FxHashMap;

#[derive(Debug, Default, MallocSizeOf)]
pub(crate) struct IdMap {
    browser_ids: FxHashMap<WebViewId, u32>,
    browsing_context_ids: FxHashMap<BrowsingContextId, u32>,
    outer_window_ids: FxHashMap<PipelineId, u32>,
}

impl IdMap {
    pub(crate) fn browser_id(&mut self, webview_id: WebViewId) -> DevtoolsBrowserId {
        let len = self
            .browser_ids
            .len()
            .checked_add(1)
            .expect("WebViewId count overflow")
            .try_into()
            .expect("DevtoolsBrowserId overflow");
        DevtoolsBrowserId(*self.browser_ids.entry(webview_id).or_insert(len))
    }
    pub(crate) fn browsing_context_id(
        &mut self,
        browsing_context_id: BrowsingContextId,
    ) -> DevtoolsBrowsingContextId {
        let len = self
            .browsing_context_ids
            .len()
            .checked_add(1)
            .expect("BrowsingContextId count overflow")
            .try_into()
            .expect("DevtoolsBrowsingContextId overflow");
        DevtoolsBrowsingContextId(
            *self
                .browsing_context_ids
                .entry(browsing_context_id)
                .or_insert(len),
        )
    }
    pub(crate) fn outer_window_id(&mut self, pipeline_id: PipelineId) -> DevtoolsOuterWindowId {
        let len = self
            .outer_window_ids
            .len()
            .checked_add(1)
            .expect("PipelineId count overflow")
            .try_into()
            .expect("DevtoolsOuterWindowId overflow");
        DevtoolsOuterWindowId(*self.outer_window_ids.entry(pipeline_id).or_insert(len))
    }
}

#[derive(Clone, Copy, Debug, PartialEq, MallocSizeOf)]
pub(crate) struct DevtoolsBrowserId(u32);

#[derive(Clone, Copy, Debug, PartialEq, MallocSizeOf)]
pub(crate) struct DevtoolsBrowsingContextId(u32);

#[derive(Clone, Copy, Debug, PartialEq, MallocSizeOf)]
pub(crate) struct DevtoolsOuterWindowId(u32);

impl DevtoolsBrowserId {
    pub(crate) fn value(&self) -> u32 {
        self.0
    }
}

impl DevtoolsBrowsingContextId {
    pub(crate) fn value(&self) -> u32 {
        self.0
    }
}

impl DevtoolsOuterWindowId {
    pub(crate) fn value(&self) -> u32 {
        self.0
    }
}

#[cfg(test)]
mod test {
    use super::*;

    #[test]
    pub(crate) fn test_id_map() {
        use std::thread;

        use base::id::{PipelineNamespace, PipelineNamespaceId};
        use crossbeam_channel::unbounded;

        macro_rules! test_sequential_id_assignment {
            ($id_type:ident, $new_id_function:expr, $map_id_function:expr) => {
                let (sender, receiver) = unbounded();
                let sender1 = sender.clone();
                let sender2 = sender.clone();
                let sender3 = sender.clone();
                let threads = [
                    thread::spawn(move || {
                        PipelineNamespace::install(PipelineNamespaceId(1));
                        sender1.send($new_id_function()).expect("Send failed");
                        sender1.send($new_id_function()).expect("Send failed");
                        sender1.send($new_id_function()).expect("Send failed");
                    }),
                    thread::spawn(move || {
                        PipelineNamespace::install(PipelineNamespaceId(2));
                        sender2.send($new_id_function()).expect("Send failed");
                        sender2.send($new_id_function()).expect("Send failed");
                        sender2.send($new_id_function()).expect("Send failed");
                    }),
                    thread::spawn(move || {
                        PipelineNamespace::install(PipelineNamespaceId(3));
                        sender3.send($new_id_function()).expect("Send failed");
                        sender3.send($new_id_function()).expect("Send failed");
                        sender3.send($new_id_function()).expect("Send failed");
                    }),
                ];
                for thread in threads {
                    thread.join().expect("Thread join failed");
                }
                let mut id_map = IdMap::default();
                assert_eq!(
                    $map_id_function(&mut id_map, receiver.recv().expect("Recv failed")),
                    $id_type(1)
                );
                assert_eq!(
                    $map_id_function(&mut id_map, receiver.recv().expect("Recv failed")),
                    $id_type(2)
                );
                assert_eq!(
                    $map_id_function(&mut id_map, receiver.recv().expect("Recv failed")),
                    $id_type(3)
                );
                assert_eq!(
                    $map_id_function(&mut id_map, receiver.recv().expect("Recv failed")),
                    $id_type(4)
                );
                assert_eq!(
                    $map_id_function(&mut id_map, receiver.recv().expect("Recv failed")),
                    $id_type(5)
                );
                assert_eq!(
                    $map_id_function(&mut id_map, receiver.recv().expect("Recv failed")),
                    $id_type(6)
                );
                assert_eq!(
                    $map_id_function(&mut id_map, receiver.recv().expect("Recv failed")),
                    $id_type(7)
                );
                assert_eq!(
                    $map_id_function(&mut id_map, receiver.recv().expect("Recv failed")),
                    $id_type(8)
                );
                assert_eq!(
                    $map_id_function(&mut id_map, receiver.recv().expect("Recv failed")),
                    $id_type(9)
                );
            };
        }

        test_sequential_id_assignment!(
            DevtoolsBrowserId,
            || WebViewId::new(base::id::TEST_PAINTER_ID),
            |id_map: &mut IdMap, id| id_map.browser_id(id)
        );
        test_sequential_id_assignment!(
            DevtoolsBrowsingContextId,
            || BrowsingContextId::new(),
            |id_map: &mut IdMap, id| id_map.browsing_context_id(id)
        );
        test_sequential_id_assignment!(
            DevtoolsOuterWindowId,
            || PipelineId::new(),
            |id_map: &mut IdMap, id| id_map.outer_window_id(id)
        );
    }
}