From b18ef7d45a9ba0c3a28fd57851d71eb7606a6f21 Mon Sep 17 00:00:00 2001
From: Samuel Sonning <samuel.sonning@gmail.com>
Date: Wed, 9 Mar 2022 09:45:36 +0100
Subject: [PATCH 1/2] make num messages configurable

---
 examples/long_lived.rs         | 2 +-
 examples/serde.rs              | 2 +-
 examples/singleshot.rs         | 2 +-
 examples/subscribe_to_topic.rs | 2 +-
 src/subscription.rs            | 3 ++-
 5 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/examples/long_lived.rs b/examples/long_lived.rs
index ba04809..61030fb 100644
--- a/examples/long_lived.rs
+++ b/examples/long_lived.rs
@@ -26,7 +26,7 @@ impl FromPubSubMessage for UpdatePacket {
 fn schedule_pubsub_pull(subscription: Arc<Subscription>) {
     task::spawn(async move {
         while subscription.client().is_running() {
-            match subscription.get_messages::<UpdatePacket>().await {
+            match subscription.get_messages::<UpdatePacket>(100).await {
                 Ok(messages) => {
                     for (result, ack_id) in messages {
                         match result {
diff --git a/examples/serde.rs b/examples/serde.rs
index d51b5df..a7f7491 100644
--- a/examples/serde.rs
+++ b/examples/serde.rs
@@ -40,7 +40,7 @@ async fn main() {
     };
 
     let order_sub = Arc::new(pubsub.subscribe(config.pubsub_subscription));
-    match order_sub.clone().get_messages::<UpdatePacket>().await {
+    match order_sub.clone().get_messages::<UpdatePacket>(100).await {
         Ok(packets) => {
             for packet in &packets {
                 println!("Received: {:?}", packet);
diff --git a/examples/singleshot.rs b/examples/singleshot.rs
index 2d677a2..2e83953 100644
--- a/examples/singleshot.rs
+++ b/examples/singleshot.rs
@@ -37,7 +37,7 @@ async fn main() {
     };
 
     let subscription = Arc::new(pubsub.subscribe(config.pubsub_subscription));
-    match subscription.get_messages::<UpdatePacket>().await {
+    match subscription.get_messages::<UpdatePacket>(100).await {
         Ok(messages) => {
             for (result, ack_id) in messages {
                 match result {
diff --git a/examples/subscribe_to_topic.rs b/examples/subscribe_to_topic.rs
index 2862376..1d45c90 100644
--- a/examples/subscribe_to_topic.rs
+++ b/examples/subscribe_to_topic.rs
@@ -35,7 +35,7 @@ async fn main() {
     println!("Subscribed to topic with: {}", sub.name);
     let packets = sub
         .clone()
-        .get_messages::<UpdatePacket>()
+        .get_messages::<UpdatePacket>(100)
         .await
         .expect("Error Checking PubSub");
 
diff --git a/src/subscription.rs b/src/subscription.rs
index 8a23537..1f18a4e 100644
--- a/src/subscription.rs
+++ b/src/subscription.rs
@@ -59,6 +59,7 @@ impl Subscription {
 
     pub async fn get_messages<T: FromPubSubMessage>(
         &self,
+        max_messages: u16,
     ) -> Result<Vec<(Result<T, error::Error>, String)>, error::Error> {
         let client = self
             .client
@@ -69,7 +70,7 @@ impl Subscription {
             .parse()
             .unwrap();
 
-        let json = r#"{"maxMessages": 100}"#;
+        let json = format!("{{\"maxMessages:\": {}}}", max_messages);
 
         let mut req = client.request(Method::POST, json);
         *req.uri_mut() = uri.clone();

From 07dd737dd72c03851b207aef5c5539aa48d04189 Mon Sep 17 00:00:00 2001
From: Samuel Sonning <samuel.sonning@gmail.com>
Date: Wed, 9 Mar 2022 10:53:38 +0100
Subject: [PATCH 2/2] fix

---
 src/subscription.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/subscription.rs b/src/subscription.rs
index 1f18a4e..1b17268 100644
--- a/src/subscription.rs
+++ b/src/subscription.rs
@@ -70,7 +70,7 @@ impl Subscription {
             .parse()
             .unwrap();
 
-        let json = format!("{{\"maxMessages:\": {}}}", max_messages);
+        let json = format!("{{\"maxMessages\": {}}}", max_messages);
 
         let mut req = client.request(Method::POST, json);
         *req.uri_mut() = uri.clone();