From 4a383cd3b838bd5d55f978d22a25b9be646c9853 Mon Sep 17 00:00:00 2001
From: minghuaw <michael.wu1107@gmail.com>
Date: Thu, 22 Feb 2024 17:04:50 -0800
Subject: [PATCH] added new example resume sender

---
 examples/event_hubs/src/bin/resume_sender.rs | 81 ++++++++++++++++++++
 examples/event_hubs/src/bin/simple_sender.rs | 23 ------
 2 files changed, 81 insertions(+), 23 deletions(-)
 create mode 100644 examples/event_hubs/src/bin/resume_sender.rs

diff --git a/examples/event_hubs/src/bin/resume_sender.rs b/examples/event_hubs/src/bin/resume_sender.rs
new file mode 100644
index 00000000..5168a98f
--- /dev/null
+++ b/examples/event_hubs/src/bin/resume_sender.rs
@@ -0,0 +1,81 @@
+use std::env;
+
+use dotenv::dotenv;
+use fe2o3_amqp::{
+    sasl_profile::SaslProfile,
+    types::{
+        messaging::{Message, Properties},
+        primitives::Binary,
+    },
+    Connection, Sender, Session,
+};
+
+#[tokio::main]
+async fn main() {
+    dotenv().ok();
+    env_logger::init();
+
+    let hostname = env::var("HOST_NAME").unwrap();
+    let sa_key_name = env::var("SHARED_ACCESS_KEY_NAME").unwrap();
+    let sa_key_value = env::var("SHARED_ACCESS_KEY_VALUE").unwrap();
+    let event_hub_name = env::var("EVENT_HUB_NAME").unwrap();
+
+    let url = format!("amqps://{}", hostname);
+    let mut connection = Connection::builder()
+        .container_id("rust-connection-1")
+        .alt_tls_establishment(true) // EventHubs uses alternative TLS establishment
+        .sasl_profile(SaslProfile::Plain {
+            username: sa_key_name,
+            password: sa_key_value,
+        })
+        .open(&url[..])
+        .await
+        .unwrap();
+    let mut session = Session::begin(&mut connection).await.unwrap();
+    let mut sender = Sender::attach(&mut session, "rust-simple-sender", event_hub_name)
+        .await
+        .unwrap();
+
+    // Message will be randomly distributed to different partitions
+    for i in 0..3 {
+        // All of the Microsoft AMQP clients represent the event body as an uninterpreted bag of bytes.
+        // A message builder can be used to specify the type of body section
+        let data = format!("Message {}", i).into_bytes();
+        let message = Message::builder()
+            .properties(
+                Properties::builder()
+                    .group_id(String::from("send_to_event_hub"))
+                    .build(),
+            )
+            .data(Binary::from(data))
+            .build();
+        let outcome = sender.send(message).await.unwrap();
+        outcome.accepted_or_else(|outcome| outcome).unwrap();
+    }
+
+    let detached = sender.detach().await.unwrap();
+
+    // Close the old session and create a new one
+    session.close().await.unwrap();
+    let mut session = Session::begin(&mut connection).await.unwrap();
+
+    let mut sender = detached.resume_on_session(&session).await.unwrap();
+
+    for i in 10..13 {
+        let data = format!("Message {}", i).into_bytes();
+        let message = Message::builder()
+            .properties(
+                Properties::builder()
+                    .group_id(String::from("send_to_event_hub"))
+                    .build(),
+            )
+            .data(Binary::from(data))
+            .build();
+        let outcome = sender.send(message).await.unwrap();
+        outcome.accepted_or_else(|outcome| outcome).unwrap();
+    }
+
+    sender.close().await.unwrap();
+    session.end().await.unwrap();
+    connection.close().await.unwrap();
+}
diff --git a/examples/event_hubs/src/bin/simple_sender.rs b/examples/event_hubs/src/bin/simple_sender.rs
index 5168a98f..5e7e4d60 100644
--- a/examples/event_hubs/src/bin/simple_sender.rs
+++ b/examples/event_hubs/src/bin/simple_sender.rs
@@ -13,7 +13,6 @@ use fe2o3_amqp::{
 #[tokio::main]
 async fn main() {
     dotenv().ok();
-    env_logger::init();
 
     let hostname = env::var("HOST_NAME").unwrap();
     let sa_key_name = env::var("SHARED_ACCESS_KEY_NAME").unwrap();
@@ -53,28 +52,6 @@ async fn main() {
         outcome.accepted_or_else(|outcome| outcome).unwrap();
     }
 
-    let detached = sender.detach().await.unwrap();
-
-    // Close the old session and create a new one
-    session.close().await.unwrap();
-    let mut session = Session::begin(&mut connection).await.unwrap();
-
-    let mut sender = detached.resume_on_session(&session).await.unwrap();
-
-    for i in 10..13 {
-        let data = format!("Message {}", i).into_bytes();
-        let message = Message::builder()
-            .properties(
-                Properties::builder()
-                    .group_id(String::from("send_to_event_hub"))
-                    .build(),
-            )
-            .data(Binary::from(data))
-            .build();
-        let outcome = sender.send(message).await.unwrap();
-        outcome.accepted_or_else(|outcome| outcome).unwrap();
-    }
-
     sender.close().await.unwrap();
     session.end().await.unwrap();
     connection.close().await.unwrap();