Set grpc timeout on all requests, not just the first one. (#355)

Co-authored-by: Bjorn Melinder <[email protected]>
ExpediaGroup · Mar 8, 2024 · 6eba3d6 · 6eba3d6
1 parent 722aff9
commit 6eba3d6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/internal/pkg/grpc/client.go b/internal/pkg/grpc/client.go
@@ -128,7 +128,8 @@ func (c *Client) SendRequest(serviceMethod string, message string, headers []str
 		return response.Response{Duration: time.Duration(0), Err: errors.New("no connection available"), Type: respType}
 	}
 
-	err = grpcurl.InvokeRPC(context.Background(), c.descriptorSource, c.conn, serviceMethod, interpolatedHeaders, loggingEventHandler, requestParser.Next)
+	ctx, _ := context.WithTimeout(context.Background(), time.Duration(c.timeoutMilliseconds)*time.Millisecond)
+	err = grpcurl.InvokeRPC(ctx, c.descriptorSource, c.conn, serviceMethod, interpolatedHeaders, loggingEventHandler, requestParser.Next)
 	endTime := time.Now()
 	if err != nil {
 		log.Printf("grpc response error: %s", err)