mirror of
https://github.com/hashicorp/terraform.git
synced 2026-02-20 00:13:30 -05:00
133 lines
3 KiB
Go
133 lines
3 KiB
Go
// Copyright IBM Corp. 2014, 2026
|
|
// SPDX-License-Identifier: BUSL-1.1
|
|
|
|
package views
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/zclconf/go-cty/cty"
|
|
|
|
"github.com/hashicorp/terraform/internal/addrs"
|
|
"github.com/hashicorp/terraform/internal/plans"
|
|
"github.com/hashicorp/terraform/internal/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
|
|
Imported int
|
|
ActionInvocation 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
|
|
h.Imported = 0
|
|
h.ActionInvocation = 0
|
|
}
|
|
|
|
func (h *countHook) PreApply(id terraform.HookResourceIdentity, dk addrs.DeposedKey, 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[id.Addr.String()] = action
|
|
|
|
return terraform.HookActionContinue, nil
|
|
}
|
|
|
|
func (h *countHook) PostApply(id terraform.HookResourceIdentity, dk addrs.DeposedKey, newState cty.Value, err error) (terraform.HookAction, error) {
|
|
h.Lock()
|
|
defer h.Unlock()
|
|
|
|
if h.pending != nil {
|
|
pendingKey := id.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(id terraform.HookResourceIdentity, dk addrs.DeposedKey, action plans.Action, priorState, plannedNewState cty.Value, err error) (terraform.HookAction, error) {
|
|
h.Lock()
|
|
defer h.Unlock()
|
|
|
|
// Skip counting if there was an error
|
|
if err != nil {
|
|
return terraform.HookActionContinue, nil
|
|
}
|
|
|
|
// We don't count anything for data resources
|
|
if id.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
|
|
}
|
|
|
|
func (h *countHook) PostApplyImport(id terraform.HookResourceIdentity, importing plans.ImportingSrc) (terraform.HookAction, error) {
|
|
h.Lock()
|
|
defer h.Unlock()
|
|
|
|
h.Imported++
|
|
return terraform.HookActionContinue, nil
|
|
}
|
|
|
|
func (h *countHook) CompleteAction(id terraform.HookActionIdentity, err error) (terraform.HookAction, error) {
|
|
h.Lock()
|
|
defer h.Unlock()
|
|
|
|
h.ActionInvocation++
|
|
return terraform.HookActionContinue, nil
|
|
}
|