use {
solana_connection_cache::client_connection::ClientConnection,
solana_streamer::sendmmsg::batch_send,
solana_transaction_error::TransportResult,
std::{
net::{SocketAddr, UdpSocket},
sync::Arc,
},
};
pub struct UdpClientConnection {
pub socket: Arc<UdpSocket>,
pub addr: SocketAddr,
}
impl UdpClientConnection {
pub fn new_from_addr(local_socket: Arc<UdpSocket>, server_addr: SocketAddr) -> Self {
Self {
socket: local_socket,
addr: server_addr,
}
}
}
impl ClientConnection for UdpClientConnection {
fn server_addr(&self) -> &SocketAddr {
&self.addr
}
fn send_data_async(&self, data: Arc<Vec<u8>>) -> TransportResult<()> {
self.socket.send_to(data.as_ref(), self.addr)?;
Ok(())
}
fn send_data_batch(&self, buffers: &[Vec<u8>]) -> TransportResult<()> {
let addr = self.server_addr();
let pkts = buffers.iter().map(|bytes| (bytes, addr));
batch_send(&self.socket, pkts)?;
Ok(())
}
fn send_data_batch_async(&self, buffers: Vec<Vec<u8>>) -> TransportResult<()> {
let addr = self.server_addr();
let pkts = buffers.iter().map(|bytes| (bytes, addr));
batch_send(&self.socket, pkts)?;
Ok(())
}
fn send_data(&self, buffer: &[u8]) -> TransportResult<()> {
self.socket.send_to(buffer, self.addr)?;
Ok(())
}
}