diff options
-rw-r--r-- | src/creator.rs | 7 | ||||
-rw-r--r-- | src/writer.rs | 8 |
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(()) } |