devtools/
id.rs

1/* This Source Code Form is subject to the terms of the Mozilla Public
2 * License, v. 2.0. If a copy of the MPL was not distributed with this
3 * file, You can obtain one at https://mozilla.org/MPL/2.0/. */
4
5use base::id::{BrowsingContextId, PipelineId, WebViewId};
6use rustc_hash::FxHashMap;
7
8#[derive(Debug, Default)]
9pub(crate) struct IdMap {
10    pub(crate) browser_ids: FxHashMap<WebViewId, u32>,
11    pub(crate) browsing_context_ids: FxHashMap<BrowsingContextId, u32>,
12    pub(crate) outer_window_ids: FxHashMap<PipelineId, u32>,
13}
14
15impl IdMap {
16    pub(crate) fn browser_id(&mut self, webview_id: WebViewId) -> DevtoolsBrowserId {
17        let len = self
18            .browser_ids
19            .len()
20            .checked_add(1)
21            .expect("WebViewId count overflow")
22            .try_into()
23            .expect("DevtoolsBrowserId overflow");
24        DevtoolsBrowserId(*self.browser_ids.entry(webview_id).or_insert(len))
25    }
26    pub(crate) fn browsing_context_id(
27        &mut self,
28        browsing_context_id: BrowsingContextId,
29    ) -> DevtoolsBrowsingContextId {
30        let len = self
31            .browsing_context_ids
32            .len()
33            .checked_add(1)
34            .expect("BrowsingContextId count overflow")
35            .try_into()
36            .expect("DevtoolsBrowsingContextId overflow");
37        DevtoolsBrowsingContextId(
38            *self
39                .browsing_context_ids
40                .entry(browsing_context_id)
41                .or_insert(len),
42        )
43    }
44    pub(crate) fn outer_window_id(&mut self, pipeline_id: PipelineId) -> DevtoolsOuterWindowId {
45        let len = self
46            .outer_window_ids
47            .len()
48            .checked_add(1)
49            .expect("PipelineId count overflow")
50            .try_into()
51            .expect("DevtoolsOuterWindowId overflow");
52        DevtoolsOuterWindowId(*self.outer_window_ids.entry(pipeline_id).or_insert(len))
53    }
54}
55
56#[derive(Clone, Copy, Debug, PartialEq)]
57pub(crate) struct DevtoolsBrowserId(u32);
58
59#[derive(Clone, Copy, Debug, PartialEq)]
60pub(crate) struct DevtoolsBrowsingContextId(u32);
61
62#[derive(Clone, Copy, Debug, PartialEq)]
63pub(crate) struct DevtoolsOuterWindowId(u32);
64
65impl DevtoolsBrowserId {
66    pub(crate) fn value(&self) -> u32 {
67        self.0
68    }
69}
70
71impl DevtoolsBrowsingContextId {
72    pub(crate) fn value(&self) -> u32 {
73        self.0
74    }
75}
76
77impl DevtoolsOuterWindowId {
78    pub(crate) fn value(&self) -> u32 {
79        self.0
80    }
81}
82
83#[test]
84pub(crate) fn test_id_map() {
85    use std::thread;
86
87    use base::id::{PipelineNamespace, PipelineNamespaceId};
88    use crossbeam_channel::unbounded;
89
90    macro_rules! test_sequential_id_assignment {
91        ($id_type:ident, $new_id_function:expr, $map_id_function:expr) => {
92            let (sender, receiver) = unbounded();
93            let sender1 = sender.clone();
94            let sender2 = sender.clone();
95            let sender3 = sender.clone();
96            let threads = [
97                thread::spawn(move || {
98                    PipelineNamespace::install(PipelineNamespaceId(1));
99                    sender1.send($new_id_function()).expect("Send failed");
100                    sender1.send($new_id_function()).expect("Send failed");
101                    sender1.send($new_id_function()).expect("Send failed");
102                }),
103                thread::spawn(move || {
104                    PipelineNamespace::install(PipelineNamespaceId(2));
105                    sender2.send($new_id_function()).expect("Send failed");
106                    sender2.send($new_id_function()).expect("Send failed");
107                    sender2.send($new_id_function()).expect("Send failed");
108                }),
109                thread::spawn(move || {
110                    PipelineNamespace::install(PipelineNamespaceId(3));
111                    sender3.send($new_id_function()).expect("Send failed");
112                    sender3.send($new_id_function()).expect("Send failed");
113                    sender3.send($new_id_function()).expect("Send failed");
114                }),
115            ];
116            for thread in threads {
117                thread.join().expect("Thread join failed");
118            }
119            let mut id_map = IdMap::default();
120            assert_eq!(
121                $map_id_function(&mut id_map, receiver.recv().expect("Recv failed")),
122                $id_type(1)
123            );
124            assert_eq!(
125                $map_id_function(&mut id_map, receiver.recv().expect("Recv failed")),
126                $id_type(2)
127            );
128            assert_eq!(
129                $map_id_function(&mut id_map, receiver.recv().expect("Recv failed")),
130                $id_type(3)
131            );
132            assert_eq!(
133                $map_id_function(&mut id_map, receiver.recv().expect("Recv failed")),
134                $id_type(4)
135            );
136            assert_eq!(
137                $map_id_function(&mut id_map, receiver.recv().expect("Recv failed")),
138                $id_type(5)
139            );
140            assert_eq!(
141                $map_id_function(&mut id_map, receiver.recv().expect("Recv failed")),
142                $id_type(6)
143            );
144            assert_eq!(
145                $map_id_function(&mut id_map, receiver.recv().expect("Recv failed")),
146                $id_type(7)
147            );
148            assert_eq!(
149                $map_id_function(&mut id_map, receiver.recv().expect("Recv failed")),
150                $id_type(8)
151            );
152            assert_eq!(
153                $map_id_function(&mut id_map, receiver.recv().expect("Recv failed")),
154                $id_type(9)
155            );
156        };
157    }
158
159    test_sequential_id_assignment!(
160        DevtoolsBrowserId,
161        || WebViewId::new(),
162        |id_map: &mut IdMap, id| id_map.browser_id(id)
163    );
164    test_sequential_id_assignment!(
165        DevtoolsBrowsingContextId,
166        || BrowsingContextId::new(),
167        |id_map: &mut IdMap, id| id_map.browsing_context_id(id)
168    );
169    test_sequential_id_assignment!(
170        DevtoolsOuterWindowId,
171        || PipelineId::new(),
172        |id_map: &mut IdMap, id| id_map.outer_window_id(id)
173    );
174}
devtools/id.rs

devtools/
id.rs
