fix: add client error unwrap

Author: franklinkim

clienterror.go

@@ -9,3 +9,11 @@ func NewClientError(err error) *ClientError {
 		error: err,
 	}
 }
+
+// Unwrap interface
+func (e *ClientError) Unwrap() error {
+	if e != nil && e.error != nil {
+		return e.error
+	}
+	return nil
+}

example/basic/client/src/service-client.ts

@@ -16,6 +16,9 @@ export class ServiceClient {
 	async boolSlice(v:Array<boolean>|null):Promise<Array<boolean>|null> {
 		return (await this.transport<{0:Array<boolean>|null}>("BoolSlice", [v]))[0]
 	}
+	async context():Promise<void> {
+		await this.transport<void>("Context", [])
+	}
 	async empty():Promise<void> {
 		await this.transport<void>("Empty", [])
 	}

example/basic/main_test.go

@@ -0,0 +1,27 @@
+package main_test
+
+import (
+	"context"
+	"net/http/httptest"
+	"testing"
+	"time"
+
+	"github.com/foomo/gotsrpc/v2/example/basic/service"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestContextCanceled(t *testing.T) {
+	ctx, cancel := context.WithCancel(context.TODO())
+	go func() {
+		time.Sleep(time.Second)
+		cancel()
+	}()
+
+	svr := httptest.NewServer(service.NewDefaultServiceGoTSRPCProxy(&service.Handler{}))
+	defer svr.Close()
+
+	client := service.NewDefaultServiceGoTSRPCClient(svr.URL)
+
+	clientErr := client.Context(ctx)
+	assert.ErrorIs(t, clientErr, context.Canceled)
+}

example/basic/service/gorpc_gen.go

@@ -1,10 +1,20 @@
 package service
 
+import (
+	"net/http"
+	"time"
+)
+
 type Handler struct{}
 
-func (h *Handler) Empty() {
+func (h *Handler) Context(w http.ResponseWriter, r *http.Request) {
+	for r.Context().Err() == nil {
+		time.Sleep(100 * time.Millisecond)
+	}
 }
 
+func (h *Handler) Empty() {}
+
 func (h *Handler) Bool(v bool) bool {
 	return v
 }

example/basic/service/gorpcclient_gen.go

@@ -1,6 +1,11 @@
 package service
 
+import (
+	"net/http"
+)
+
 type Service interface {
+	Context(w http.ResponseWriter, r *http.Request)
 	Empty()
 	Bool(v bool) bool
 	BoolPtr(v bool) *bool

example/basic/service/gotsrpc_gen.go

@@ -1,7 +1,9 @@
 package frontend
 
 import (
+	"fmt"
 	"net/http"
+	"time"
 
 	"github.com/foomo/gotsrpc/v2/example/errors/service/backend"
 	"github.com/foomo/gotsrpc/v2/example/errors/service/frontend"
@@ -18,6 +20,12 @@ func New(client backend.ServiceGoTSRPCClient) *Handler {
 }
 
 func (h *Handler) Simple(w http.ResponseWriter, r *http.Request) (e *frontend.ErrSimple) {
+	fmt.Println("==========> incoming")
+	time.Sleep(time.Second)
+	if r.Context().Err() != nil {
+		fmt.Println("==========>" + r.Context().Err().Error())
+	}
+	fmt.Println("<========== outgoing")
 	return
 }