/
downstream_request.go
39 lines (31 loc) · 1.26 KB
/
downstream_request.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
package tf6serverlogging
import (
"context"
"time"
"github.com/hashicorp/terraform-plugin-go/internal/logging"
"github.com/hashicorp/terraform-plugin-go/tfprotov6/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)
}