implement remove

jpicht · Mar 14, 2022 · ea726af · ea726af
1 parent 9b5e98d
commit ea726af
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 2 deletions.
diff --git a/pkg/azcat/remove.go b/pkg/azcat/remove.go
@@ -0,0 +1,21 @@
+package azcat
+
+import (
+ "context"
+
+ "github.com/Azure/azure-sdk-for-go/sdk/storage/azblob"
+ "github.com/sirupsen/logrus"
+)
+
+func Remove(containerName, blobName string, client *azblob.ServiceClient) {
+ log.WithFields(logrus.Fields{
+ "container": containerName,
+ "blob": blobName,
+ }).Debug("remove")
+ containerClient := client.NewContainerClient(containerName)
+ blobClient := containerClient.NewBlobClient(blobName)
+ _, err := blobClient.Delete(context.TODO(), nil)
+ if err != nil {
+ log.WithError(err).Fatal("Delete failed")
+ }
+}
diff --git a/pkg/azcat/run.go b/pkg/azcat/run.go
@@ -10,6 +10,8 @@ func Run(mode Mode, bloburl azblob.BlobURLParts, client *azblob.ServiceClient) {
  Read(bloburl.ContainerName, bloburl.BlobName, client)
  case EMode.Write():
  Write(bloburl.ContainerName, bloburl.BlobName, client)
+ case EMode.Remove():
+ Remove(bloburl.ContainerName, bloburl.BlobName, client)
  default:
  log.WithField("mode", mode.String()).Fatal("Not implemented")
  }

diff --git a/pkg/azcat/write.go b/pkg/azcat/write.go
@@ -15,9 +15,9 @@ func Write(containerName, blobName string, client *azblob.ServiceClient) {
  }).Debug("write")
  containerClient := client.NewContainerClient(containerName)
  blobClient := containerClient.NewBlockBlobClient(blobName)
- _, err := blobClient.UploadStreamToBlockBlob(context.TODO(), os.Stdin, azblob.UploadStreamToBlockBlobOptions{})
 
+ _, err := blobClient.UploadStreamToBlockBlob(context.TODO(), os.Stdin, azblob.UploadStreamToBlockBlobOptions{})
  if err != nil {
- log.WithError(err).Fatal("Download failed")
+ log.WithError(err).Fatal("Write failed")
  }
 }