add configurable browser timeout

tedpearson · Oct 11, 2022 · a88f435 · a88f435
1 parent 7998642
commit a88f435
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/internal/app/browser.go b/internal/app/browser.go
@@ -11,9 +11,13 @@ import (
 )
 
 func DownloadCsv(config *Config, startDate string, endDate string) (string, error) {
+	timeout, err := time.ParseDuration(config.Timeout)
+	if err != nil {
+		return "", err
+	}
 	ctx, cancel := chromedp.NewContext(context.Background(), chromedp.WithLogf(log.Printf))
 	defer cancel()
-	ctx, cancel = context.WithTimeout(ctx, 30*time.Second)
+	ctx, cancel = context.WithTimeout(ctx, timeout)
 	defer cancel()
 
 	done := make(chan string, 1)
@@ -29,7 +33,7 @@ func DownloadCsv(config *Config, startDate string, endDate string) (string, erro
 		}
 	})
 
-	err := chromedp.Run(ctx,
+	err = chromedp.Run(ctx,
 		chromedp.Navigate(config.LoginUrl),
 		chromedp.SetValue("#LoginUsernameTextBox", config.Username, chromedp.NodeVisible),
 		chromedp.SetValue("#LoginPasswordTextBox", config.Password),

diff --git a/internal/app/main.go b/internal/app/main.go
@@ -27,6 +27,7 @@ type Config struct {
 	LoginUrl    string
 	DownloadDir string
 	ExtractDays int
+	Timeout     string
 	InfluxDB    InfluxDB
 }