Allow setting custom properties in service_bus send_message() (Azure#1758)

* Allow setting custom properties in service_bus send_message()

* Format corrections.

Author: bryanpendleton

sdk/messaging_servicebus/src/service_bus/mod.rs

@@ -15,11 +15,7 @@ use azure_core::{
     CollectedResponse, HttpClient, Method, Request, StatusCode, Url,
 };
 use serde::{Deserialize, Serialize};
-use std::{
-    str::FromStr,
-    time::Duration,
-    {ops::Add, sync::Arc},
-};
+use std::{collections::HashMap, ops::Add, str::FromStr, sync::Arc, time::Duration};
 use time::OffsetDateTime;
 use url::form_urlencoded::{self, Serializer};
 
@@ -315,6 +311,7 @@ impl FromStr for BrokerProperties {
 pub struct SendMessageOptions {
     pub content_type: Option<String>,
     pub broker_properties: Option<SettableBrokerProperties>,
+    pub custom_properties: Option<HashMap<String, String>>,
 }
 
 impl headers::AsHeaders for SendMessageOptions {
@@ -334,6 +331,14 @@ impl headers::AsHeaders for SendMessageOptions {
             ));
         }
 
+        if let Some(custom_properties) = &self.custom_properties {
+            headers.extend(
+                custom_properties
+                    .iter()
+                    .map(|(k, v)| (k.to_owned().into(), v.into())),
+            );
+        }
+
         headers.into_iter()
     }
 }

sdk/messaging_servicebus/tests/service_bus.rs

@@ -50,6 +50,27 @@ async fn send_message_with_content_type() {
         .expect("Failed to send message");
 }
 
+#[tokio::test]
+async fn send_message_with_custom_property_test() {
+    let client = create_client().unwrap();
+    client
+        .send_message(
+            "hello, world!",
+            Some(SendMessageOptions {
+                custom_properties: Some(
+                    std::collections::HashMap::from([(
+                        "custom_property_key".into(),
+                        "custom_property_value".into(),
+                    )])
+                    .into(),
+                ),
+                ..Default::default()
+            }),
+        )
+        .await
+        .expect("Failed to send message");
+}
+
 #[tokio::test]
 async fn receive_and_delete_message_test() {
     let client = create_client().unwrap();