docker/pgcat
1
0
Fork 0
mirror of https://github.com/postgresml/pgcat.git synced 2026-03-23 01:16:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

simplify TableAccess::run

Browse Source
This commit is contained in:
Kevin Zimmerman
2023-07-25 19:34:37 -05:00
parent 1c26aa3547
commit 33724ea670
1 changed files with 4 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/plugins/table_access.rs
View File

@@ -30,21 +30,18 @@ impl<'a> Plugin for TableAccess<'a> {
return Ok(PluginOutput::Allow);
}
let mut found = None;
visit_relations(ast, |relation| {
let control_flow = visit_relations(ast, |relation| {
let relation = relation.to_string();
let table_name = relation.split('.').last().unwrap().to_string();
if self.tables.contains(&table_name) {
found = Some(table_name);
ControlFlow::<()>::Break(())
ControlFlow::Break(table_name)
} else {
ControlFlow::<()>::Continue(())
ControlFlow::Continue(())
}
});
if let Some(found) = found {
if let ControlFlow::Break(found) = control_flow {
debug!("Blocking access to table \"{found}\"");
Ok(PluginOutput::Deny(format!(