.github/workflows/ci.yml

@@ -18,9 +18,9 @@ jobs:
     name: Go ${{ matrix.go }}
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v6
+      - uses: actions/checkout@v4
       - name: Set up Go
-        uses: actions/setup-go@v6
+        uses: actions/setup-go@v5
         with:
           go-version: ${{ matrix.go }}
         id: go

.github/workflows/scip.yml

@@ -8,7 +8,7 @@ jobs:
     runs-on: ubuntu-latest
     container: sourcegraph/scip-go
     steps:
-      - uses: actions/checkout@v6
+      - uses: actions/checkout@v4
       - name: Get src-cli
         run: curl -L https://sourcegraph.com/.api/src-cli/src_linux_amd64 -o /usr/local/bin/src;
           chmod +x /usr/local/bin/src

conn.go

@@ -27,7 +27,6 @@ type Conn struct {
 
 	sending sync.Mutex
 
-	cancelCtx  context.CancelFunc
 	disconnect chan struct{}
 
 	logger Logger
@@ -44,19 +43,13 @@ var _ JSONRPC2 = (*Conn)(nil)
 // JSON-RPC protocol is symmetric, so a Conn runs on both ends of a
 // client-server connection.
 //
-// NewConn consumes stream, so you should call Close on the returned
-// Conn not on the given stream or its underlying connection.
-//
-// Conn is closed when the given context's Done channel is closed.
+// NewClient consumes conn, so you should call Close on the returned
+// client not on the given conn.
 func NewConn(ctx context.Context, stream ObjectStream, h Handler, opts ...ConnOpt) *Conn {
-
-	ctx, cancel := context.WithCancel(ctx)
-
 	c := &Conn{
 		stream:     stream,
 		h:          h,
 		pending:    map[ID]*call{},
-		cancelCtx:  cancel,
 		disconnect: make(chan struct{}),
 		logger:     log.New(os.Stderr, "", log.LstdFlags),
 	}
@@ -67,12 +60,6 @@ func NewConn(ctx context.Context, stream ObjectStream, h Handler, opts ...ConnOp
 		opt(c)
 	}
 	go c.readMessages(ctx)
-
-	go func() {
-		<-ctx.Done()
-		c.close(nil)
-	}()
-
 	return c
 }
 
@@ -195,7 +182,6 @@ func (c *Conn) close(cause error) error {
 	}
 
 	close(c.disconnect)
-	c.cancelCtx()
 	c.closed = true
 	return c.stream.Close()
 }

conn_test.go

@@ -14,58 +14,6 @@ import (
 	"github.com/sourcegraph/jsonrpc2"
 )
 
-func TestConn(t *testing.T) {
-
-	t.Run("closes when context is done", func(t *testing.T) {
-		ctx, cancel := context.WithCancel(context.Background())
-
-		connA, connB := Pipe(ctx, noopHandler{}, noopHandler{})
-		defer connA.Close()
-		defer connB.Close()
-
-		cancel()
-		<-connA.DisconnectNotify()
-
-		got := connA.Close()
-		want := jsonrpc2.ErrClosed
-		if got != want {
-			t.Fatalf("got %v, want %v", got, want)
-		}
-	})
-
-	t.Run("cancels context when closed", func(t *testing.T) {
-		ctxCanceled := make(chan struct{})
-
-		handler := handlerFunc(func(ctx context.Context, c *jsonrpc2.Conn, r *jsonrpc2.Request) {
-			// Block until the context is canceled.
-			<-ctx.Done()
-			close(ctxCanceled)
-		})
-
-		connA, connB := Pipe(context.Background(), noopHandler{}, jsonrpc2.AsyncHandler(handler))
-		defer connA.Close()
-		defer connB.Close()
-
-		// Send a notification from connA to connB to trigger connB's handler
-		// function.
-		if err := connA.Notify(context.Background(), "foo", nil, nil); err != nil {
-			t.Fatal(err)
-		}
-
-		// Disconnect connA from connB.
-		if err := connA.Close(); err != nil {
-			t.Fatal(err)
-		}
-
-		select {
-		case <-ctxCanceled:
-			// Test passed, the handler's context was canceled.
-		case <-time.After(time.Second):
-			t.Fatal("context not canceled")
-		}
-	})
-}
-
 var paramsTests = []struct {
 	sendParams interface{}
 	wantParams *json.RawMessage
@@ -250,12 +198,12 @@ func testParams(t *testing.T, want *json.RawMessage, fn func(c *jsonrpc2.Conn) e
 		wg.Done()
 	})
 
-	connA, connB := Pipe(context.Background(), noopHandler{}, handler)
-	defer connA.Close()
-	defer connB.Close()
+	client, server := newClientServer(handler)
+	defer client.Close()
+	defer server.Close()
 
 	wg.Add(1)
-	if err := fn(connA); err != nil {
+	if err := fn(client); err != nil {
 		t.Error(err)
 	}
 	wg.Wait()
@@ -294,11 +242,18 @@ func assertRawJSONMessage(t *testing.T, got *json.RawMessage, want *json.RawMess
 	}
 }
 
-// Pipe returns two jsonrpc2.Conn, connected via a synchronous, in-memory, full
-// duplex network connection.
-func Pipe(ctx context.Context, handlerA, handlerB jsonrpc2.Handler) (connA *jsonrpc2.Conn, connB *jsonrpc2.Conn) {
-	a, b := net.Pipe()
-	connA = jsonrpc2.NewConn(ctx, jsonrpc2.NewPlainObjectStream(a), handlerA)
-	connB = jsonrpc2.NewConn(ctx, jsonrpc2.NewPlainObjectStream(b), handlerB)
-	return connA, connB
+func newClientServer(handler jsonrpc2.Handler) (client *jsonrpc2.Conn, server *jsonrpc2.Conn) {
+	ctx := context.Background()
+	connA, connB := net.Pipe()
+	client = jsonrpc2.NewConn(
+		ctx,
+		jsonrpc2.NewPlainObjectStream(connA),
+		noopHandler{},
+	)
+	server = jsonrpc2.NewConn(
+		ctx,
+		jsonrpc2.NewPlainObjectStream(connB),
+		handler,
+	)
+	return client, server
 }

jsonrpc2.go

@@ -59,10 +59,10 @@ const (
 
 // Handler handles JSON-RPC requests and notifications.
 type Handler interface {
-	// Handle is called to handle a request. No other requests are handled until
-	// it returns. If you do not require strict ordering behavior of received
-	// RPCs, it is suggested to wrap your handler in AsyncHandler. The context
-	// is automatically canceled when the connection closes.
+	// Handle is called to handle a request. No other requests are handled
+	// until it returns. If you do not require strict ordering behavior
+	// of received RPCs, it is suggested to wrap your handler in
+	// AsyncHandler.
 	Handle(context.Context, *Conn, *Request)
 }