mirror of
https://github.com/golang/go
synced 2024-10-01 08:28:43 -06:00
65e3620a7a
Change-Id: I5de33edc352a5202fdf9e38538a224dd6adafedb Reviewed-on: https://go-review.googlesource.com/c/tools/+/190799 Run-TryBot: Ian Cottrell <iancottrell@google.com> TryBot-Result: Gobot Gobot <gobot@golang.org> Reviewed-by: Rebecca Stambler <rstambler@golang.org> Reviewed-by: Emmanuel Odeke <emm.odeke@gmail.com>
46 lines
1.1 KiB
Go
46 lines
1.1 KiB
Go
package protocol
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"golang.org/x/tools/internal/telemetry"
|
|
"golang.org/x/tools/internal/telemetry/export"
|
|
"golang.org/x/tools/internal/xcontext"
|
|
)
|
|
|
|
func init() {
|
|
export.AddExporters(logExporter{})
|
|
}
|
|
|
|
type contextKey int
|
|
|
|
const (
|
|
clientKey = contextKey(iota)
|
|
)
|
|
|
|
func WithClient(ctx context.Context, client Client) context.Context {
|
|
return context.WithValue(ctx, clientKey, client)
|
|
}
|
|
|
|
// logExporter sends the log event back to the client if there is one stored on the
|
|
// context.
|
|
type logExporter struct{}
|
|
|
|
func (logExporter) StartSpan(context.Context, *telemetry.Span) {}
|
|
func (logExporter) FinishSpan(context.Context, *telemetry.Span) {}
|
|
func (logExporter) Metric(context.Context, telemetry.MetricData) {}
|
|
func (logExporter) Flush() {}
|
|
|
|
func (logExporter) Log(ctx context.Context, event telemetry.Event) {
|
|
client, ok := ctx.Value(clientKey).(Client)
|
|
if !ok {
|
|
return
|
|
}
|
|
msg := &LogMessageParams{Type: Info, Message: fmt.Sprint(event)}
|
|
if event.Error != nil {
|
|
msg.Type = Error
|
|
}
|
|
go client.LogMessage(xcontext.Detach(ctx), msg)
|
|
}
|