mirror of
https://github.com/opentffoundation/opentf.git
synced 2025-12-22 11:17:35 -05:00
4218231438
This covers most of the logic required to turn a source graph into a compiled graph ready for execution. There's currently only support for one of the opcodes though, so subsequent commits will sketch those out more and then add some tests and fix any problems that inevitably exist here but aren't yet visible because there are no tests. Signed-off-by: Martin Atkins <mart@degeneration.co.uk>
70 lines
2.5 KiB
Go
70 lines
2.5 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 grapheval
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/apparentlymart/go-workgraph/workgraph"
|
|
)
|
|
|
|
// ContextWithWorker returns a child of the given context that is associated
|
|
// with the given [workgraph.Worker].
|
|
//
|
|
// This is a very low-level API for use by code that is interacting directly
|
|
// with the [workgraph] package API. Callers should prefer to use the
|
|
// higher-level wrappers in this package whenever possible, because they
|
|
// manage context-based worker tracking automatically on the caller's behalf.
|
|
func ContextWithWorker(parent context.Context, worker *workgraph.Worker) context.Context {
|
|
return context.WithValue(parent, workerContextKey, worker)
|
|
}
|
|
|
|
// ContextWithNewWorker is like [ContextWithWorker] except that it internally
|
|
// creates a new worker and associates that with the returned context.
|
|
//
|
|
// This is a good way to create the top-level context needed for first entry
|
|
// into a call graph that relies on the self-reference-detecting functions
|
|
// elsewhere in this package. Those functions will then create other workers
|
|
// as necessary.
|
|
func ContextWithNewWorker(parent context.Context) context.Context {
|
|
return ContextWithWorker(parent, workgraph.NewWorker())
|
|
}
|
|
|
|
// WorkerFromContext returns a pointer to the [workgraph.Worker] associated
|
|
// with the given context, or panics if the context has no worker.
|
|
func WorkerFromContext(ctx context.Context) *workgraph.Worker {
|
|
worker, ok := ctx.Value(workerContextKey).(*workgraph.Worker)
|
|
if !ok {
|
|
panic("no worker handle in this context")
|
|
}
|
|
return worker
|
|
}
|
|
|
|
// ContextWithRequestTracker returns a child of the given context that is
|
|
// associated with the given [RequestTracker].
|
|
//
|
|
// Pass promises derived from the result to other functions in this package that
|
|
// perform self-reference and unresolved request detection to improve the
|
|
// error messages returned when those error situations occur.
|
|
func ContextWithRequestTracker(parent context.Context, tracker RequestTracker) context.Context {
|
|
return context.WithValue(parent, trackerContextKey, tracker)
|
|
}
|
|
|
|
// RequestTrackerFromContext returns the request tracker associated with the
|
|
// given context, or nil if there is no request tracker.
|
|
func RequestTrackerFromContext(ctx context.Context) RequestTracker {
|
|
tracker, ok := ctx.Value(trackerContextKey).(RequestTracker)
|
|
if !ok {
|
|
return nil
|
|
}
|
|
return tracker
|
|
}
|
|
|
|
type contextKey rune
|
|
|
|
const workerContextKey = contextKey('W')
|
|
const trackerContextKey = contextKey('T')
|