udp.rs

use metrics::counter;
use vector_lib::NamedInternalEvent;
use vector_lib::internal_event::{
    ComponentEventsDropped, InternalEvent, UNINTENTIONAL, error_stage, error_type,
};

use crate::internal_events::SocketOutgoingConnectionError;

// TODO: Get rid of this. UDP is connectionless, so there's no "successful" connect event, only
// successfully binding a socket that can be used for receiving.
#[derive(Debug, NamedInternalEvent)]
pub struct UdpSocketConnectionEstablished;

impl InternalEvent for UdpSocketConnectionEstablished {
    fn emit(self) {
        debug!(message = "Connected.");
        counter!("connection_established_total", "mode" => "udp").increment(1);
    }
}

// TODO: Get rid of this. UDP is connectionless, so there's no "unsuccessful" connect event, only
// unsuccessfully binding a socket that can be used for receiving.
#[derive(NamedInternalEvent)]
pub struct UdpSocketOutgoingConnectionError<E> {
    pub error: E,
}

impl<E: std::error::Error> InternalEvent for UdpSocketOutgoingConnectionError<E> {
    fn emit(self) {
        // ## skip check-duplicate-events ##
        // ## skip check-validity-events ##
        emit!(SocketOutgoingConnectionError { error: self.error });
    }
}

#[derive(Debug, NamedInternalEvent)]
pub struct UdpSendIncompleteError {
    pub data_size: usize,
    pub sent: usize,
}

impl InternalEvent for UdpSendIncompleteError {
    fn emit(self) {
        let reason = "Could not send all data in one UDP datagram.";
        error!(
            message = reason,
            data_size = self.data_size,
            sent = self.sent,
            dropped = self.data_size - self.sent,
            error_type = error_type::WRITER_FAILED,
            stage = error_stage::SENDING,
        );
        counter!(
            "component_errors_total",
            "error_type" => error_type::WRITER_FAILED,
            "stage" => error_stage::SENDING,
        )
        .increment(1);
        // deprecated
        counter!("connection_send_errors_total", "mode" => "udp").increment(1);

        emit!(ComponentEventsDropped::<UNINTENTIONAL> { count: 1, reason });
    }
}

#[derive(Debug, NamedInternalEvent)]
pub struct UdpChunkingError {
    pub error: vector_common::Error,
    pub data_size: usize,
}

impl InternalEvent for UdpChunkingError {
    fn emit(self) {
        let reason = "Could not chunk UDP datagram.";
        error!(
            message = reason,
            data_size = self.data_size,
            error = self.error,
            error_type = error_type::WRITER_FAILED,
            stage = error_stage::SENDING,
        );
        counter!(
            "component_errors_total",
            "error_type" => error_type::WRITER_FAILED,
            "stage" => error_stage::SENDING,
        )
        .increment(1);

        emit!(ComponentEventsDropped::<UNINTENTIONAL> { count: 1, reason });
    }
}