solana_udp_client/
udp_client.rs

1//! Simple client that communicates with the given UDP port with UDP and provides
2//! an interface for sending data
3
4use {
5    solana_connection_cache::client_connection::ClientConnection,
6    solana_streamer::sendmmsg::batch_send,
7    solana_transaction_error::TransportResult,
8    std::{
9        net::{SocketAddr, UdpSocket},
10        sync::Arc,
11    },
12};
13
14pub struct UdpClientConnection {
15    pub socket: Arc<UdpSocket>,
16    pub addr: SocketAddr,
17}
18
19impl UdpClientConnection {
20    pub fn new_from_addr(local_socket: Arc<UdpSocket>, server_addr: SocketAddr) -> Self {
21        Self {
22            socket: local_socket,
23            addr: server_addr,
24        }
25    }
26}
27
28impl ClientConnection for UdpClientConnection {
29    fn server_addr(&self) -> &SocketAddr {
30        &self.addr
31    }
32
33    fn send_data_async(&self, data: Vec<u8>) -> TransportResult<()> {
34        self.socket.send_to(data.as_ref(), self.addr)?;
35        Ok(())
36    }
37
38    fn send_data_batch(&self, buffers: &[Vec<u8>]) -> TransportResult<()> {
39        let addr = self.server_addr();
40        let pkts = buffers.iter().map(|bytes| (bytes, addr));
41        Ok(batch_send(&self.socket, pkts)?)
42    }
43
44    fn send_data_batch_async(&self, buffers: Vec<Vec<u8>>) -> TransportResult<()> {
45        let addr = self.server_addr();
46        let pkts = buffers.iter().map(|bytes| (bytes, addr));
47        Ok(batch_send(&self.socket, pkts)?)
48    }
49
50    fn send_data(&self, buffer: &[u8]) -> TransportResult<()> {
51        self.socket.send_to(buffer, self.addr)?;
52        Ok(())
53    }
54}
solana_udp_client/udp_client.rs

solana_udp_client/
udp_client.rs
