29 lines
627 B
Go
29 lines
627 B
Go
|
package terraform
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/hashicorp/terraform/dag"
|
||
|
|
)
|
||
|
|
|
||
|
|
// CountBoundaryTransformer adds a node that depends on everything else
|
||
|
|
// so that it runs last in order to clean up the state for nodes that
|
||
|
|
// are on the "count boundary": "foo.0" when only one exists becomes "foo"
|
||
|
|
type CountBoundaryTransformer struct{}
|
||
|
|
|
||
|
|
func (t *CountBoundaryTransformer) Transform(g *Graph) error {
|
||
|
|
node := &NodeCountBoundary{}
|
||
|
|
g.Add(node)
|
||
|
|
|
||
|
|
// Depends on everything
|
||
|
|
for _, v := range g.Vertices() {
|
||
|
|
// Don't connect to ourselves
|
||
|
|
if v == node {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
|
||
|
|
// Connect!
|
||
|
|
g.Connect(dag.BasicEdge(node, v))
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|