Switch to a lock free method of storing packets.

src/rust/lqos_heimdall/src/perf_interface.rs

@@ -4,7 +4,7 @@ use zerocopy::FromBytes;
 
 use crate::{flows::record_flow, timeline::store_on_timeline};
 
-#[derive(FromBytes, Debug, Clone)]
+#[derive(FromBytes, Debug, Clone, PartialEq, Eq, Hash)]
 #[repr(C)]
 pub struct HeimdallEvent {
   pub timestamp: u64,

src/rust/lqos_heimdall/src/timeline.rs

@@ -1,40 +1,39 @@
-use std::{sync::RwLock, time::Duration};
-
+use std::time::Duration;
+use dashmap::DashSet;
 use lqos_bus::PacketHeader;
 use lqos_utils::{unix_time::time_since_boot, XdpIpAddress};
 use once_cell::sync::Lazy;
 use crate::perf_interface::HeimdallEvent;
 
-impl From<&HeimdallEvent> for PacketHeader {
-    fn from(value: &HeimdallEvent) -> Self {
-        Self {
-            timestamp: value.timestamp,
-            src: value.src.as_ip().to_string(),
-            dst: value.dst.as_ip().to_string(),
-            src_port: value.src_port,
-            dst_port: value.dst_port,
-            ip_protocol: value.ip_protocol,
-            tos: value.tos,
-            size: value.size,
+impl HeimdallEvent {
+    fn as_header(&self) -> PacketHeader {
+      PacketHeader {
+            timestamp: self.timestamp,
+            src: self.src.as_ip().to_string(),
+            dst: self.dst.as_ip().to_string(),
+            src_port: self.src_port,
+            dst_port: self.dst_port,
+            ip_protocol: self.ip_protocol,
+            tos: self.tos,
+            size: self.size,
         }
     }
 }
 
 struct Timeline {
-  data: RwLock<Vec<HeimdallEvent>>,
+  data: DashSet<HeimdallEvent>,
 }
 
 impl Timeline {
   fn new() -> Self {
-    Self { data: RwLock::new(Vec::new()) }
+    Self { data: DashSet::new() }
   }
 }
 
 static TIMELINE: Lazy<Timeline> = Lazy::new(Timeline::new);
 
 pub(crate) fn store_on_timeline(event: HeimdallEvent) {
-  let mut lock = TIMELINE.data.write().unwrap();
-  lock.push(event); // We're moving here deliberately
+  TIMELINE.data.insert(event); // We're moving here deliberately
 }
 
 pub(crate) fn expire_timeline() {
@@ -42,18 +41,15 @@ pub(crate) fn expire_timeline() {
     let since_boot = Duration::from(now);
     let ten_secs_ago = since_boot - Duration::from_secs(10);
     let expire = ten_secs_ago.as_nanos() as u64;
-    let mut lock = TIMELINE.data.write().unwrap();
-    lock.retain(|v| v.timestamp > expire);
+    TIMELINE.data.retain(|v| v.timestamp > expire);
   }
 }
 
 pub fn ten_second_packet_dump(ip: XdpIpAddress) -> Vec<PacketHeader> {
   TIMELINE
     .data
-    .read()
-    .unwrap()
     .iter()
     .filter(|e| e.src == ip || e.dst == ip)
-    .map(|e| e.into())
+    .map(|e| e.as_header())
     .collect()
 }