Add missing conditional headers to blob and container operations (#1280)

Author: JWT95

sdk/storage_blobs/src/blob/operations/acquire_lease.rs

@@ -7,7 +7,10 @@ operation! {
     client: BlobClient,
     lease_duration: LeaseDuration,
     ?lease_id: LeaseId,
-    ?proposed_lease_id: ProposedLeaseId
+    ?proposed_lease_id: ProposedLeaseId,
+    ?if_modified_since: IfModifiedSinceCondition,
+    ?if_match: IfMatchCondition,
+    ?if_tags: IfTags
 }
 
 impl AcquireLeaseBuilder {
@@ -22,6 +25,9 @@ impl AcquireLeaseBuilder {
             headers.add(self.lease_duration);
             headers.add(self.proposed_lease_id);
             headers.add(self.lease_id);
+            headers.add(self.if_modified_since);
+            headers.add(self.if_match);
+            headers.add(self.if_tags);
 
             let mut request =
                 self.client

sdk/storage_blobs/src/blob/operations/change_lease.rs

@@ -6,6 +6,9 @@ operation! {
     ChangeLease,
     client: BlobLeaseClient,
     proposed_lease_id: ProposedLeaseId,
+    ?if_modified_since: IfModifiedSinceCondition,
+    ?if_match: IfMatchCondition,
+    ?if_tags: IfTags
 }
 
 impl ChangeLeaseBuilder {
@@ -19,6 +22,9 @@ impl ChangeLeaseBuilder {
             headers.insert(LEASE_ACTION, "change");
             headers.add(self.client.lease_id());
             headers.add(self.proposed_lease_id);
+            headers.add(self.if_modified_since);
+            headers.add(self.if_match);
+            headers.add(self.if_tags);
 
             let mut request =
                 self.client

sdk/storage_blobs/src/blob/operations/get_page_ranges.rs

@@ -6,7 +6,7 @@ use time::OffsetDateTime;
 operation! {
     GetPageRanges,
     client: BlobClient,
-        ?if_modified_since: IfModifiedSinceCondition,
+    ?if_modified_since: IfModifiedSinceCondition,
     ?if_match: IfMatchCondition,
     ?if_tags: IfTags,
     ?blob_versioning: BlobVersioning,

sdk/storage_blobs/src/blob/operations/put_block_blob.rs

@@ -16,7 +16,10 @@ operation! {
     ?access_tier: AccessTier,
     ?tags: Tags,
     ?lease_id: LeaseId,
-    ?encryption_scope: EncryptionScope
+    ?encryption_scope: EncryptionScope,
+    ?if_modified_since: IfModifiedSinceCondition,
+    ?if_match: IfMatchCondition,
+    ?if_tags: IfTags
 }
 
 impl PutBlockBlobBuilder {
@@ -40,6 +43,9 @@ impl PutBlockBlobBuilder {
             headers.add(self.access_tier);
             headers.add(self.lease_id);
             headers.add(self.encryption_scope);
+            headers.add(self.if_modified_since);
+            headers.add(self.if_match);
+            headers.add(self.if_tags);
 
             let mut request = self.client.finalize_request(
                 url,

sdk/storage_blobs/src/blob/operations/put_block_list.rs

@@ -16,7 +16,10 @@ operation! {
     ?metadata: Metadata,
     ?access_tier: AccessTier,
     ?tags: Tags,
-    ?lease_id: LeaseId
+    ?lease_id: LeaseId,
+    ?if_modified_since: IfModifiedSinceCondition,
+    ?if_match: IfMatchCondition,
+    ?if_tags: IfTags
 }
 
 impl PutBlockListBuilder {
@@ -51,6 +54,9 @@ impl PutBlockListBuilder {
             }
             headers.add(self.access_tier);
             headers.add(self.lease_id);
+            headers.add(self.if_modified_since);
+            headers.add(self.if_match);
+            headers.add(self.if_tags);
 
             let mut request = self.client.finalize_request(
                 url,

sdk/storage_blobs/src/blob/operations/release_lease.rs

@@ -1,10 +1,13 @@
 use crate::prelude::*;
-use azure_core::{headers::*, RequestId};
+use azure_core::{headers::*, prelude::*, RequestId};
 use time::OffsetDateTime;
 
 operation! {
     ReleaseLease,
     client: BlobLeaseClient,
+    ?if_modified_since: IfModifiedSinceCondition,
+    ?if_match: IfMatchCondition,
+    ?if_tags: IfTags
 }
 
 impl ReleaseLeaseBuilder {
@@ -17,6 +20,9 @@ impl ReleaseLeaseBuilder {
             let mut headers = Headers::new();
             headers.insert(LEASE_ACTION, "release");
             headers.add(self.client.lease_id());
+            headers.add(self.if_modified_since);
+            headers.add(self.if_match);
+            headers.add(self.if_tags);
 
             let mut request =
                 self.client

sdk/storage_blobs/src/blob/operations/renew_lease.rs

@@ -5,6 +5,9 @@ use time::OffsetDateTime;
 operation! {
     RenewLease,
     client: BlobLeaseClient,
+    ?if_modified_since: IfModifiedSinceCondition,
+    ?if_match: IfMatchCondition,
+    ?if_tags: IfTags
 }
 
 impl RenewLeaseBuilder {
@@ -17,6 +20,9 @@ impl RenewLeaseBuilder {
             let mut headers = Headers::new();
             headers.insert(LEASE_ACTION, "renew");
             headers.add(self.client.lease_id());
+            headers.add(self.if_modified_since);
+            headers.add(self.if_match);
+            headers.add(self.if_tags);
 
             let mut request =
                 self.client

sdk/storage_blobs/src/blob/operations/set_blob_tier.rs

@@ -7,7 +7,8 @@ operation! {
     client: BlobClient,
     access_tier: AccessTier,
     ?rehydrate_priority: RehydratePriority,
-    ?blob_versioning: BlobVersioning
+    ?blob_versioning: BlobVersioning,
+    ?if_tags: IfTags
 }
 
 impl SetBlobTierBuilder {
@@ -23,6 +24,7 @@ impl SetBlobTierBuilder {
                 self.rehydrate_priority
                     .unwrap_or(RehydratePriority::Standard),
             );
+            headers.add(self.if_tags);
 
             let mut request =
                 self.client

sdk/storage_blobs/src/blob/operations/snapshot_blob.rs

@@ -14,6 +14,7 @@ operation! {
     ?metadata: Metadata,
     ?if_modified_since: IfModifiedSinceCondition,
     ?if_match: IfMatchCondition,
+    ?if_tags: IfTags,
     ?lease_id: LeaseId
 }
 
@@ -28,6 +29,7 @@ impl SnapshotBlobBuilder {
             headers.add(self.lease_id);
             headers.add(self.if_modified_since);
             headers.add(self.if_match);
+            headers.add(self.if_tags);
             if let Some(metadata) = &self.metadata {
                 for m in metadata.iter() {
                     headers.add(m);

sdk/storage_blobs/src/container/operations/break_lease.rs

@@ -7,7 +7,8 @@ operation! {
     BreakLease,
     client: ContainerClient,
     ?lease_break_period: LeaseBreakPeriod,
-    ?lease_id: LeaseId
+    ?lease_id: LeaseId,
+    ?if_modified_since: IfModifiedSinceCondition
 }
 
 impl BreakLeaseBuilder {
@@ -22,6 +23,7 @@ impl BreakLeaseBuilder {
             headers.insert(LEASE_ACTION, "break");
             headers.add(self.lease_id);
             headers.add(self.lease_break_period);
+            headers.add(self.if_modified_since);
 
             let mut request = self
                 .client