feat: prevent proxy context cancelled

docker-compose.yml

@@ -78,9 +78,9 @@ services:
   spicedb:
     image: quay.io/authzed/spicedb:v1.0.0
     ports:
-      - "8081:8080"
-      - "50052:50051"
-      - "50054:50053"
+      - "8080:8080"
+      - "50051:50051"
+      - "50053:50053"
     command:
       spicedb serve --grpc-preshared-key "shield" --grpc-no-tls --datastore-engine postgres
       --datastore-conn-uri postgres://spicedb:@pg2:5432/spicedb?sslmode=disable

internal/proxy/roundtripper.go

@@ -1,6 +1,7 @@
 package proxy
 
 import (
+	"context"
 	"crypto/tls"
 	"net"
 	"net/http"
@@ -28,6 +29,30 @@ type h2cTransportWrapper struct {
 	hook hook.Service
 }
 
+type Context struct {
+	ctx context.Context
+}
+
+func (c Context) Deadline() (time.Time, bool) {
+	return time.Time{}, false
+}
+
+func (c Context) Done() <-chan struct{} {
+	return nil
+}
+
+func (c Context) Err() error {
+	return nil
+}
+
+func (c Context) Value(key interface{}) interface{} {
+	return c.ctx.Value(key)
+}
+
+func WithoutCancel(ctx context.Context) context.Context {
+	return Context{ctx: ctx}
+}
+
 func (t *h2cTransportWrapper) RoundTrip(req *http.Request) (*http.Response, error) {
 	// we need to apply errors if it failed in Director
 	if err, ok := req.Context().Value(ctxRequestErrorKey).(error); ok {
@@ -42,6 +67,8 @@ func (t *h2cTransportWrapper) RoundTrip(req *http.Request) (*http.Response, erro
 		transport = t.grpcTransport
 	}
 
+	req = req.WithContext(WithoutCancel(req.Context()))
+
 	res, err := transport.RoundTrip(req)
 	if err != nil {
 		return res, err