github.com/YoungNK/go-ethereum@v1.9.7/internal/build/azure.go (about) 1 // Copyright 2016 The go-ethereum Authors 2 // This file is part of the go-ethereum library. 3 // 4 // The go-ethereum library is free software: you can redistribute it and/or modify 5 // it under the terms of the GNU Lesser General Public License as published by 6 // the Free Software Foundation, either version 3 of the License, or 7 // (at your option) any later version. 8 // 9 // The go-ethereum library is distributed in the hope that it will be useful, 10 // but WITHOUT ANY WARRANTY; without even the implied warranty of 11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 12 // GNU Lesser General Public License for more details. 13 // 14 // You should have received a copy of the GNU Lesser General Public License 15 // along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>. 16 17 package build 18 19 import ( 20 "context" 21 "fmt" 22 "net/url" 23 "os" 24 25 "github.com/Azure/azure-storage-blob-go/azblob" 26 ) 27 28 // AzureBlobstoreConfig is an authentication and configuration struct containing 29 // the data needed by the Azure SDK to interact with a speicifc container in the 30 // blobstore. 31 type AzureBlobstoreConfig struct { 32 Account string // Account name to authorize API requests with 33 Token string // Access token for the above account 34 Container string // Blob container to upload files into 35 } 36 37 // AzureBlobstoreUpload uploads a local file to the Azure Blob Storage. Note, this 38 // method assumes a max file size of 64MB (Azure limitation). Larger files will 39 // need a multi API call approach implemented. 40 // 41 // See: https://msdn.microsoft.com/en-us/library/azure/dd179451.aspx#Anchor_3 42 func AzureBlobstoreUpload(path string, name string, config AzureBlobstoreConfig) error { 43 if *DryRunFlag { 44 fmt.Printf("would upload %q to %s/%s/%s\n", path, config.Account, config.Container, name) 45 return nil 46 } 47 // Create an authenticated client against the Azure cloud 48 credential, err := azblob.NewSharedKeyCredential(config.Account, config.Token) 49 if err != nil { 50 return err 51 } 52 53 pipeline := azblob.NewPipeline(credential, azblob.PipelineOptions{}) 54 55 u, _ := url.Parse(fmt.Sprintf("https://%s.blob.core.windows.net", config.Account)) 56 service := azblob.NewServiceURL(*u, pipeline) 57 58 container := service.NewContainerURL(config.Container) 59 blockblob := container.NewBlockBlobURL(name) 60 61 // Stream the file to upload into the designated blobstore container 62 in, err := os.Open(path) 63 if err != nil { 64 return err 65 } 66 defer in.Close() 67 68 _, err = blockblob.Upload(context.Background(), in, azblob.BlobHTTPHeaders{}, azblob.Metadata{}, azblob.BlobAccessConditions{}) 69 return err 70 } 71 72 // AzureBlobstoreList lists all the files contained within an azure blobstore. 73 func AzureBlobstoreList(config AzureBlobstoreConfig) ([]azblob.BlobItem, error) { 74 credential, err := azblob.NewSharedKeyCredential(config.Account, config.Token) 75 if err != nil { 76 return nil, err 77 } 78 79 pipeline := azblob.NewPipeline(credential, azblob.PipelineOptions{}) 80 81 u, _ := url.Parse(fmt.Sprintf("https://%s.blob.core.windows.net", config.Account)) 82 service := azblob.NewServiceURL(*u, pipeline) 83 84 // List all the blobs from the container and return them 85 container := service.NewContainerURL(config.Container) 86 87 res, err := container.ListBlobsFlatSegment(context.Background(), azblob.Marker{}, azblob.ListBlobsSegmentOptions{ 88 MaxResults: 1024 * 1024 * 1024, // Yes, fetch all of them 89 }) 90 if err != nil { 91 return nil, err 92 } 93 return res.Segment.BlobItems, nil 94 } 95 96 // AzureBlobstoreDelete iterates over a list of files to delete and removes them 97 // from the blobstore. 98 func AzureBlobstoreDelete(config AzureBlobstoreConfig, blobs []azblob.BlobItem) error { 99 if *DryRunFlag { 100 for _, blob := range blobs { 101 fmt.Printf("would delete %s (%s) from %s/%s\n", blob.Name, blob.Properties.LastModified, config.Account, config.Container) 102 } 103 return nil 104 } 105 // Create an authenticated client against the Azure cloud 106 credential, err := azblob.NewSharedKeyCredential(config.Account, config.Token) 107 if err != nil { 108 return err 109 } 110 111 pipeline := azblob.NewPipeline(credential, azblob.PipelineOptions{}) 112 113 u, _ := url.Parse(fmt.Sprintf("https://%s.blob.core.windows.net", config.Account)) 114 service := azblob.NewServiceURL(*u, pipeline) 115 116 container := service.NewContainerURL(config.Container) 117 118 // Iterate over the blobs and delete them 119 for _, blob := range blobs { 120 blockblob := container.NewBlockBlobURL(blob.Name) 121 if _, err := blockblob.Delete(context.Background(), azblob.DeleteSnapshotsOptionInclude, azblob.BlobAccessConditions{}); err != nil { 122 return err 123 } 124 } 125 return nil 126 }