FilesExpand file tree

 master

/

BlobClientWrapper.cs

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

92 lines (85 loc) · 4.29 KB

 master

/

BlobClientWrapper.cs

Copy path

Top

File metadata and controls

• Code
• Blame

92 lines (85 loc) · 4.29 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

using Azure.Storage.Blobs;

using Azure.Storage.Blobs.Models;

using Azure.Storage.Blobs.Specialized;

using Medallion.Threading.Internal;

using System;

using System.Collections.Generic;

using System.IO;

using System.Text;

using System.Threading;

using System.Threading.Tasks;

namespace Medallion.Threading.Azure

{

/// <summary>

/// Adds <see cref="SyncViaAsync"/> support to <see cref="BlobBaseClient"/>

/// </summary>

internal class BlobClientWrapper

{

private readonly BlobBaseClient _blobClient;

public BlobClientWrapper(BlobBaseClient blobClient)

{

this._blobClient = blobClient;

}

public string Name => this._blobClient.Name;

public BlobLeaseClientWrapper GetBlobLeaseClient() => new BlobLeaseClientWrapper(this._blobClient.GetBlobLeaseClient());

public async ValueTask<IDictionary<string, string>> GetMetadataAsync(string leaseId, CancellationToken cancellationToken)

{

var conditions = new BlobRequestConditions { LeaseId = leaseId };

var properties = SyncViaAsync.IsSynchronous

? this._blobClient.GetProperties(conditions, cancellationToken)

: await this._blobClient.GetPropertiesAsync(conditions, cancellationToken).ConfigureAwait(false);

return properties.Value.Metadata;

}

public ValueTask CreateIfNotExistsAsync(IDictionary<string, string> metadata, CancellationToken cancellationToken)

{

switch (this._blobClient)

{

case BlobClient blobClient:

if (SyncViaAsync.IsSynchronous)

{

blobClient.Upload(Stream.Null, metadata: metadata, cancellationToken: cancellationToken);

return default;

}

return new ValueTask(blobClient.UploadAsync(Stream.Null, metadata: metadata, cancellationToken: cancellationToken));

case BlockBlobClient blockBlobClient:

if (SyncViaAsync.IsSynchronous)

{

blockBlobClient.Upload(Stream.Null, metadata: metadata, cancellationToken: cancellationToken);

return default;

}

return new ValueTask(blockBlobClient.UploadAsync(Stream.Null, metadata: metadata, cancellationToken: cancellationToken));

case PageBlobClient pageBlobClient:

if (SyncViaAsync.IsSynchronous)

{

pageBlobClient.CreateIfNotExists(size: 0, metadata: metadata, cancellationToken: cancellationToken);

return default;

}

return new ValueTask(pageBlobClient.CreateIfNotExistsAsync(size: 0, metadata: metadata, cancellationToken: cancellationToken));

case AppendBlobClient appendBlobClient:

if (SyncViaAsync.IsSynchronous)

{

appendBlobClient.CreateIfNotExists(metadata: metadata, cancellationToken: cancellationToken);

return default;

}

return new ValueTask(appendBlobClient.CreateIfNotExistsAsync(metadata: metadata, cancellationToken: cancellationToken));

default:

throw new InvalidOperationException(

this._blobClient.GetType() == typeof(BlobBaseClient)

? $"Unable to create a lock blob given client type {typeof(BlobBaseClient)}. Either ensure that the blob exists or use a non-base client type such as {typeof(BlobClient)}"

+ " which specifies the type of blob to create"

: $"Unexpected blob client type {this._blobClient.GetType()}"

);

}

}

public ValueTask DeleteIfExistsAsync(string? leaseId = null)

{

var conditions = leaseId != null ? new BlobRequestConditions { LeaseId = leaseId } : null;

if (SyncViaAsync.IsSynchronous)

{

this._blobClient.DeleteIfExists(conditions: conditions);

return default;

}

return new ValueTask(this._blobClient.DeleteIfExistsAsync(conditions: conditions));

}

}

}