Set segment write timeout

livepeer · Jan 25, 2024 · 2d73713 · 2d73713
1 parent 58cf59f
commit 2d73713
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/core/uploader.go b/core/uploader.go
@@ -30,6 +30,8 @@ func UploadRetryBackoff() backoff.BackOff {
 	return backoff.WithMaxRetries(newExponentialBackOffExecutor(), 3)
 }
 
+const segmentWriteTimeout = 5 * time.Minute
+
 func Upload(input io.Reader, outputURI string, waitBetweenWrites, writeTimeout time.Duration) error {
 	storageDriver, err := drivers.ParseOSURL(outputURI, true)
 	if err != nil {
@@ -60,7 +62,7 @@ func Upload(input io.Reader, outputURI string, waitBetweenWrites, writeTimeout t
 		}
 
 		err = backoff.Retry(func() error {
-			_, err := session.SaveData(context.Background(), "", bytes.NewReader(fileContents), fields, writeTimeout)
+			_, err := session.SaveData(context.Background(), "", bytes.NewReader(fileContents), fields, segmentWriteTimeout)
 			if err != nil {
 				glog.Errorf("failed upload attempt: %v", err)
 			}