mirror of
https://github.com/opentffoundation/opentf.git
synced 2025-12-19 17:59:05 -05:00
75c7834ca6
Most of our transformers are pure compute and so don't really have a strong need to generate trace spans under our current focus of only exposing user-facing concepts and external requests in our traces, but unfortunately some of them indirectly depend on provider schema, which in turn means that they can potentially be unlucky enough to be the trigger for making all of the provider requests needed to fill the schema cache and therefore would end up with provider request spans being reported beneath them. As usual with these interface updates, this initial change focuses only on changing the interface and updating its direct callers and implementers to match, without any further refactoring or attempts to plumb contexts to or from other functions that don't have them yet. That means there are a few new context.TODO() calls here that we'll tidy up in a later commit that hopefully won't involve all of the noise that is caused by changing an interface API. Signed-off-by: Martin Atkins <mart@degeneration.co.uk>
75 lines
2.3 KiB
Go
75 lines
2.3 KiB
Go
// Copyright (c) The OpenTofu Authors
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
// Copyright (c) 2023 HashiCorp, Inc.
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
package tofu
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
|
|
"github.com/opentofu/opentofu/internal/dag"
|
|
"github.com/opentofu/opentofu/internal/states"
|
|
)
|
|
|
|
// GraphNodeAttachResourceState is an interface that can be implemented
|
|
// to request that a ResourceState is attached to the node.
|
|
//
|
|
// Due to a historical naming inconsistency, the type ResourceState actually
|
|
// represents the state for a particular _instance_, while InstanceState
|
|
// represents the values for that instance during a particular phase
|
|
// (e.g. primary vs. deposed). Consequently, GraphNodeAttachResourceState
|
|
// is supported only for nodes that represent resource instances, even though
|
|
// the name might suggest it is for containing resources.
|
|
type GraphNodeAttachResourceState interface {
|
|
GraphNodeResourceInstance
|
|
|
|
// Sets the state
|
|
AttachResourceState(*states.Resource)
|
|
}
|
|
|
|
// AttachStateTransformer goes through the graph and attaches
|
|
// state to nodes that implement the interfaces above.
|
|
type AttachStateTransformer struct {
|
|
State *states.State // State is the root state
|
|
}
|
|
|
|
func (t *AttachStateTransformer) Transform(_ context.Context, g *Graph) error {
|
|
// If no state, then nothing to do
|
|
if t.State == nil {
|
|
log.Printf("[DEBUG] Not attaching any node states: overall state is nil")
|
|
return nil
|
|
}
|
|
|
|
for _, v := range g.Vertices() {
|
|
// Nodes implement this interface to request state attachment.
|
|
an, ok := v.(GraphNodeAttachResourceState)
|
|
if !ok {
|
|
continue
|
|
}
|
|
addr := an.ResourceInstanceAddr()
|
|
|
|
rs := t.State.Resource(addr.ContainingResource())
|
|
if rs == nil {
|
|
log.Printf("[DEBUG] Resource state not found for node %q, instance %s", dag.VertexName(v), addr)
|
|
continue
|
|
}
|
|
|
|
is := rs.Instance(addr.Resource.Key)
|
|
if is == nil {
|
|
// We don't actually need this here, since we'll attach the whole
|
|
// resource state, but we still check because it'd be weird
|
|
// for the specific instance we're attaching to not to exist.
|
|
log.Printf("[DEBUG] Resource instance state not found for node %q, instance %s", dag.VertexName(v), addr)
|
|
continue
|
|
}
|
|
|
|
// make sure to attach a copy of the state, so instances can modify the
|
|
// same ResourceState.
|
|
an.AttachResourceState(rs.DeepCopy())
|
|
}
|
|
|
|
return nil
|
|
}
|