// UncoveredDeploymentFlowNodes includes nodes that either services or deployment
// configs, or which haven't previously been covered.
func UncoveredDeploymentFlowNodes(covered osgraph.NodeSet) osgraph.NodeFunc {
return func(g osgraph.Interface, node graph.Node) bool {
switch node.(type) {
case *deploygraph.DeploymentConfigNode, *kubegraph.ServiceNode:
return true
}
return !covered.Has(node.ID())
}
}
// UncoveredDeploymentFlowEdges preserves (and duplicates) edges that were not
// covered by a deployment flow. As a special case, it preserves edges between
// Services and DeploymentConfigs.
func UncoveredDeploymentFlowEdges(covered osgraph.NodeSet) osgraph.EdgeFunc {
return func(g osgraph.Interface, head, tail graph.Node, edgeKind string) bool {
if edgeKind == kubeedges.ExposedThroughServiceEdgeKind {
return osgraph.AddReversedEdge(g, head, tail, osgraph.ReferencedByEdgeKind)
}
if covered.Has(head.ID()) && covered.Has(tail.ID()) {
return false
}
return osgraph.AddReversedEdge(g, head, tail, osgraph.ReferencedByEdgeKind)
}
}
// subgraphWithoutPrunableImages creates a subgraph from g with prunable image
// nodes excluded.
func subgraphWithoutPrunableImages(g graph.Graph, prunableImageIDs graph.NodeSet) graph.Graph {
return g.Subgraph(
func(g graph.Interface, node gonum.Node) bool {
return !prunableImageIDs.Has(node.ID())
},
func(g graph.Interface, head, tail gonum.Node, edgeKinds util.StringSet) bool {
if prunableImageIDs.Has(head.ID()) {
return false
}
if prunableImageIDs.Has(tail.ID()) {
return false
}
return true
},
)
}
// subgraphWithoutPrunableImages creates a subgraph from g with prunable image
// nodes excluded.
func subgraphWithoutPrunableImages(g graph.Graph, prunableImageIDs graph.NodeSet) graph.Graph {
return g.Subgraph(
func(g graph.Interface, node gonum.Node) bool {
return !prunableImageIDs.Has(node.ID())
},
func(g graph.Interface, from, to gonum.Node, edgeKinds sets.String) bool {
if prunableImageIDs.Has(from.ID()) {
return false
}
if prunableImageIDs.Has(to.ID()) {
return false
}
return true
},
)
}
