feat: deserialize duration from milliseconds as a u64 in toml

book/src/sinks/gcp_pubsub.md

@@ -9,6 +9,12 @@ A sink that sends each event as a message to a PubSub topic. Each event is json-
 type = "GcpPubSub"
 credentials = "oura-test-347101-ff3f7b2d69cc.json"
 topic = "test"
+
+[sink.retry_policy]
+max_retries = 30
+backoff_unit =  5000
+backoff_factor = 2
+max_backoff = 100000
 ```
 
 ### Section: `sink`

book/src/sinks/webhook.md

@@ -12,9 +12,13 @@ type = "Webhook"
 url = "https://endpoint:5000/events"
 authorization = "user:pass"
 timeout = 30000
-error_policy = "Retry"
+error_policy = "Continue"
+
+[sink.retry_policy]
 max_retries = 30
-backoff_delay = 5000
+backoff_unit =  5000
+backoff_factor = 2
+max_backoff = 100000
 
 [sink.headers]
 extra_header_1 = "abc"

src/utils/retry.rs

@@ -1,15 +1,26 @@
 use std::{fmt::Debug, ops::Mul, time::Duration};
 
-use serde::Deserialize;
+use serde::{Deserialize, Deserializer};
 
 #[derive(Debug, Deserialize, Default, Copy, Clone)]
 pub struct Policy {
     pub max_retries: u32,
+    #[serde(deserialize_with = "deserialize_duration")]
     pub backoff_unit: Duration,
     pub backoff_factor: u32,
+    #[serde(deserialize_with = "deserialize_duration")]
     pub max_backoff: Duration,
 }
 
+fn deserialize_duration<'de, D>(deserializer: D) -> Result<Duration, D::Error>
+where
+    D: Deserializer<'de>,
+{
+    let millis = u64::deserialize(deserializer)?;
+
+    Ok(Duration::from_millis(millis))
+}
+
 fn compute_backoff_delay(policy: &Policy, retry: u32) -> Duration {
     let units = policy.backoff_factor.pow(retry);
     let backoff = policy.backoff_unit.mul(units);