vector/internal_events/
websocket_server.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
use std::fmt::Debug;

use futures::channel::mpsc::TrySendError;
use metrics::{counter, gauge};
use tokio_tungstenite::tungstenite::Message;
use vector_lib::internal_event::InternalEvent;

use vector_lib::internal_event::{error_stage, error_type};

#[derive(Debug)]
pub struct WsListenerConnectionEstablished {
    pub client_count: usize,
}

impl InternalEvent for WsListenerConnectionEstablished {
    fn emit(self) {
        debug!(message = "Websocket client connected. Client count: {self.client_count}");
        counter!("connection_established_total").increment(1);
        gauge!("active_clients").set(self.client_count as f64);
    }

    fn name(&self) -> Option<&'static str> {
        Some("WsListenerConnectionEstablished")
    }
}

#[derive(Debug)]
pub struct WsListenerConnectionShutdown {
    pub client_count: usize,
}

impl InternalEvent for WsListenerConnectionShutdown {
    fn emit(self) {
        info!(message = "Client connection closed. Client count: {self.client_count}.");
        counter!("connection_shutdown_total").increment(1);
        gauge!("active_clients").set(self.client_count as f64);
    }

    fn name(&self) -> Option<&'static str> {
        Some("WsListenerConnectionShutdown")
    }
}

#[derive(Debug)]
pub struct WsListenerSendError {
    pub error: TrySendError<Message>,
}

impl InternalEvent for WsListenerSendError {
    fn emit(self) {
        error!(
            message = "WebSocket message send error.",
            error = %self.error,
            error_code = "ws_server_connection_error",
            error_type = error_type::WRITER_FAILED,
            stage = error_stage::SENDING,
            internal_log_rate_limit = true,
        );
        counter!(
            "component_errors_total",
            "error_code" => "ws_server_connection_error",
            "error_type" => error_type::WRITER_FAILED,
            "stage" => error_stage::SENDING,
        )
        .increment(1);
    }

    fn name(&self) -> Option<&'static str> {
        Some("WsListenerConnectionError")
    }
}