forks/go-ethereum
1
0
Fork 1
mirror of https://github.com/ethereum/go-ethereum.git synced 2026-02-26 15:47:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

internal/build: add timeouts and unify context usage in Azure blob helpers

Browse source
This commit is contained in:
radik878 2025-10-02 15:59:48 +03:00 committed by GitHub
parent fc8c8c1314
commit a28e25c63f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 11 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
internal/build/azure.go
View file

@ -20,6 +20,7 @@ import (
"context"
"fmt"
"os"
"time"
"github.com/Azure/azure-sdk-for-go/sdk/storage/azblob"
"github.com/Azure/azure-sdk-for-go/sdk/storage/azblob/container"
@ -61,7 +62,9 @@ func AzureBlobstoreUpload(path string, name string, config AzureBlobstoreConfig)
}
defer in.Close()
_, err = client.UploadFile(context.Background(), config.Container, name, in, nil)
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Minute)
defer cancel()
_, err = client.UploadFile(ctx, config.Container, name, in, nil)
return err
}
@ -80,8 +83,10 @@ func AzureBlobstoreList(config AzureBlobstoreConfig) ([]*container.BlobItem, err
pager := client.NewListBlobsFlatPager(config.Container, nil)
var blobs []*container.BlobItem
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Minute)
defer cancel()
for pager.More() {
page, err := pager.NextPage(context.TODO())
page, err := pager.NextPage(ctx)
if err != nil {
return nil, err
}
@ -111,9 +116,12 @@ func AzureBlobstoreDelete(config AzureBlobstoreConfig, blobs []*container.BlobIt
}
// Iterate over the blobs and delete them
for _, blob := range blobs {
if _, err := client.DeleteBlob(context.Background(), config.Container, *blob.Name, nil); err != nil {
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Minute)
if _, err := client.DeleteBlob(ctx, config.Container, *blob.Name, nil); err != nil {
cancel()
return err
}
cancel()
fmt.Printf("deleted %s (%s)\n", *blob.Name, blob.Properties.LastModified)
}
return nil