mirror of
https://github.com/golang/go
synced 2024-11-18 20:44:45 -07:00
7504fd2219
ID's are now by value not pointer, which caused it to not use the Format method, resulting in broken id strings The id maps need to be crossover (set and get go to different maps for a given direction of message) Change-Id: Ide2b63ec1b009ae3587ee10e8bce018732ea342c Reviewed-on: https://go-review.googlesource.com/c/tools/+/229244 Run-TryBot: Ian Cottrell <iancottrell@google.com> TryBot-Result: Gobot Gobot <gobot@golang.org> Reviewed-by: Peter Weinberger <pjw@google.com>
81 lines
2.4 KiB
Go
81 lines
2.4 KiB
Go
// Copyright 2018 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 (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"golang.org/x/tools/internal/jsonrpc2"
|
|
"golang.org/x/tools/internal/telemetry/event"
|
|
"golang.org/x/tools/internal/xcontext"
|
|
)
|
|
|
|
var (
|
|
// RequestCancelledError should be used when a request is cancelled early.
|
|
RequestCancelledError = jsonrpc2.NewError(-32800, "JSON RPC cancelled")
|
|
)
|
|
|
|
// ClientDispatcher returns a Client that dispatches LSP requests across the
|
|
// given jsonrpc2 connection.
|
|
func ClientDispatcher(conn *jsonrpc2.Conn) Client {
|
|
return &clientDispatcher{Conn: conn}
|
|
}
|
|
|
|
// ServerDispatcher returns a Server that dispatches LSP requests across the
|
|
// given jsonrpc2 connection.
|
|
func ServerDispatcher(conn *jsonrpc2.Conn) Server {
|
|
return &serverDispatcher{Conn: conn}
|
|
}
|
|
|
|
func Handlers(handler jsonrpc2.Handler) jsonrpc2.Handler {
|
|
return CancelHandler(
|
|
CancelHandler(
|
|
jsonrpc2.AsyncHandler(
|
|
jsonrpc2.MustReplyHandler(handler))))
|
|
}
|
|
|
|
func CancelHandler(handler jsonrpc2.Handler) jsonrpc2.Handler {
|
|
handler, canceller := jsonrpc2.CancelHandler(handler)
|
|
return func(ctx context.Context, reply jsonrpc2.Replier, req jsonrpc2.Request) error {
|
|
if req.Method() != "$/cancelRequest" {
|
|
return handler(ctx, reply, req)
|
|
}
|
|
var params CancelParams
|
|
if err := json.Unmarshal(req.Params(), ¶ms); err != nil {
|
|
return sendParseError(ctx, reply, err)
|
|
}
|
|
if n, ok := params.ID.(float64); ok {
|
|
canceller(jsonrpc2.NewIntID(int64(n)))
|
|
} else if s, ok := params.ID.(string); ok {
|
|
canceller(jsonrpc2.NewStringID(s))
|
|
} else {
|
|
return sendParseError(ctx, reply, fmt.Errorf("request ID %v malformed", params.ID))
|
|
}
|
|
return reply(ctx, nil, nil)
|
|
}
|
|
}
|
|
|
|
func Call(ctx context.Context, conn *jsonrpc2.Conn, method string, params interface{}, result interface{}) error {
|
|
id, err := conn.Call(ctx, method, params, result)
|
|
if ctx.Err() != nil {
|
|
cancelCall(ctx, conn, id)
|
|
}
|
|
return err
|
|
}
|
|
|
|
func cancelCall(ctx context.Context, conn *jsonrpc2.Conn, id jsonrpc2.ID) {
|
|
ctx = xcontext.Detach(ctx)
|
|
ctx, done := event.StartSpan(ctx, "protocol.canceller")
|
|
defer done()
|
|
// Note that only *jsonrpc2.ID implements json.Marshaler.
|
|
conn.Notify(ctx, "$/cancelRequest", &CancelParams{ID: &id})
|
|
}
|
|
|
|
func sendParseError(ctx context.Context, reply jsonrpc2.Replier, err error) error {
|
|
return reply(ctx, nil, fmt.Errorf("%w: %s", jsonrpc2.ErrParse, err))
|
|
}
|