kata-containers/regorus
1
0
Fork 0
mirror of https://github.com/microsoft/regorus.git synced 2026-08-05 02:16:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use Ref for storing ast nodes in collections. (#37)

Browse Source 
cmp, eq is done as pointer comparison which makes retrieval super fast.

Signed-off-by: Anand Krishnamoorthi <anakrish@microsoft.com>
This commit is contained in:
Anand Krishnamoorthi
2023-11-05 22:32:00 -08:00
committed by GitHub
parent d3fd0a3a78
commit a751cd69e1
7 changed files with 137 additions and 116 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/scheduler.rs
+10 -13
View File
@@ -377,16 +377,16 @@ fn get_rule_prefix<'a>(expr: &Expr<'a>) -> Result<&'a str> {
pub struct Analyzer<'a> {
packages: BTreeMap<String, Scope<'a>>,
locals: BTreeMap<&'a Query<'a>, Scope<'a>>,
locals: BTreeMap<Ref<'a, Query<'a>>, Scope<'a>>,
scopes: Vec<Scope<'a>>,
order: BTreeMap<&'a Query<'a>, Vec<u16>>,
order: BTreeMap<Ref<'a, Query<'a>>, Vec<u16>>,
functions: FunctionTable<'a>,
}
#[derive(Clone)]
pub struct Schedule<'a> {
pub scopes: BTreeMap<&'a Query<'a>, Scope<'a>>,
pub order: BTreeMap<&'a Query<'a>, Vec<u16>>,
pub scopes: BTreeMap<Ref<'a, Query<'a>>, Scope<'a>>,
pub order: BTreeMap<Ref<'a, Query<'a>>, Vec<u16>>,
}
impl<'a> Default for Analyzer<'a> {
@@ -535,11 +535,8 @@ impl<'a> Analyzer<'a> {
RuleHead::Set { key, .. } => (key.as_ref(), None, scope),
RuleHead::Func { args, assign, .. } => {
for a in args.iter() {
match a {
Var(v) => {
scope.locals.insert(v.text());
}
_ => unimplemented!("non var arguments"),
if let Var(v) = a {
scope.locals.insert(v.text());
}
}
(None, assign.as_ref().map(|a| &a.value), scope)
@@ -672,7 +669,7 @@ impl<'a> Analyzer<'a> {
let compr_scope = match compr {
Expr::ArrayCompr { query, term, .. } | Expr::SetCompr { query, term, .. } => {
self.analyze_query(None, Some(term), query, Scope::default())?;
self.locals.get(query)
self.locals.get(&Ref::make(query))
}
Expr::ObjectCompr {
query, key, value, ..
@@ -683,7 +680,7 @@ impl<'a> Analyzer<'a> {
query,
Scope::default(),
)?;
self.locals.get(query)
self.locals.get(&Ref::make(query))
}
_ => break,
};
@@ -1041,14 +1038,14 @@ impl<'a> Analyzer<'a> {
let res = schedule(&mut infos[..]);
match res {
Ok(SortResult::Order(ord)) => {
self.order.insert(query, ord);
self.order.insert(Ref::make(query), ord);
}
Err(err) => {
bail!(query.span.error(&err.to_string()))
}
_ => (),
}
self.locals.insert(query, scope);
self.locals.insert(Ref::make(query), scope);
Ok(())
}