[Storage] get_account_info all clients, set/get_tags for BlobClient, set_properties for BlobServiceClient (#2795)

Author: vincenttran-msft

sdk/storage/azure_storage_blob/assets.json

@@ -1,6 +1,6 @@
 {
   "AssetsRepo": "Azure/azure-sdk-assets",
   "AssetsRepoPrefixPath": "rust",
-  "Tag": "rust/azure_storage_blob_873d66a4ea",
+  "Tag": "rust/azure_storage_blob_cb9b90a35f",
   "TagPrefix": "rust/azure_storage_blob"
 }

sdk/storage/azure_storage_blob/src/clients/append_blob_client.rs

@@ -55,9 +55,9 @@ impl AppendBlobClient {
 
         let client = GeneratedAppendBlobClient::new(
             endpoint,
-            credential.clone(),
-            container_name.clone(),
-            blob_name.clone(),
+            credential,
+            container_name,
+            blob_name,
             Some(options),
         )?;
         Ok(Self {

sdk/storage/azure_storage_blob/src/clients/blob_client.rs

@@ -5,16 +5,18 @@ use crate::{
     generated::clients::BlobClient as GeneratedBlobClient,
     generated::models::{
         BlobClientAcquireLeaseResult, BlobClientBreakLeaseResult, BlobClientChangeLeaseResult,
-        BlobClientDownloadResult, BlobClientGetPropertiesResult, BlobClientReleaseLeaseResult,
-        BlobClientRenewLeaseResult, BlockBlobClientCommitBlockListResult,
-        BlockBlobClientStageBlockResult, BlockBlobClientUploadResult,
+        BlobClientDownloadResult, BlobClientGetAccountInfoResult, BlobClientGetPropertiesResult,
+        BlobClientReleaseLeaseResult, BlobClientRenewLeaseResult,
+        BlockBlobClientCommitBlockListResult, BlockBlobClientStageBlockResult,
+        BlockBlobClientUploadResult,
     },
     models::{
         AccessTier, BlobClientAcquireLeaseOptions, BlobClientBreakLeaseOptions,
         BlobClientChangeLeaseOptions, BlobClientDeleteOptions, BlobClientDownloadOptions,
-        BlobClientGetPropertiesOptions, BlobClientReleaseLeaseOptions, BlobClientRenewLeaseOptions,
-        BlobClientSetMetadataOptions, BlobClientSetPropertiesOptions, BlobClientSetTierOptions,
-        BlockBlobClientCommitBlockListOptions, BlockBlobClientUploadOptions, BlockList,
+        BlobClientGetAccountInfoOptions, BlobClientGetPropertiesOptions, BlobClientGetTagsOptions,
+        BlobClientReleaseLeaseOptions, BlobClientRenewLeaseOptions, BlobClientSetMetadataOptions,
+        BlobClientSetPropertiesOptions, BlobClientSetTagsOptions, BlobClientSetTierOptions,
+        BlobTags, BlockBlobClientCommitBlockListOptions, BlockBlobClientUploadOptions, BlockList,
         BlockListType, BlockLookupList,
     },
     pipeline::StorageHeadersPolicy,
@@ -24,10 +26,11 @@ use azure_core::{
     credentials::TokenCredential,
     http::{
         policies::{BearerTokenCredentialPolicy, Policy},
-        NoFormat, RequestContent, Response, Url, XmlFormat,
+        JsonFormat, NoFormat, RequestContent, Response, Url, XmlFormat,
     },
     Bytes, Result,
 };
+use std::collections::HashMap;
 use std::sync::Arc;
 
 /// A client to interact with a specific Azure storage blob, although that blob may not yet exist.
@@ -63,9 +66,9 @@ impl BlobClient {
 
         let client = GeneratedBlobClient::new(
             endpoint,
-            credential.clone(),
-            container_name.clone(),
-            blob_name.clone(),
+            credential,
+            container_name,
+            blob_name,
             Some(options),
         )?;
         Ok(Self {
@@ -304,4 +307,51 @@ impl BlobClient {
     ) -> Result<Response<BlobClientRenewLeaseResult, NoFormat>> {
         self.client.renew_lease(lease_id, options).await
     }
+
+    /// Sets tags on a blob. Note that each call to this operation replaces all existing tags. To remove
+    /// all tags from the blob, call this operation with no tags specified.
+    ///
+    /// # Arguments
+    ///
+    /// * `tags` - Name-value pairs associated with the blob as tag. Tags are case-sensitive.
+    ///   The tag set may contain at most 10 tags.  Tag keys must be between 1 and 128 characters,
+    ///   and tag values must be between 0 and 256 characters.
+    ///   Valid tag key and value characters include: lowercase and uppercase letters, digits (0-9),
+    ///   space (' '), plus (+), minus (-), period (.), solidus (/), colon (:), equals (=), underscore (_)
+    /// * `options` - Optional configuration for the request.
+    pub async fn set_tags(
+        &self,
+        tags: HashMap<String, String>,
+        options: Option<BlobClientSetTagsOptions<'_>>,
+    ) -> Result<Response<(), NoFormat>> {
+        let blob_tags: BlobTags = tags.into();
+        self.client
+            .set_tags(RequestContent::try_from(blob_tags)?, options)
+            .await
+    }
+
+    /// Gets the tags on a blob.
+    ///
+    /// # Arguments
+    ///
+    /// * `options` - Optional configuration for the request.
+    pub async fn get_tags(
+        &self,
+        options: Option<BlobClientGetTagsOptions<'_>>,
+    ) -> Result<Response<BlobTags, XmlFormat>> {
+        self.client.get_tags(options).await
+    }
+
+    /// Gets information related to the Storage account in which the blob resides.
+    /// This includes the `sku_name` and `account_kind`.
+    ///
+    /// # Arguments
+    ///
+    /// * `options` - Optional configuration for the request.
+    pub async fn get_account_info(
+        &self,
+        options: Option<BlobClientGetAccountInfoOptions<'_>>,
+    ) -> Result<Response<BlobClientGetAccountInfoResult, NoFormat>> {
+        self.client.get_account_info(options).await
+    }
 }

sdk/storage/azure_storage_blob/src/clients/blob_container_client.rs

@@ -5,16 +5,17 @@ use crate::{
     generated::clients::BlobContainerClient as GeneratedBlobContainerClient,
     generated::models::{
         BlobContainerClientAcquireLeaseResult, BlobContainerClientBreakLeaseResult,
-        BlobContainerClientChangeLeaseResult, BlobContainerClientGetPropertiesResult,
-        BlobContainerClientReleaseLeaseResult, BlobContainerClientRenewLeaseResult,
+        BlobContainerClientChangeLeaseResult, BlobContainerClientGetAccountInfoResult,
+        BlobContainerClientGetPropertiesResult, BlobContainerClientReleaseLeaseResult,
+        BlobContainerClientRenewLeaseResult,
     },
     models::{
         BlobContainerClientAcquireLeaseOptions, BlobContainerClientBreakLeaseOptions,
         BlobContainerClientChangeLeaseOptions, BlobContainerClientCreateOptions,
-        BlobContainerClientDeleteOptions, BlobContainerClientGetPropertiesOptions,
-        BlobContainerClientListBlobFlatSegmentOptions, BlobContainerClientReleaseLeaseOptions,
-        BlobContainerClientRenewLeaseOptions, BlobContainerClientSetMetadataOptions,
-        ListBlobsFlatSegmentResponse,
+        BlobContainerClientDeleteOptions, BlobContainerClientGetAccountInfoOptions,
+        BlobContainerClientGetPropertiesOptions, BlobContainerClientListBlobFlatSegmentOptions,
+        BlobContainerClientReleaseLeaseOptions, BlobContainerClientRenewLeaseOptions,
+        BlobContainerClientSetMetadataOptions, ListBlobsFlatSegmentResponse,
     },
     pipeline::StorageHeadersPolicy,
     BlobClient, BlobContainerClientOptions,
@@ -233,4 +234,17 @@ impl BlobContainerClient {
     ) -> Result<Response<BlobContainerClientRenewLeaseResult, NoFormat>> {
         self.client.renew_lease(lease_id, options).await
     }
+
+    /// Gets information related to the Storage account in which the container resides.
+    /// This includes the `sku_name` and `account_kind`.
+    ///
+    /// # Arguments
+    ///
+    /// * `options` - Optional configuration for the request.
+    pub async fn get_account_info(
+        &self,
+        options: Option<BlobContainerClientGetAccountInfoOptions<'_>>,
+    ) -> Result<Response<BlobContainerClientGetAccountInfoResult, NoFormat>> {
+        self.client.get_account_info(options).await
+    }
 }

sdk/storage/azure_storage_blob/src/clients/blob_service_client.rs

@@ -3,8 +3,10 @@
 
 use crate::{
     generated::clients::BlobServiceClient as GeneratedBlobServiceClient,
+    generated::models::BlobServiceClientGetAccountInfoResult,
     models::{
-        BlobServiceClientGetPropertiesOptions, BlobServiceClientListContainersSegmentOptions,
+        BlobServiceClientGetAccountInfoOptions, BlobServiceClientGetPropertiesOptions,
+        BlobServiceClientListContainersSegmentOptions, BlobServiceClientSetPropertiesOptions,
         ListContainersSegmentResponse, StorageServiceProperties,
     },
     pipeline::StorageHeadersPolicy,
@@ -14,7 +16,7 @@ use azure_core::{
     credentials::TokenCredential,
     http::{
         policies::{BearerTokenCredentialPolicy, Policy},
-        PageIterator, Response, Url, XmlFormat,
+        NoFormat, PageIterator, RequestContent, Response, Url, XmlFormat,
     },
     Result,
 };
@@ -95,4 +97,33 @@ impl BlobServiceClient {
     ) -> Result<PageIterator<Response<ListContainersSegmentResponse, XmlFormat>>> {
         self.client.list_containers_segment(options)
     }
+
+    /// Sets properties for a Storage account's Blob service endpoint, including properties for Storage Analytics and CORS rules.
+    ///
+    /// # Arguments
+    ///
+    /// * `storage_service_properties` - The Storage service properties to set.
+    /// * `options` - Optional configuration for the request.
+    pub async fn set_properties(
+        &self,
+        storage_service_properties: RequestContent<StorageServiceProperties, XmlFormat>,
+        options: Option<BlobServiceClientSetPropertiesOptions<'_>>,
+    ) -> Result<Response<(), NoFormat>> {
+        self.client
+            .set_properties(storage_service_properties, options)
+            .await
+    }
+
+    /// Gets information related to the Storage account.
+    /// This includes the `sku_name` and `account_kind`.
+    ///
+    /// # Arguments
+    ///
+    /// * `options` - Optional configuration for the request.
+    pub async fn get_account_info(
+        &self,
+        options: Option<BlobServiceClientGetAccountInfoOptions<'_>>,
+    ) -> Result<Response<BlobServiceClientGetAccountInfoResult, NoFormat>> {
+        self.client.get_account_info(options).await
+    }
 }

sdk/storage/azure_storage_blob/src/clients/block_blob_client.rs

@@ -61,9 +61,9 @@ impl BlockBlobClient {
 
         let client = GeneratedBlockBlobClient::new(
             endpoint,
-            credential.clone(),
-            container_name.clone(),
-            blob_name.clone(),
+            credential,
+            container_name,
+            blob_name,
             Some(options),
         )?;
         Ok(Self {

sdk/storage/azure_storage_blob/src/clients/page_blob_client.rs

@@ -58,9 +58,9 @@ impl PageBlobClient {
 
         let client = GeneratedPageBlobClient::new(
             endpoint,
-            credential.clone(),
-            container_name.clone(),
-            blob_name.clone(),
+            credential,
+            container_name,
+            blob_name,
             Some(options),
         )?;
         Ok(Self {