From 7c6f8b9082cfc9d0446b6c927d4477d3d8a8161b Mon Sep 17 00:00:00 2001
From: Ben Bangert <ben@groovie.org>
Date: Thu, 26 Apr 2018 17:37:05 -0700
Subject: [PATCH] feat: use chrono epoch times instead of monotonic time

Add's a currently unused ns_since_epoch for future use.

Closes #1180
---
 autopush_rs/Cargo.toml      |  2 +-
 autopush_rs/src/client.rs   | 13 ++++++-------
 autopush_rs/src/util/mod.rs | 18 ++++++++++++++++++
 3 files changed, 25 insertions(+), 8 deletions(-)

diff --git a/autopush_rs/Cargo.toml b/autopush_rs/Cargo.toml
index 3c2e32b0..1b1ad0a6 100644
--- a/autopush_rs/Cargo.toml
+++ b/autopush_rs/Cargo.toml
@@ -9,7 +9,7 @@ crate-type = ["cdylib"]
 [dependencies]
 bytes = "0.4.6"
 cadence = "0.13.2"
-chrono = "0.4.1"
+chrono = "0.4.2"
 env_logger = { version = "0.5.6", default-features = false }
 error-chain = "0.11.0"
 futures = "0.1.21"
diff --git a/autopush_rs/src/client.rs b/autopush_rs/src/client.rs
index 5cd31349..c7255dee 100644
--- a/autopush_rs/src/client.rs
+++ b/autopush_rs/src/client.rs
@@ -20,7 +20,6 @@ use rusoto_dynamodb::{AttributeValue, DynamoDb};
 use rusoto_dynamodb::{UpdateItemError, UpdateItemInput, UpdateItemOutput};
 use state_machine_future::RentToOwn;
 use tokio_core::reactor::Timeout;
-use time;
 use uuid::Uuid;
 use woothee::parser::Parser;
 
@@ -28,7 +27,7 @@ use call;
 use errors::*;
 use protocol::{ClientMessage, Notification, ServerMessage, ServerNotification};
 use server::Server;
-use util::parse_user_agent;
+use util::{ms_since_epoch, parse_user_agent, sec_since_epoch};
 use util::megaphone::{ClientServices, Service, ServiceClientInit};
 
 const MAX_EXPIRY: u64 = 2592000;
@@ -339,7 +338,7 @@ where
         };
 
         let AwaitHello { data, tx, rx, .. } = hello.take();
-        let connected_at = time::precise_time_ns() / 1000;
+        let connected_at = ms_since_epoch();
         transition!(AwaitProcessHello {
             response: data.srv.hello(&connected_at, uaid.as_ref()),
             data,
@@ -478,8 +477,8 @@ where
                 Err(_) => break,
             }
         }
-        let now = time::precise_time_ns() / 1000;
-        let elapsed = now - webpush.connected_at;
+        let now = ms_since_epoch();
+        let elapsed = (now - webpush.connected_at) / 1_000;
         let parser = Parser::new();
         let (ua_result, metrics_os, metrics_browser) = parse_user_agent(&parser, &user_agent);
         srv.metrics
@@ -833,7 +832,7 @@ where
         let srv = increment_storage.data.srv.clone();
         let ddb_response = retry_if(
             move || {
-                let expiry = (time::get_time().sec as u64) + MAX_EXPIRY;
+                let expiry = sec_since_epoch() + MAX_EXPIRY;
                 let mut attr_values = HashMap::new();
                 attr_values.insert(
                     ":timestamp".to_string(),
@@ -923,7 +922,7 @@ where
         webpush.unacked_stored_highest = timestamp;
         if messages.len() > 0 {
             // Filter out TTL expired messages
-            let now = time::get_time().sec as u32;
+            let now = sec_since_epoch() as u32;
             messages.retain(|ref msg| now < msg.ttl + msg.timestamp);
             webpush.flags.increment_storage = !include_topic && timestamp.is_some();
             // If there's still messages send them out
diff --git a/autopush_rs/src/util/mod.rs b/autopush_rs/src/util/mod.rs
index 1d4d2bf8..bebf272a 100644
--- a/autopush_rs/src/util/mod.rs
+++ b/autopush_rs/src/util/mod.rs
@@ -2,6 +2,7 @@
 use std::io;
 use std::time::Duration;
 
+use chrono::Utc;
 use hostname::get_hostname;
 use futures::future::{Either, Future, IntoFuture};
 use slog;
@@ -92,6 +93,23 @@ pub fn init_logging(json: bool) {
     slog_stdlog::init().ok();
 }
 
+/// Get the time since the UNIX epoch in seconds
+pub fn sec_since_epoch() -> u64 {
+    Utc::now().timestamp() as u64
+}
+
+/// Get the time since the UNIX epoch in milliseconds
+pub fn ms_since_epoch() -> u64 {
+    Utc::now().timestamp_millis() as u64
+}
+
+/// Get the time since the UNIX epoch in microseconds
+#[allow(dead_code)]
+pub fn us_since_epoch() -> u64 {
+    let now = Utc::now();
+    (now.timestamp() as u64) * 1_000_000 + (now.timestamp_subsec_micros() as u64)
+}
+
 pub fn reset_logging() {
     let logger = slog::Logger::root(slog::Discard, o!());
     slog_scope::set_global_logger(logger).cancel_reset();