oryxc/src/depmap.rs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

use boxcar;
use dashmap::DashMap;

use crate::compiler::Job;
use crate::prelude::*;

pub struct DepMap(DashMap<Dependency, boxcar::Vec<Job>>);

#[derive(Eq, Hash, PartialEq)]
pub enum Dependency {
	Symbol(SymbolId),
}

impl DepMap {
	pub fn with_capacity(n: usize) -> Self {
		return Self(DashMap::with_capacity(n));
	}

	pub fn add(&self, d: Dependency, j: Job) {
		self.0
			.entry(d)
			.and_modify(|v| {
				v.push(j.clone());
			})
			.or_insert_with(|| boxcar::vec![j]);
	}

	pub fn pop(&self, d: Dependency) -> Option<boxcar::Vec<Job>> {
		return self.0.remove(&d).map(|(_, v)| v);
	}
}