Return ErrWebrpcClientDisconnected on stream ctx timeout/cancelation

webrpc · Apr 17, 2024 · edb53fa · edb53fa
1 parent 931e7fe
commit edb53fa
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/_examples/golang-basics/example.gen.go b/_examples/golang-basics/example.gen.go
diff --git a/_examples/golang-imports/api.gen.go b/_examples/golang-imports/api.gen.go
diff --git a/client.go.tmpl b/client.go.tmpl
@@ -171,6 +171,9 @@ func (r *streamReader) handleReadError(err error) error {
 	if errors.Is(err, io.ErrUnexpectedEOF) {
 		return ErrWebrpcStreamLost.WithCause(err)
 	}
+	if errors.Is(err, context.Canceled) || errors.Is(err, context.DeadlineExceeded){
+		return ErrWebrpcClientDisconnected.WithCause(err)
+	}
 	return ErrWebrpcBadResponse.WithCause(fmt.Errorf("reading stream: %w", err))
 }