Useless Option<HeaderMap>

nats-io · Sep 20, 2023 · 17dfb59 · 17dfb59
1 parent 98fd93a
commit 17dfb59
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 56 deletions.
diff --git a/async-nats/src/client.rs b/async-nats/src/client.rs
@@ -150,15 +150,8 @@ impl Client {
     /// # }
     /// ```
     pub async fn publish(&self, subject: String, payload: Bytes) -> Result<(), PublishError> {
-        self.sender
-            .send(Command::Publish {
-                subject,
-                payload,
-                respond: None,
-                headers: None,
-            })
-            .await?;
-        Ok(())
+        self.publish_with_headers(subject, HeaderMap::new(), payload)
+            .await
     }
 
     /// Publish a [Message] with headers to a given subject.
@@ -191,7 +184,7 @@ impl Client {
                 subject,
                 payload,
                 respond: None,
-                headers: Some(headers),
+                headers,
             })
             .await?;
         Ok(())
@@ -223,15 +216,8 @@ impl Client {
         reply: String,
         payload: Bytes,
     ) -> Result<(), PublishError> {
-        self.sender
-            .send(Command::Publish {
-                subject,
-                payload,
-                respond: Some(reply),
-                headers: None,
-            })
-            .await?;
-        Ok(())
+        self.publish_with_reply_and_headers(subject, reply, HeaderMap::new(), payload)
+            .await
     }
 
     /// Publish a [Message] to a given subject with headers and specified response subject
@@ -269,7 +255,7 @@ impl Client {
                 subject,
                 payload,
                 respond: Some(reply),
-                headers: Some(headers),
+                headers,
             })
             .await?;
         Ok(())
@@ -338,14 +324,9 @@ impl Client {
         if let Some(inbox) = request.inbox {
             let timeout = request.timeout.unwrap_or(self.request_timeout);
             let mut sub = self.subscribe(inbox.clone()).await?;
-            let payload: Bytes = request.payload.unwrap_or_else(Bytes::new);
-            match request.headers {
-                Some(headers) => {
-                    self.publish_with_reply_and_headers(subject, inbox, headers, payload)
-                        .await?
-                }
-                None => self.publish_with_reply(subject, inbox, payload).await?,
-            }
+            self.publish_with_reply_and_headers(subject, inbox, request.headers, request.payload)
+                .await?;
+
             let request = match timeout {
                 Some(timeout) => {
                     tokio::time::timeout(timeout, sub.next())
@@ -372,9 +353,9 @@ impl Client {
         } else {
             let (sender, receiver) = oneshot::channel();
 
-            let payload = request.payload.unwrap_or_else(Bytes::new);
             let respond = self.new_inbox();
             let headers = request.headers;
+            let payload = request.payload;
 
             self.sender
                 .send(Command::Request {
@@ -542,8 +523,8 @@ impl Client {
 /// Used for building customized requests.
 #[derive(Default)]
 pub struct Request {
-    payload: Option<Bytes>,
-    headers: Option<HeaderMap>,
+    payload: Bytes,
+    headers: HeaderMap,
     timeout: Option<Option<Duration>>,
     inbox: Option<String>,
 }
@@ -566,7 +547,7 @@ impl Request {
     /// # }
     /// ```
     pub fn payload(mut self, payload: Bytes) -> Request {
-        self.payload = Some(payload);
+        self.payload = payload;
         self
     }
 
@@ -591,7 +572,7 @@ impl Request {
     /// # }
     /// ```
     pub fn headers(mut self, headers: HeaderMap) -> Request {
-        self.headers = Some(headers);
+        self.headers = headers;
         self
     }
 

diff --git a/async-nats/src/connection.rs b/async-nats/src/connection.rs
@@ -431,30 +431,24 @@ impl Connection {
                 respond,
                 headers,
             } => {
-                let verb = match headers.as_ref() {
-                    Some(headers) if !headers.is_empty() => "HPUB",
-                    _ => "PUB",
-                };
+                let verb = if !headers.is_empty() { "HPUB" } else { "PUB" };
 
                 small_write!("{verb} {subject} ");
 
                 if let Some(respond) = respond {
                     small_write!("{respond} ");
                 }
 
-                match headers {
-                    Some(headers) if !headers.is_empty() => {
-                        let headers = headers.to_bytes();
+                if !headers.is_empty() {
+                    let headers = headers.to_bytes();
 
-                        let headers_len = headers.len();
-                        let total_len = headers_len + payload.len();
-                        small_write!("{headers_len} {total_len}\r\n");
-                        self.write(headers);
-                    }
-                    _ => {
-                        let payload_len = payload.len();
-                        small_write!("{payload_len}\r\n");
-                    }
+                    let headers_len = headers.len();
+                    let total_len = headers_len + payload.len();
+                    small_write!("{headers_len} {total_len}\r\n");
+                    self.write(headers);
+                } else {
+                    let payload_len = payload.len();
+                    small_write!("{payload_len}\r\n");
                 }
 
                 self.write(Bytes::clone(payload));
@@ -958,7 +952,7 @@ mod write_op {
                     subject: "FOO.BAR".into(),
                     payload: "Hello World".into(),
                     respond: None,
-                    headers: None,
+                    headers: HeaderMap::new(),
                 }]
                 .iter(),
             )
@@ -977,7 +971,7 @@ mod write_op {
                     subject: "FOO.BAR".into(),
                     payload: "Hello World".into(),
                     respond: Some("INBOX.67".into()),
-                    headers: None,
+                    headers: HeaderMap::new(),
                 }]
                 .iter(),
             )
@@ -995,10 +989,10 @@ mod write_op {
                     subject: "FOO.BAR".into(),
                     payload: "Hello World".into(),
                     respond: Some("INBOX.67".into()),
-                    headers: Some(HeaderMap::from_iter([(
+                    headers: HeaderMap::from_iter([(
                         "Header".parse().unwrap(),
                         "X".parse().unwrap(),
-                    )])),
+                    )]),
                 }]
                 .iter(),
             )

diff --git a/async-nats/src/lib.rs b/async-nats/src/lib.rs
@@ -268,13 +268,13 @@ pub(crate) enum Command {
         subject: String,
         payload: Bytes,
         respond: Option<String>,
-        headers: Option<HeaderMap>,
+        headers: HeaderMap,
     },
     Request {
         subject: String,
         payload: Bytes,
         respond: String,
-        headers: Option<HeaderMap>,
+        headers: HeaderMap,
         sender: oneshot::Sender<Message>,
     },
     Subscribe {
@@ -299,7 +299,7 @@ pub(crate) enum ClientOp {
         subject: String,
         payload: Bytes,
         respond: Option<String>,
-        headers: Option<HeaderMap>,
+        headers: HeaderMap,
     },
     Subscribe {
         sid: u64,