pass in the current time

2 files changed, 11 insertions, 4 deletions

diff --git a/src/creator.rs b/src/creator.rs
index bda3719..02d0c12 100644
--- a/src/creator.rs
+++ b/src/creator.rs
@@ -9,8 +9,11 @@ impl Creator {
         Default::default()
     }
 
-    pub fn frame(&mut self, data: &[u8]) -> crate::error::Result<Vec<u8>> {
-        let cur_time = std::time::Instant::now();
+    pub fn frame(
+        &mut self,
+        cur_time: std::time::Instant,
+        data: &[u8],
+    ) -> crate::error::Result<Vec<u8>> {
         let base_time = if let Some(base_time) = &self.base_time {
             base_time
         } else {
diff --git a/src/writer.rs b/src/writer.rs
index 7d2ecb4..1717032 100644
--- a/src/writer.rs
+++ b/src/writer.rs
@@ -15,8 +15,12 @@ impl<W: tokio::io::AsyncWrite> Writer<W> {
         }
     }
 
-    pub fn frame(&mut self, data: &[u8]) -> crate::error::Result<()> {
-        let bytes = self.creator.frame(data)?;
+    pub fn frame(
+        &mut self,
+        time: std::time::Instant,
+        data: &[u8],
+    ) -> crate::error::Result<()> {
+        let bytes = self.creator.frame(time, data)?;
         self.to_write.extend(bytes.iter());
         Ok(())
     }