starpc 0.49.18 → 0.50.0

package/srpc/server-rpc.ts

@@ -3,6 +3,7 @@ import type { Sink, Source } from 'it-stream-types'
 import type { CallData, CallStart } from './rpcproto.pb.js'
 import { CommonRPC } from './common-rpc.js'
 import { InvokeFn } from './handler.js'
+import { ServerInvocation } from './server-invocation.js'
 import { LookupMethod } from './mux.js'
 
 // ServerRPC is an ongoing RPC from the server side.
@@ -43,12 +44,15 @@ export class ServerRPC extends CommonRPC {
     }
     return super.handleCallData(packet)
   }
-
   // invokeRPC starts invoking the RPC handler.
   private async invokeRPC(invokeFn: InvokeFn) {
     const dataSink = this._createDataSink()
+    const invocation = new ServerInvocation(
+      this.invocationSignal,
+      (ownerSignal) => this.waitTerminal(ownerSignal),
+    )
     try {
-      await invokeFn(this.rpcDataSource, dataSink)
+      await invokeFn(this.rpcDataSource, dataSink, invocation)
     } catch (err) {
       this.close(err as Error)
     }

package/srpc/stream-yamux.go

@@ -4,7 +4,7 @@ import (
 	"errors"
 	"time"
 
-	yamux "github.com/libp2p/go-yamux/v4"
+	yamux "github.com/libp2p/go-yamux/v5"
 )
 
 // yamuxStream wraps a yamux.Stream to implement MuxedStream.

package/srpc/stream.go

@@ -23,6 +23,10 @@ type Stream interface {
 	Close() error
 }
 
+type receiptTerminalStream interface {
+	receiptTerminalKind() (TerminalKind, bool)
+}
+
 // StreamRecv is a stream that can receive typed messages.
 //
 // T is the response type.
@@ -69,6 +73,14 @@ func (s *streamWithClose) Close() error {
 	return err2
 }
 
+func (s *streamWithClose) receiptTerminalKind() (TerminalKind, bool) {
+	receipt, ok := s.Stream.(receiptTerminalStream)
+	if !ok {
+		return 0, false
+	}
+	return receipt.receiptTerminalKind()
+}
+
 // streamWithContext is a Stream with a wrapped Context function.
 type streamWithContext struct {
 	Stream
@@ -85,6 +97,14 @@ func (s *streamWithContext) Context() context.Context {
 	return s.ctx
 }
 
+func (s *streamWithContext) receiptTerminalKind() (TerminalKind, bool) {
+	receipt, ok := s.Stream.(receiptTerminalStream)
+	if !ok {
+		return 0, false
+	}
+	return receipt.receiptTerminalKind()
+}
+
 // _ is a type assertion
 var (
 	_ Stream = (*streamWithClose)(nil)

package/srpc/websocket.go

@@ -4,7 +4,7 @@ import (
 	"context"
 
 	"github.com/aperturerobotics/go-websocket"
-	"github.com/libp2p/go-yamux/v4"
+	"github.com/libp2p/go-yamux/v5"
 )
 
 // NewWebSocketConn wraps a websocket into a MuxedConn.