41 lines
1.3 KiB
Go
41 lines
1.3 KiB
Go
package tf5serverlogging
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/hashicorp/terraform-plugin-go/internal/logging"
|
|
"github.com/hashicorp/terraform-plugin-go/tfprotov5/internal/diag"
|
|
)
|
|
|
|
// DownstreamRequest sets a request duration start time context key and
|
|
// generates a TRACE "Sending request downstream" log.
|
|
func DownstreamRequest(ctx context.Context) context.Context {
|
|
requestStart := time.Now()
|
|
ctx = context.WithValue(ctx, ContextKeyDownstreamRequestStartTime{}, requestStart)
|
|
|
|
logging.ProtocolTrace(ctx, "Sending request downstream")
|
|
|
|
return ctx
|
|
}
|
|
|
|
// DownstreamResponse generates the following logging:
|
|
//
|
|
// - TRACE "Received downstream response" log with request duration and
|
|
// diagnostic severity counts
|
|
// - Per-diagnostic logs
|
|
//
|
|
func DownstreamResponse(ctx context.Context, diagnostics diag.Diagnostics) {
|
|
responseFields := map[string]interface{}{
|
|
logging.KeyDiagnosticErrorCount: diagnostics.ErrorCount(),
|
|
logging.KeyDiagnosticWarningCount: diagnostics.WarningCount(),
|
|
}
|
|
|
|
if requestStart, ok := ctx.Value(ContextKeyDownstreamRequestStartTime{}).(time.Time); ok {
|
|
responseFields[logging.KeyRequestDurationMs] = time.Since(requestStart).Milliseconds()
|
|
}
|
|
|
|
logging.ProtocolTrace(ctx, "Received downstream response", responseFields)
|
|
diagnostics.Log(ctx)
|
|
}
|