blob: a9ef48d0f0be38fdcb673870ba3997e4d767ff59 [file] [log] [blame]
// Copyright 2019 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package protocol
import (
"bytes"
"context"
"golang.org/x/tools/internal/event"
"golang.org/x/tools/internal/event/core"
"golang.org/x/tools/internal/event/export"
"golang.org/x/tools/internal/event/label"
"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 core.Event, lm label.Map, mt MessageType) context.Context {
client, ok := ctx.Value(clientKey).(Client)
if !ok {
return ctx
}
buf := &bytes.Buffer{}
p := export.Printer{}
p.WriteEvent(buf, ev, lm)
msg := &LogMessageParams{Type: mt, Message: buf.String()}
// Handle messages generated via event.Error, which won't have a level Label.
if event.IsError(ev) {
msg.Type = Error
}
// TODO(adonovan): the goroutine here could cause log
// messages to be delivered out of order! Use a queue.
go client.LogMessage(xcontext.Detach(ctx), msg)
return ctx
}