lourenco/mgmt
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

pgraph: Filter graph should not include unfiltered vertices

Browse Source 
This adds a test for another sneaky aspect of filter graph: It should
not pull in a vertex because another across the edge was included. If we
want this behaviour, then we need a different function or a config
option for this filter function.
This commit is contained in:
James Shubin
2024-01-07 18:05:41 -05:00
parent cb021e3a25
commit 58607e2ca0
2 changed files with 31 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
pgraph/pgraph.go
View File

@@ -512,7 +512,7 @@ func (g *Graph) FilterGraph(name string, vertices []Vertex) (*Graph, error) {
}
for k2, e := range x {
//log.Printf("Filter: %s -> %s # %s", k1.Name, k2.Name, e.Name)
if contains || VertexContains(k2, vertices) {
if contains && VertexContains(k2, vertices) {
newGraph.AddEdge(k1, k2, e)
}
}
@@ -538,7 +538,8 @@ func (g *Graph) DisconnectedGraphs() ([]*Graph, error) {
// dfs through the graph
dfs := g.DFS(start)
// filter all the collected elements into a new graph
newgraph, err := g.FilterGraph(g.Name, dfs)
// TODO: is this method of filtering correct here? && or || ?
newGraph, err := g.FilterGraph(g.Name, dfs)
if err != nil {
return nil, errwrap.Wrapf(err, "could not run DisconnectedGraphs() properly")
}
@@ -546,7 +547,7 @@ func (g *Graph) DisconnectedGraphs() ([]*Graph, error) {
d = append(d, dfs...) // extend
// append this new graph to the list
graphs = append(graphs, newgraph)
graphs = append(graphs, newGraph)
// if we've found all the elements, then we're done
// otherwise loop through to continue...