Merge pull request kayleg#20 from BigJerBD/feature/add-attributes-to-publish

feature: Add attributes to published EncodedMessage

Author: kayleg

src/message.rs

@@ -1,10 +1,13 @@
 use crate::error;
 use base64;
 use serde_derive::{Deserialize, Serialize};
+use std::collections::HashMap;
 
 #[derive(Deserialize, Clone, Serialize)]
 pub struct EncodedMessage {
     data: String,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    attributes: Option<HashMap<String, String>>,
 }
 
 pub trait FromPubSubMessage
@@ -19,10 +22,10 @@ impl EncodedMessage {
         base64::decode(&self.data)
     }
 
-    pub fn new<T: serde::Serialize>(data: &T) -> Self {
+    pub fn new<T: serde::Serialize>(data: &T, attributes: Option<HashMap<String, String>>) -> Self {
         let json = serde_json::to_string(data).unwrap();
         let data = base64::encode(&json);
-        EncodedMessage { data }
+        EncodedMessage { data, attributes }
     }
 }
 

src/topic.rs

@@ -61,14 +61,20 @@ impl Topic {
     pub async fn publish<T: serde::Serialize>(
         &self,
         data: T,
+    ) -> Result<PublishMessageResponse, error::Error> {
+        self.publish_message(EncodedMessage::new(&data, None)).await
+    }
+
+    pub async fn publish_message(
+        &self,
+        message: EncodedMessage,
     ) -> Result<PublishMessageResponse, error::Error> {
         let uri: hyper::Uri = format!("{}/v1/{}:publish", *PUBSUB_HOST, self.name)
             .parse()
             .unwrap();
 
-        let new_message = EncodedMessage::new(&data);
         let payload = PublishMessageRequest {
-            messages: vec![new_message],
+            messages: vec![message],
         };
 
         self.perform_request::<PublishMessageRequest, PublishMessageResponse>(