mirror of
https://github.com/golang/go
synced 2024-11-19 07:04:43 -07:00
cb106d260e
This allows early exporters to adjust the event for later ones. This is used to lookup key values from the context if needed. Also add a Query type event which is intended to perform all event modifications but nothing else, and is used to lookup values from the context. This cleans up a weirdness where the current lookup presumes there will be an exporter with a matching mechanism. Change-Id: I835d1e0b2511553c30f94b7becfe7b7b5462c111 Reviewed-on: https://go-review.googlesource.com/c/tools/+/223657 Run-TryBot: Ian Cottrell <iancottrell@google.com> TryBot-Result: Gobot Gobot <gobot@golang.org> Reviewed-by: Emmanuel Odeke <emm.odeke@gmail.com>
36 lines
692 B
Go
36 lines
692 B
Go
package protocol
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"golang.org/x/tools/internal/telemetry/event"
|
|
"golang.org/x/tools/internal/xcontext"
|
|
)
|
|
|
|
type contextKey int
|
|
|
|
const (
|
|
clientKey = contextKey(iota)
|
|
)
|
|
|
|
func WithClient(ctx context.Context, client Client) context.Context {
|
|
return context.WithValue(ctx, clientKey, client)
|
|
}
|
|
|
|
func LogEvent(ctx context.Context, ev event.Event) (context.Context, event.Event) {
|
|
if !ev.IsLog() {
|
|
return ctx, ev
|
|
}
|
|
client, ok := ctx.Value(clientKey).(Client)
|
|
if !ok {
|
|
return ctx, ev
|
|
}
|
|
msg := &LogMessageParams{Type: Info, Message: fmt.Sprint(ev)}
|
|
if ev.Error != nil {
|
|
msg.Type = Error
|
|
}
|
|
go client.LogMessage(xcontext.Detach(ctx), msg)
|
|
return ctx, ev
|
|
}
|