feat(udp-receiver): add buffer

network-programmability/udp/udp-receiver/Cargo.toml

@@ -6,6 +6,7 @@ edition = "2021"
 
 [dependencies]
 prost = "0.13.3"
+socket2 = "0.5.7"
 tokio = { version = "1.41.0", features = ["full"] }
 
 [dev-dependencies]

network-programmability/udp/udp-receiver/src/main.rs

@@ -1,5 +1,7 @@
 use prost::Message;
+use socket2::{Socket, Domain, Type};
 use std::error::Error;
+use std::net::SocketAddr;
 use tokio::net::UdpSocket;
 
 pub mod iot {
@@ -10,16 +12,32 @@ use iot::Motor;
 
 #[tokio::main]
 async fn main() -> Result<(), Box<dyn Error>> {
-    let socket = UdpSocket::bind("127.0.0.1:50537").await?;
+    // Create a standard UDP socket using socket2
+    let socket = Socket::new(Domain::IPV4, Type::DGRAM, None)?;
+    socket.set_reuse_address(true)?;
+
+    // Increase the buffer size to 20 MiB
+    socket.set_recv_buffer_size(20 * 1024 * 1024)?;
+
+    // Bind the socket to the address
+    let addr: SocketAddr = "127.0.0.1:50537".parse()?;
+    socket.bind(&addr.into())?;
+
+    // Convert socket2::Socket into tokio::net::UdpSocket
+    let socket = UdpSocket::from_std(socket.into())?;
     println!("Listening on 127.0.0.1:50537");
 
     let mut buffer = vec![0u8; 1024]; // Buffer to hold received data
+    let mut message_count = 0; // Counter for received messages
 
     loop {
-        let (amt, src) = socket.recv_from(&mut buffer).await?;
+        let (amt, _src) = socket.recv_from(&mut buffer).await?;
         match Motor::decode(&buffer[..amt]) {
-            Ok(motor) => {
-                println!("Received from {}: {:?}", src, motor);
+            Ok(_motor) => {
+                message_count += 1;
+                if message_count % 10000 == 0 {
+                    println!("Total messages received: {}", message_count);
+                }
             }
             Err(e) => {
                 eprintln!("Failed to decode protobuf message: {}", e);