cloudevents · sbcd90 · Nov 8, 2020 · Nov 9, 2020 · Nov 30, 2020 · Dec 22, 2020
diff --git a/Cargo.toml b/Cargo.toml
@@ -45,11 +45,13 @@ members = [
   "cloudevents-sdk-actix-web",
   "cloudevents-sdk-reqwest",
   "cloudevents-sdk-rdkafka",
-  "cloudevents-sdk-warp"
+  "cloudevents-sdk-warp",
+  "cloudevents-sdk-paho-mqtt"
 ]
 exclude = [
   "example-projects/actix-web-example",
   "example-projects/reqwest-wasm-example",
   "example-projects/rdkafka-example",
   "example-projects/warp-example",
+  "example-projects/paho-mqtt-example"
 ]
diff --git a/cloudevents-sdk-paho-mqtt/Cargo.toml b/cloudevents-sdk-paho-mqtt/Cargo.toml
@@ -0,0 +1,18 @@
+[package]
+name = "cloudevents-sdk-paho-mqtt"
+version = "0.3.0"
+authors = ["Francesco Guardiani <[email protected]>"]
+license-file = "../LICENSE"
+edition = "2018"
+description = "CloudEvents official Rust SDK - Mqtt integration"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
+cloudevents-sdk = { version = "0.3.0", path = ".." }
+lazy_static = "1.4.0"
+paho-mqtt = "0.8"
+chrono = { version = "^0.4", features = ["serde"] }
+
+[dev-dependencies]
+serde_json = "^1.0"
diff --git a/cloudevents-sdk-paho-mqtt/README.md b/cloudevents-sdk-paho-mqtt/README.md
@@ -0,0 +1,42 @@
+# CloudEvents SDK Rust - paho-mqtt [![Crates badge]][crates.io] [![Docs badge]][docs.rs]
+
+Integration of [CloudEvents SDK](https://github.com/cloudevents/sdk-rust/) with [paho-mqtt](https://www.eclipse.org/paho/).
+
+Look at [CloudEvents SDK README](https://github.com/cloudevents/sdk-rust/) for more info.
+
+## Development & Contributing
+
+If you're interested in contributing to sdk-rust, look at [Contributing documentation](../CONTRIBUTING.md)
+
+## Community
+
+## Sample usage
+
+- Check the example [paho-mqtt-example](../example-projects/paho-mqtt-example)
+
+### MQTT V3
+- Start the MQTT V3 Consumer
+
+```
+run --package <package-name> --bin <binary-name> -- --mode consumerV3 --broker tcp://localhost:1883 --topic test
+```
+
+- Start the MQTT V3 Producer
+
+```
+run --package <package-name> --bin <binary-name> -- --broker tcp://localhost:1883 --topic test --mode producerV3
+```
+
+### MQTT V5
+- Start the MQTT V5 Consumer
+
+```
+run --package <package-name> --bin <binary-name> -- --mode consumerV5 --broker tcp://localhost:1883 --topic test
+```
+
+- Start the MQTT V5 Producer
+
+```
+run --package <package-name> --bin <binary-name> -- --broker tcp://localhost:1883 --topic test --mode producerV5
+```
+
diff --git a/cloudevents-sdk-paho-mqtt/src/headers.rs b/cloudevents-sdk-paho-mqtt/src/headers.rs
@@ -0,0 +1,34 @@
+use cloudevents::event::SpecVersion;
+use lazy_static::lazy_static;
+use std::collections::HashMap;
+
+macro_rules! attribute_name_to_header {
+    ($attribute:expr) => {
+        format!("ce_{}", $attribute)
+    };
+}
+
+fn attributes_to_headers(it: impl Iterator<Item = &'static str>) -> HashMap<&'static str, String> {
+    it.map(|s| {
+        if s == "datacontenttype" {
+            (s, String::from("content-type"))
+        } else {
+            (s, attribute_name_to_header!(s))
+        }
+    })
+    .collect()
+}
+
+lazy_static! {
+    pub(crate) static ref ATTRIBUTES_TO_MQTT_HEADERS: HashMap<&'static str, String> =
+        attributes_to_headers(SpecVersion::all_attribute_names());
+}
+
+pub(crate) static SPEC_VERSION_HEADER: &'static str = "ce_specversion";
+pub(crate) static CLOUDEVENTS_JSON_HEADER: &'static str = "application/cloudevents+json";
+pub(crate) static CONTENT_TYPE: &'static str = "content-type";
+
+pub enum MqttVersion {
+    MQTT_3,
+    MQTT_5,
+}
diff --git a/cloudevents-sdk-paho-mqtt/src/lib.rs b/cloudevents-sdk-paho-mqtt/src/lib.rs
@@ -0,0 +1,14 @@
+//! This library provides Mqtt protocol bindings for CloudEvents
+//! using the [paho.mqtt.rust](https://github.com/eclipse/paho.mqtt.rust) library.\\
+#[macro_use]
+mod headers;
+mod mqtt_consumer_record;
+mod mqtt_producer_record;
+
+pub use mqtt_consumer_record::record_to_event;
+pub use mqtt_consumer_record::ConsumerMessageDeserializer;
+pub use mqtt_consumer_record::MessageExt;
+
+pub use headers::MqttVersion;
+pub use mqtt_producer_record::MessageBuilderExt;
+pub use mqtt_producer_record::MessageRecord;
diff --git a/cloudevents-sdk-paho-mqtt/src/mqtt_consumer_record.rs b/cloudevents-sdk-paho-mqtt/src/mqtt_consumer_record.rs
@@ -0,0 +1,179 @@
+use crate::headers;
+use cloudevents::event::SpecVersion;
+use cloudevents::message::{
+    BinaryDeserializer, BinarySerializer, Encoding, MessageAttributeValue, MessageDeserializer,
+    Result, StructuredDeserializer, StructuredSerializer,
+};
+use cloudevents::{message, Event};
+use paho_mqtt::{Message, Properties, PropertyCode};
+use std::convert::TryFrom;
+
+pub struct ConsumerMessageDeserializer<'a> {
+    pub(crate) headers: &'a Properties,
+    pub(crate) payload: Option<Vec<u8>>,
+}
+
+impl<'a> ConsumerMessageDeserializer<'a> {
+    fn get_mqtt_headers(message: &Message) -> &Properties {
+        message.properties()
+    }
+
+    pub fn new(message: &Message) -> Result<ConsumerMessageDeserializer> {
+        Ok(ConsumerMessageDeserializer {
+            headers: Self::get_mqtt_headers(message),
+            payload: Some(message.payload()).map(|s| Vec::from(s)),
+        })
+    }
+}
+
+impl<'a> BinaryDeserializer for ConsumerMessageDeserializer<'a> {
+    fn deserialize_binary<R: Sized, V: BinarySerializer<R>>(self, mut visitor: V) -> Result<R> {
+        if self.encoding() != Encoding::BINARY {
+            return Err(message::Error::WrongEncoding {});
+        }
+
+        let spec_version = SpecVersion::try_from(
+            self.headers
+                .find_user_property(headers::SPEC_VERSION_HEADER)
+                .unwrap()
+                .as_str(),
+        )?;
+
+        visitor = visitor.set_spec_version(spec_version.clone())?;
+
+        let attributes = spec_version.attribute_names();
+
+        if let Some(hv) = self.headers.find_user_property(headers::CONTENT_TYPE) {
+            visitor = visitor.set_attribute("datacontenttype", MessageAttributeValue::String(hv))?
+        }
+
+        for (hn, hv) in self
+            .headers
+            .user_iter()
+            .filter(|(hn, _)| headers::SPEC_VERSION_HEADER != *hn && hn.starts_with("ce_"))
+        {
+            let name = &hn["ce_".len()..];
+
+            if attributes.contains(&name) {
+                visitor = visitor.set_attribute(name, MessageAttributeValue::String(hv))?
+            } else {
+                visitor = visitor.set_extension(name, MessageAttributeValue::String(hv))?
+            }
+        }
+
+        if self.payload != None {
+            visitor.end_with_data(self.payload.unwrap())
+        } else {
+            visitor.end()
+        }
+    }
+}
+
+impl<'a> StructuredDeserializer for ConsumerMessageDeserializer<'a> {
+    fn deserialize_structured<R: Sized, V: StructuredSerializer<R>>(self, visitor: V) -> Result<R> {
+        visitor.set_structured_event(self.payload.unwrap())
+    }
+}
+
+impl<'a> MessageDeserializer for ConsumerMessageDeserializer<'a> {
+    fn encoding(&self) -> Encoding {
+        match self.headers.iter(PropertyCode::UserProperty).count() == 0 {
+            true => Encoding::STRUCTURED,
+            false => Encoding::BINARY,
+        }
+    }
+}
+
+pub fn record_to_event(msg: &Message) -> Result<Event> {
+    MessageDeserializer::into_event(ConsumerMessageDeserializer::new(msg)?)
+}
+
+pub trait MessageExt {
+    fn to_event(&self) -> Result<Event>;
+}
+
+impl MessageExt for Message {
+    fn to_event(&self) -> Result<Event> {
+        record_to_event(self)
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    use crate::headers::MqttVersion::{MQTT_3, MQTT_5};
+    use crate::MessageBuilderExt;
+    use chrono::Utc;
+    use cloudevents::event::Data;
+    use cloudevents::{EventBuilder, EventBuilderV10};
+    use paho_mqtt::MessageBuilder;
+    use serde_json::json;
+
+    #[test]
+    fn test_binary_record() {
+        let time = Utc::now();
+
+        let expected = EventBuilderV10::new()
+            .id("0001")
+            .ty("example.test")
+            .time(time)
+            .source("http://localhost")
+            .data(
+                "application/json",
+                Data::Binary(String::from("{\"hello\":\"world\"}").into_bytes()),
+            )
+            .extension("someint", "10")
+            .build()
+            .unwrap();
+
+        let event = EventBuilderV10::new()
+            .id("0001")
+            .ty("example.test")
+            .time(time)
+            .source("http://localhost")
+            .extension("someint", "10")
+            .data("application/json", json!({"hello": "world"}))
+            .build()
+            .unwrap();
+
+        let msg = MessageBuilder::new()
+            .topic("test")
+            .event(event, MQTT_5)
+            .qos(1)
+            .finalize();
+
+        assert_eq!(msg.to_event().unwrap(), expected)
+    }
+
+    #[test]
+    fn test_structured_record() {
+        let j = json!({"hello": "world"});
+
+        let expected = EventBuilderV10::new()
+            .id("0001")
+            .ty("example.test")
+            .source("http://localhost")
+            .data("application/cloudevents+json", j.clone())
+            .extension("someint", "10")
+            .build()
+            .unwrap();
+
+        let input = EventBuilderV10::new()
+            .id("0001")
+            .ty("example.test")
+            .source("http://localhost")
+            .data("application/cloudevents+json", j.clone())
+            .extension("someint", "10")
+            .build()
+            .unwrap();
+
+        let msg = MessageBuilder::new()
+            .topic("test")
+            .event(input, MQTT_3)
+            .qos(1)
+            .finalize();
+
+        assert_eq!(msg.to_event().unwrap(), expected)
+    }
+}