refactor: const Duration

bytebeamio · Jun 11, 2024 · 79d3821 · 79d3821
1 parent fa48079
commit 79d3821
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git a/uplink/src/collector/device_shadow.rs b/uplink/src/collector/device_shadow.rs
@@ -1,5 +1,3 @@
-use std::time::Duration;
-
 use log::{error, trace};
 use serde::Serialize;
 
@@ -49,8 +47,7 @@ impl DeviceShadow {
         let ping_addr = "8.8.8.8".parse().unwrap();
         let ping_payload = [0; 64];
 
-        let mut device_shadow_interval =
-            tokio::time::interval(Duration::from_secs(self.config.interval));
+        let mut device_shadow_interval = tokio::time::interval(self.config.interval);
 
         loop {
             _ = device_shadow_interval.tick().await;

diff --git a/uplink/src/config.rs b/uplink/src/config.rs
@@ -13,11 +13,11 @@ use crate::collector::journalctl::JournalCtlConfig;
 #[cfg(target_os = "android")]
 use crate::collector::logcat::LogcatConfig;
 
-pub const DEFAULT_TIMEOUT: u64 = 60;
+pub const DEFAULT_TIMEOUT: Duration = Duration::from_secs(60);
 
 #[inline]
 fn default_timeout() -> Duration {
-    Duration::from_secs(DEFAULT_TIMEOUT)
+    DEFAULT_TIMEOUT
 }
 
 #[inline]
@@ -227,7 +227,7 @@ impl From<&ActionRoute> for ActionRoute {
 
 #[derive(Clone, Debug, Deserialize)]
 pub struct DeviceShadowConfig {
-    pub interval: u64,
+    pub interval: Duration,
 }
 
 impl Default for DeviceShadowConfig {