scouter-events 0.21.0

Logic for setting up and running Scouter event-driven consumers and producers
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

use pyo3::prelude::*;
use scouter_types::TransportType;

#[pyclass]
#[derive(Clone, Debug)]
pub struct RabbitMQConfig {
    #[pyo3(get, set)]
    pub address: String,

    #[pyo3(get, set)]
    pub queue: String,

    #[pyo3(get, set)]
    pub max_retries: i32,

    #[pyo3(get)]
    pub transport_type: TransportType,
}

#[pymethods]
impl RabbitMQConfig {
    #[new]
    #[pyo3(signature = (host=None, port=None, username=None, password=None, queue="scouter_monitoring".to_string(), max_retries=3))]
    pub fn new(
        host: Option<String>,
        port: Option<u16>,
        username: Option<String>,
        password: Option<String>,
        queue: Option<String>,
        max_retries: Option<i32>,
    ) -> Self {
        // build address
        let address = std::env::var("RABBITMQ_ADDR").unwrap_or_else(|_| {
            let host = host.unwrap_or_else(|| {
                std::env::var("RABBITMQ_HOST").unwrap_or_else(|_| "localhost".to_string())
            });

            let port = port.unwrap_or_else(|| {
                std::env::var("RABBITMQ_PORT")
                    .unwrap_or_else(|_| "5672".to_string())
                    .parse::<u16>()
                    .unwrap()
            });

            let username = username.or_else(|| std::env::var("RABBITMQ_USERNAME").ok());

            let password = password.or_else(|| std::env::var("RABBITMQ_PASSWORD").ok());

            match (username, password) {
                (Some(user), Some(pass)) => {
                    format!("amqp://{user}:{pass}@{host}:{port}/%2f")
                }
                _ => {
                    format!("amqp://{host}:{port}/%2f")
                }
            }
        });

        let queue = queue.unwrap_or_else(|| {
            std::env::var("RABBITMQ_QUEUE").unwrap_or_else(|_| "scouter_monitoring".to_string())
        });

        RabbitMQConfig {
            address,
            queue,
            max_retries: max_retries.unwrap_or(3),
            transport_type: TransportType::RabbitMQ,
        }
    }
}