feat(go): endpoint level timeout

Author: Fluf22

clients/algoliasearch-client-go/algolia/transport/configuration.go

@@ -20,3 +20,9 @@ type Configuration struct {
 	Compression                     compression.Compression
 	ExposeIntermediateNetworkErrors bool
 }
+
+type RequestConfiguration struct {
+	ReadTimeout    *time.Duration
+	WriteTimeout   *time.Duration
+	ConnectTimeout *time.Duration
+}

clients/algoliasearch-client-go/algolia/transport/transport.go

@@ -44,7 +44,7 @@ func New(cfg Configuration) *Transport {
 	return transport
 }
 
-func (t *Transport) Request(ctx context.Context, req *http.Request, k call.Kind) (*http.Response, []byte, error) {
+func (t *Transport) Request(ctx context.Context, req *http.Request, k call.Kind, c *RequestConfiguration) (*http.Response, []byte, error) {
 	var intermediateNetworkErrors []error
 
 	// Add Content-Encoding header, if needed
@@ -59,9 +59,28 @@ func (t *Transport) Request(ctx context.Context, req *http.Request, k call.Kind)
 		// before the early returns, but when we do so, we do it **after**
 		// reading the body content of the response. Otherwise, a `context
 		// cancelled` error may happen when the body is read.
-		perRequestCtx, cancel := context.WithTimeout(ctx, h.timeout)
+		var (
+			ctxTimeout     time.Duration
+			connectTimeout time.Duration
+		)
+
+		if c != nil && k == call.Read && c.ReadTimeout != nil { //nolint:gocritic
+			ctxTimeout = *c.ReadTimeout
+		} else if c != nil && k == call.Write && c.WriteTimeout != nil {
+			ctxTimeout = *c.WriteTimeout
+		} else {
+			ctxTimeout = h.timeout
+		}
+
+		if c != nil && c.ConnectTimeout != nil {
+			connectTimeout = *c.ConnectTimeout
+		} else {
+			connectTimeout = t.connectTimeout
+		}
+
+		perRequestCtx, cancel := context.WithTimeout(ctx, ctxTimeout)
 		req = req.WithContext(perRequestCtx)
-		res, err := t.request(req, h, h.timeout, t.connectTimeout)
+		res, err := t.request(req, h, ctxTimeout, connectTimeout)
 
 		code := 0
 		if res != nil {

generators/src/main/java/com/algolia/codegen/AlgoliaGoGenerator.java

@@ -2,6 +2,7 @@
 
 import com.algolia.codegen.exceptions.*;
 import com.algolia.codegen.lambda.ScreamingSnakeCaseLambda;
+import com.algolia.codegen.lambda.ToSecondsLambda;
 import com.algolia.codegen.utils.*;
 import com.google.common.collect.ImmutableMap;
 import com.google.common.collect.Iterables;
@@ -66,7 +67,12 @@ public void processOpts() {
 
   @Override
   protected ImmutableMap.Builder<String, Mustache.Lambda> addMustacheLambdas() {
-    return super.addMustacheLambdas().put("screamingSnakeCase", new ScreamingSnakeCaseLambda());
+    ImmutableMap.Builder<String, Mustache.Lambda> lambdas = super.addMustacheLambdas();
+
+    lambdas.put("screamingSnakeCase", new ScreamingSnakeCaseLambda());
+    lambdas.put("toSeconds", new ToSecondsLambda());
+
+    return lambdas;
   }
 
   @Override

templates/go/api.mustache

@@ -19,8 +19,8 @@ import (
   "errors"
   "slices"
   "sort"
-  "time"
   {{/isSearchClient}}
+  "time"
 
   "github.com/algolia/algoliasearch-client-go/v4/algolia/utils"
 )
@@ -507,7 +507,11 @@ func (c *APIClient) {{nickname}}WithHTTPInfo({{#hasParams}}r {{#structPrefix}}{{
 		return nil, nil, err
 	}
 
-	return c.callAPI(req, {{#vendorExtensions}}{{#x-use-read-transporter}}true{{/x-use-read-transporter}}{{^x-use-read-transporter}}false{{/x-use-read-transporter}}{{/vendorExtensions}})
+	return c.callAPI(
+    req,
+    {{#vendorExtensions}}{{#x-use-read-transporter}}true{{/x-use-read-transporter}}{{^x-use-read-transporter}}false{{/x-use-read-transporter}},{{/vendorExtensions}}
+    {{#vendorExtensions.x-timeouts}}utils.ToPtr(transport.RequestConfiguration{ReadTimeout: utils.ToPtr({{#lambda.toSeconds}}{{{read}}}{{/lambda.toSeconds}} * time.Second), WriteTimeout: utils.ToPtr({{#lambda.toSeconds}}{{{write}}}{{/lambda.toSeconds}} * time.Second), ConnectTimeout: utils.ToPtr({{#lambda.toSeconds}}{{{connect}}}{{/lambda.toSeconds}} * time.Second)}){{/vendorExtensions.x-timeouts}}{{^vendorExtensions.x-timeouts}}nil{{/vendorExtensions.x-timeouts}},
+  )
 }
 
 /*

templates/go/client.mustache

@@ -129,13 +129,13 @@ func (c *APIClient) AddDefaultHeader(key string, value string) {
 }
 
 // callAPI do the request.
-func (c *APIClient) callAPI(request *http.Request, useReadTransporter bool) (*http.Response, []byte, error) {
+func (c *APIClient) callAPI(request *http.Request, useReadTransporter bool, requestConfiguration *transport.RequestConfiguration) (*http.Response, []byte, error) {
 	callKind := call.Write
   if useReadTransporter || request.Method == http.MethodGet {
     callKind = call.Read
   }
 
-  resp, body, err := c.transport.Request(request.Context(), request, callKind)
+  resp, body, err := c.transport.Request(request.Context(), request, callKind, requestConfiguration)
 	if err != nil {
 		return nil, nil, fmt.Errorf("failed to do request: %w", err)
 	}