increase gcp upload timeout (#612)

livekit · Feb 15, 2024 · bfc460d · bfc460d
1 parent 10a2cbc
commit bfc460d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pkg/pipeline/sink/uploader/gcp.go b/pkg/pipeline/sink/uploader/gcp.go
@@ -30,6 +30,8 @@ import (
 	"github.com/livekit/protocol/livekit"
 )
 
+const gcpTimeout = time.Minute
+
 type GCPUploader struct {
 	conf   *livekit.GCPUpload
 	client *storage.Client
@@ -73,7 +75,7 @@ func (u *GCPUploader) upload(localFilepath, storageFilepath string, _ types.Outp
 	var ctx context.Context
 	if stat.Size() <= googleapi.DefaultUploadChunkSize {
 		var cancel context.CancelFunc
-		ctx, cancel = context.WithTimeout(context.Background(), time.Second*32)
+		ctx, cancel = context.WithTimeout(context.Background(), gcpTimeout)
 		defer cancel()
 	} else {
 		ctx = context.Background()
@@ -87,6 +89,7 @@ func (u *GCPUploader) upload(localFilepath, storageFilepath string, _ types.Outp
 		}),
 		storage.WithPolicy(storage.RetryAlways),
 	).NewWriter(ctx)
+	wc.ChunkRetryDeadline = gcpTimeout
 
 	if _, err = io.Copy(wc, file); err != nil {
 		return "", 0, wrap("GCP", err)