hohn/codeql
1
0
Fork 0
mirror of https://github.com/github/codeql.git synced 2026-02-19 08:23:45 +01:00
Code Issues Packages Projects Releases Wiki Activity

Simplify hashmap insertion

Browse Source
This commit is contained in:
Nick Rolfe
2020-10-28 11:20:47 +00:00
parent 679ca6d0f1
commit b4f9599dd9
1 changed files with 6 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
generator/src/ql_gen.rs
View File

@@ -42,16 +42,9 @@ fn create_supertype_map(nodes: &[node_types::Entry]) -> SupertypeMap {
let subtype_class_name = dbscheme_name_to_class_name(&node_types::escape_name(
&node_types::node_type_name(&n_member.kind, n_member.named),
));
match map.get_mut(&subtype_class_name) {
Some(set) => {
set.insert(supertype_class_name.clone());
}
None => {
let mut set: BTreeSet<String> = BTreeSet::new();
set.insert(supertype_class_name.clone());
map.insert(subtype_class_name.clone(), set);
}
}
map.entry(subtype_class_name)
.or_insert_with(|| BTreeSet::new())
.insert(supertype_class_name.clone());
}
}
node_types::Entry::Table { type_name, fields } => {
@@ -76,16 +69,9 @@ fn create_supertype_map(nodes: &[node_types::Entry]) -> SupertypeMap {
node_types::node_type_name(&field_type.kind, field_type.named);
let member_class_name =
dbscheme_name_to_class_name(&node_types::escape_name(&member_name));
match map.get_mut(&member_class_name) {
Some(set) => {
set.insert(supertype_name.clone());
}
None => {
let mut set: BTreeSet<String> = BTreeSet::new();
set.insert(supertype_name.clone());
map.insert(member_class_name.clone(), set);
}
}
map.entry(member_class_name)
.or_insert_with(|| BTreeSet::new())
.insert(supertype_name.clone());
}
}
}