52 lines
1.1 KiB
Go
52 lines
1.1 KiB
Go
|
package terraform
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"github.com/hashicorp/terraform/dag"
|
||
|
|
)
|
||
|
|
|
||
|
|
// ResourceCountTransformer is a GraphTransformer that expands the count
|
||
|
|
// out for a specific resource.
|
||
|
|
//
|
||
|
|
// This assumes that the count is already interpolated.
|
||
|
|
type ResourceCountTransformer struct {
|
||
|
|
Concrete ConcreteResourceNodeFunc
|
||
|
|
|
||
|
|
Count int
|
||
|
|
Addr *ResourceAddress
|
||
|
|
}
|
||
|
|
|
||
|
|
func (t *ResourceCountTransformer) Transform(g *Graph) error {
|
||
|
|
// Don't allow the count to be negative
|
||
|
|
if t.Count < 0 {
|
||
|
|
return fmt.Errorf("negative count: %d", t.Count)
|
||
|
|
}
|
||
|
|
|
||
|
|
// For each count, build and add the node
|
||
|
|
for i := 0; i < t.Count; i++ {
|
||
|
|
// Set the index. If our count is 1 we special case it so that
|
||
|
|
// we handle the "resource.0" and "resource" boundary properly.
|
||
|
|
index := i
|
||
|
|
if t.Count == 1 {
|
||
|
|
index = -1
|
||
|
|
}
|
||
|
|
|
||
|
|
// Build the resource address
|
||
|
|
addr := t.Addr.Copy()
|
||
|
|
addr.Index = index
|
||
|
|
|
||
|
|
// Build the abstract node and the concrete one
|
||
|
|
abstract := &NodeAbstractResource{Addr: addr}
|
||
|
|
var node dag.Vertex = abstract
|
||
|
|
if f := t.Concrete; f != nil {
|
||
|
|
node = f(abstract)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Add it to the graph
|
||
|
|
g.Add(node)
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|