feat: Add option to pass external context into Go SDK's HTTP calls

Author: JCPistell

go/rtl/auth.go

@@ -67,8 +67,13 @@ func NewAuthSessionWithTransport(config ApiSettings, transport http.RoundTripper
 		AuthStyle:    oauth2.AuthStyleInParams,
 	}
 
+	bgCtx := context.Background()
+	if config.Context != nil {
+		bgCtx = config.Context
+	}
+
 	ctx := context.WithValue(
-		context.Background(),
+		bgCtx,
 		oauth2.HTTPClient,
 		// Will set "x-looker-appid" Header on TokenURL requests
 		&http.Client{Transport: appIdHeaderTransport},
@@ -123,18 +128,25 @@ func (s *AuthSession) Do(result interface{}, method, ver, path string, reqPars m
 		}
 	}
 
-	// create request context with timeout
-	var timeoutInSeconds int32 = 120 //seconds
-	if s.Config.Timeout != 0 {
-		timeoutInSeconds = s.Config.Timeout
-	}
-	if options != nil && options.Timeout != 0 {
-		timeoutInSeconds = options.Timeout
+	var ctx context.Context
+	if options != nil && options.Context != nil {
+		ctx = options.Context
+	} else if s.Config.Context != nil {
+		ctx = s.Config.Context
+	} else {
+		// create request context with timeout
+		var timeoutInSeconds int32 = 120 //seconds
+		if s.Config.Timeout != 0 {
+			timeoutInSeconds = s.Config.Timeout
+		}
+		if options != nil && options.Timeout != 0 {
+			timeoutInSeconds = options.Timeout
+		}
+		var cncl context.CancelFunc
+		ctx, cncl = context.WithTimeout(context.Background(), time.Second*time.Duration(timeoutInSeconds))
+		defer cncl()
 	}
 
-	ctx, cncl := context.WithTimeout(context.Background(), time.Second*time.Duration(timeoutInSeconds))
-	defer cncl()
-
 	// create new request
 	req, err := http.NewRequestWithContext(ctx, method, u, bytes.NewBufferString(bodyString))
 	if err != nil {

go/rtl/auth_test.go

@@ -635,6 +635,65 @@ func TestAuthSession_Do_Timeout(t *testing.T) {
 			t.Errorf("Do() call did not error with context.DeadlineExceeded, got=%v", err)
 		}
 	})
+
+	t.Run("Do() follows Context set in AuthSession config", func(t *testing.T) {
+		mux := http.NewServeMux()
+		setupApi40Login(mux, foreverValidTestToken, http.StatusOK)
+		server := httptest.NewServer(mux)
+		defer server.Close()
+
+		mux.HandleFunc("/api"+apiVersion+path, func(w http.ResponseWriter, r *http.Request) {
+			time.Sleep(4 * time.Second)
+		})
+
+		ctx, cncl := context.WithTimeout(context.Background(), 1*time.Second)
+		defer cncl()
+
+		session := NewAuthSession(ApiSettings{
+			BaseUrl:    server.URL,
+			ApiVersion: apiVersion,
+			Context:    ctx,
+		})
+
+		err := session.Do(nil, "GET", apiVersion, path, nil, nil, nil)
+
+		if err == nil {
+			t.Errorf("Do() call did not error/timeout")
+		} else if !errors.Is(err, context.DeadlineExceeded) {
+			t.Errorf("Do() call did not error with context.DeadlineExceeded, got=%v", err)
+		}
+	})
+
+	t.Run("Do() follows Context set in Do()'s options", func(t *testing.T) {
+		mux := http.NewServeMux()
+		setupApi40Login(mux, foreverValidTestToken, http.StatusOK)
+		server := httptest.NewServer(mux)
+		defer server.Close()
+
+		mux.HandleFunc("/api"+apiVersion+path, func(w http.ResponseWriter, r *http.Request) {
+			time.Sleep(4 * time.Second)
+		})
+
+		ctx, cncl := context.WithTimeout(context.Background(), 1*time.Second)
+		defer cncl()
+
+		session := NewAuthSession(ApiSettings{
+			BaseUrl:    server.URL,
+			ApiVersion: apiVersion,
+		})
+
+		options := ApiSettings{
+			Context: ctx,
+		}
+
+		err := session.Do(nil, "GET", apiVersion, path, nil, nil, &options)
+
+		if err == nil {
+			t.Errorf("Do() call did not error/timeout")
+		} else if !errors.Is(err, context.DeadlineExceeded) {
+			t.Errorf("Do() call did not error with context.DeadlineExceeded, got=%v", err)
+		}
+	})
 }
 
 func TestSetQuery(t *testing.T) {

go/rtl/settings.go

@@ -1,11 +1,13 @@
 package rtl
 
 import (
+	"context"
 	"fmt"
-	"gopkg.in/ini.v1"
 	"os"
 	"strconv"
 	"strings"
+
+	"gopkg.in/ini.v1"
 )
 
 var defaultSectionName string = "Looker"
@@ -20,6 +22,7 @@ type ApiSettings struct {
 	ClientSecret string `ini:"client_secret"`
 	ApiVersion   string `ini:"api_version"`
 	Headers      map[string]string
+	Context      context.Context
 }
 
 var defaultSettings ApiSettings = ApiSettings{