mirror of
https://github.com/opentffoundation/opentf.git
synced 2026-02-17 01:00:39 -05:00
5ca118b4e6
CountHook is an implementation of terraform.Hook which is used to calculate how many resources were added, changed, or destroyed during an apply. This hook was previously injected in the local backend code, which means that the apply command code has no access to these counts. This commit moves the CountHook code into the command package, and removes an unused instance of the hook in the plan code path. The goal here is moving UI code into the command package.
107 lines
2.3 KiB
Go
107 lines
2.3 KiB
Go
package command
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/zclconf/go-cty/cty"
|
|
|
|
"github.com/hashicorp/terraform/addrs"
|
|
"github.com/hashicorp/terraform/plans"
|
|
"github.com/hashicorp/terraform/states"
|
|
"github.com/hashicorp/terraform/terraform"
|
|
)
|
|
|
|
// CountHook is a hook that counts the number of resources
|
|
// added, removed, changed during the course of an apply.
|
|
type CountHook struct {
|
|
Added int
|
|
Changed int
|
|
Removed int
|
|
|
|
ToAdd int
|
|
ToChange int
|
|
ToRemove int
|
|
ToRemoveAndAdd int
|
|
|
|
pending map[string]plans.Action
|
|
|
|
sync.Mutex
|
|
terraform.NilHook
|
|
}
|
|
|
|
var _ terraform.Hook = (*CountHook)(nil)
|
|
|
|
func (h *CountHook) Reset() {
|
|
h.Lock()
|
|
defer h.Unlock()
|
|
|
|
h.pending = nil
|
|
h.Added = 0
|
|
h.Changed = 0
|
|
h.Removed = 0
|
|
}
|
|
|
|
func (h *CountHook) PreApply(addr addrs.AbsResourceInstance, gen states.Generation, action plans.Action, priorState, plannedNewState cty.Value) (terraform.HookAction, error) {
|
|
h.Lock()
|
|
defer h.Unlock()
|
|
|
|
if h.pending == nil {
|
|
h.pending = make(map[string]plans.Action)
|
|
}
|
|
|
|
h.pending[addr.String()] = action
|
|
|
|
return terraform.HookActionContinue, nil
|
|
}
|
|
|
|
func (h *CountHook) PostApply(addr addrs.AbsResourceInstance, gen states.Generation, newState cty.Value, err error) (terraform.HookAction, error) {
|
|
h.Lock()
|
|
defer h.Unlock()
|
|
|
|
if h.pending != nil {
|
|
pendingKey := addr.String()
|
|
if action, ok := h.pending[pendingKey]; ok {
|
|
delete(h.pending, pendingKey)
|
|
|
|
if err == nil {
|
|
switch action {
|
|
case plans.CreateThenDelete, plans.DeleteThenCreate:
|
|
h.Added++
|
|
h.Removed++
|
|
case plans.Create:
|
|
h.Added++
|
|
case plans.Delete:
|
|
h.Removed++
|
|
case plans.Update:
|
|
h.Changed++
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return terraform.HookActionContinue, nil
|
|
}
|
|
|
|
func (h *CountHook) PostDiff(addr addrs.AbsResourceInstance, gen states.Generation, action plans.Action, priorState, plannedNewState cty.Value) (terraform.HookAction, error) {
|
|
h.Lock()
|
|
defer h.Unlock()
|
|
|
|
// We don't count anything for data resources
|
|
if addr.Resource.Resource.Mode == addrs.DataResourceMode {
|
|
return terraform.HookActionContinue, nil
|
|
}
|
|
|
|
switch action {
|
|
case plans.CreateThenDelete, plans.DeleteThenCreate:
|
|
h.ToRemoveAndAdd += 1
|
|
case plans.Create:
|
|
h.ToAdd += 1
|
|
case plans.Delete:
|
|
h.ToRemove += 1
|
|
case plans.Update:
|
|
h.ToChange += 1
|
|
}
|
|
|
|
return terraform.HookActionContinue, nil
|
|
}
|