1// Copyright 2018 The Go Authors. All rights reserved. 2// Use of this source code is governed by a BSD-style 3// license that can be found in the LICENSE file. 4 5package protocol 6 7import ( 8 "context" 9 10 "golang.org/x/tools/internal/jsonrpc2" 11 "golang.org/x/tools/internal/lsp/xlog" 12) 13 14const defaultMessageBufferSize = 20 15const defaultRejectIfOverloaded = false 16 17func canceller(ctx context.Context, conn *jsonrpc2.Conn, id jsonrpc2.ID) { 18 conn.Notify(context.Background(), "$/cancelRequest", &CancelParams{ID: id}) 19} 20 21func NewClient(stream jsonrpc2.Stream, client Client) (*jsonrpc2.Conn, Server, xlog.Logger) { 22 log := xlog.New(NewLogger(client)) 23 conn := jsonrpc2.NewConn(stream) 24 conn.Capacity = defaultMessageBufferSize 25 conn.RejectIfOverloaded = defaultRejectIfOverloaded 26 conn.Handler = clientHandler(log, client) 27 conn.Canceler = jsonrpc2.Canceler(canceller) 28 return conn, &serverDispatcher{Conn: conn}, log 29} 30 31func NewServer(stream jsonrpc2.Stream, server Server) (*jsonrpc2.Conn, Client, xlog.Logger) { 32 conn := jsonrpc2.NewConn(stream) 33 client := &clientDispatcher{Conn: conn} 34 log := xlog.New(NewLogger(client)) 35 conn.Capacity = defaultMessageBufferSize 36 conn.RejectIfOverloaded = defaultRejectIfOverloaded 37 conn.Handler = serverHandler(log, server) 38 conn.Canceler = jsonrpc2.Canceler(canceller) 39 return conn, client, log 40} 41 42func sendParseError(ctx context.Context, log xlog.Logger, req *jsonrpc2.Request, err error) { 43 if _, ok := err.(*jsonrpc2.Error); !ok { 44 err = jsonrpc2.NewErrorf(jsonrpc2.CodeParseError, "%v", err) 45 } 46 if err := req.Reply(ctx, nil, err); err != nil { 47 log.Errorf(ctx, "%v", err) 48 } 49} 50