Work on router optimizationsrouter

1 files changed, 24 insertions, 6 deletions

diff --git a/src/wireguard/router/worker.rs b/src/wireguard/router/worker.rs
index d95050e..bbb644c 100644
--- a/src/wireguard/router/worker.rs
+++ b/src/wireguard/router/worker.rs
@@ -1,13 +1,31 @@
-use super::Device;
-
 use super::super::{tun, udp, Endpoint};
 use super::types::Callbacks;
 
-use super::receive::ReceieveJob;
+use super::queue::ParallelJob;
+use super::receive::ReceiveJob;
 use super::send::SendJob;
 
-fn worker<E: Endpoint, C: Callbacks, T: tun::Writer, B: udp::Writer<E>>(
-    device: Device<E, C, T, B>,
+use crossbeam_channel::Receiver;
+
+pub enum JobUnion<E: Endpoint, C: Callbacks, T: tun::Writer, B: udp::Writer<E>> {
+    Outbound(SendJob<E, C, T, B>),
+    Inbound(ReceiveJob<E, C, T, B>),
+}
+
+pub fn worker<E: Endpoint, C: Callbacks, T: tun::Writer, B: udp::Writer<E>>(
+    receiver: Receiver<JobUnion<E, C, T, B>>,
 ) {
-    // fetch job
+    loop {
+        match receiver.recv() {
+            Err(_) => break,
+            Ok(JobUnion::Inbound(job)) => {
+                job.parallel_work();
+                job.queue().consume();
+            }
+            Ok(JobUnion::Outbound(job)) => {
+                job.parallel_work();
+                job.queue().consume();
+            }
+        }
+    }
 }