Merge pull request #2 from Overmuse/SR/error_context

add anyhow context for errors

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "reporting"
-version = "0.1.2"
+version = "0.1.3"
 authors = ["Sebastian Rollen <[email protected]>"]
 edition = "2018"
 

src/main.rs

@@ -1,5 +1,5 @@
 use alpaca::AlpacaMessage;
-use anyhow::Result;
+use anyhow::{Context, Result};
 use dogstatsd::Client;
 use futures::StreamExt;
 use kafka_settings::consumer;
@@ -20,20 +20,29 @@ async fn main() -> Result<()> {
         .finish();
     set_global_default(subscriber).expect("Failed to set subscriber");
     info!("Starting reporting");
-    let settings = settings::Settings::new()?;
-    let consumer = consumer(&settings.kafka)?;
+    let settings = settings::Settings::new().context("Failed to load settings")?;
+    let consumer = consumer(&settings.kafka).context("Failed to create kafka consumer")?;
     debug!("Creating dogstatsd client");
-    let client = Client::new("0.0.0.0:0", &settings.app.target_address).await?;
+    let client = Client::new("127.0.0.1:0", &settings.app.target_address)
+        .await
+        .context("Failed to create dogstatsd client")?;
     while let Some(message) = consumer.stream().next().await {
-        let message = message?;
+        let message = message.context("Error from kafka")?;
         if let Some(payload) = message.payload() {
             trace!("Received payload");
-            let alpaca_message: AlpacaMessage = serde_json::from_slice(payload)?;
+            let alpaca_message: AlpacaMessage =
+                serde_json::from_slice(payload).context("Failed to deserialize alpaca message")?;
             if let AlpacaMessage::TradeUpdates(order_update) = alpaca_message {
                 debug!("Received trade update, generating metrics");
-                let metrics = order_update.order.to_metrics()?;
+                let metrics = order_update
+                    .order
+                    .to_metrics()
+                    .context("Failed to convert order to metrics")?;
                 for metric in metrics {
-                    client.send(metric).await?;
+                    client
+                        .send(metric)
+                        .await
+                        .context("Failed to send metrics")?;
                 }
             }
         }

src/types.rs

@@ -1,5 +1,5 @@
 use alpaca::Order;
-use anyhow::Result;
+use anyhow::{Context, Result};
 use dogstatsd::Metric;
 
 pub trait ToMetrics {
@@ -9,10 +9,16 @@ pub trait ToMetrics {
 impl ToMetrics for Order {
     fn to_metrics(&self) -> Result<Vec<Metric>> {
         let metrics = vec![Metric::increase("order")
-            .add_key_value("status", serde_plain::to_string(&self.status)?)
+            .add_key_value(
+                "status",
+                serde_plain::to_string(&self.status).context("Failed to serialize order status")?,
+            )
             .add_key_value("id", &self.client_order_id)
             .add_key_value("ticker", &self.symbol)
-            .add_key_value("side", serde_plain::to_string(&self.side)?)];
+            .add_key_value(
+                "side",
+                serde_plain::to_string(&self.side).context("Failed to serialize order side")?,
+            )];
         Ok(metrics)
     }
 }