derust 0.3.9

Easy way to start your Rust asynchronous application server using Tokio and Axum frameworks.
Documentation 
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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127

use crate::httpx::{AppContext, HttpError, HttpTags};
use crate::metricx::{current_gauge, MetricTags, start_stopwatch};
use crate::shutdown_signal;
use axum::http::StatusCode;
use outbox_pattern_processor::error::OutboxPatternProcessorError;
use outbox_pattern_processor::outbox_processor::OutboxProcessor;
use outbox_pattern_processor::outbox_resources::OutboxProcessorResources;
use std::time::Duration;
use tracing::{error, info};
use wg::WaitGroup;

pub async fn run<T>(
    wg: WaitGroup,
    context: AppContext<T>,
    outbox_processor_resources: OutboxProcessorResources,
    #[cfg(any(feature = "statsd", feature = "prometheus"))] metrics_monitor_enabled: bool,
    #[cfg(any(feature = "statsd", feature = "prometheus"))] outbox_metrics_monitor_interval_in_secs: Option<u64>,
) where
    T: Clone + Send + Sync + 'static,
{
    if outbox_processor_resources.scheduled_clear_locked_partition.unwrap_or(false) {
        tokio::spawn(run_clear_locked_partition(wg.add(1), outbox_processor_resources.clone()));
    }

    #[cfg(any(feature = "statsd", feature = "prometheus"))]
    if metrics_monitor_enabled {
        tokio::spawn(metrics_monitor(wg.add(1), context.clone(), outbox_processor_resources.clone(), outbox_metrics_monitor_interval_in_secs));
    }

    info!("Started embedded outbox-pattern-processor");

    let _ = OutboxProcessor::new(outbox_processor_resources.clone())
        .with_graceful_shutdown(shutdown_signal())
        .init_process()
        .await;

    wg.done();

    info!("Embedded outbox-pattern-processor stopped");
}

async fn run_clear_locked_partition(
    wg: WaitGroup,
    outbox_processor_resources: OutboxProcessorResources,
) {
    info!("Started embedded outbox-pattern-processor clear locked partition");

    let _ = OutboxProcessor::new(outbox_processor_resources)
        .with_graceful_shutdown(shutdown_signal())
        .init_processed_locked_cleaner()
        .await;

    wg.done();

    info!("Embedded outbox-pattern-processor clear locked partition stopped");
}

#[cfg(any(feature = "statsd", feature = "prometheus"))]
async fn metrics_monitor<T>(
    wg: WaitGroup,
    context: AppContext<T>,
    outbox_processor_resources: OutboxProcessorResources,
    outbox_metrics_monitor_interval_in_secs: Option<u64>,
) where
    T: Clone + Send + Sync + 'static,
{
    info!("Started embedded outbox-pattern-processor-monitor");

    let mut shutdown_signal = Box::into_pin(Box::new(shutdown_signal()));

    loop {
        tokio::select! {
            _ = one_shot_metrics_monitor(&context) => {
                tokio::time::sleep(Duration::from_secs(outbox_metrics_monitor_interval_in_secs.unwrap_or(5))).await; // TODO ajuste na duration
            }
            _ = &mut shutdown_signal => {
                break;
            }
        }
    }

    wg.done();

    info!("Embedded outbox-pattern-processor-monitor stopped");
}

#[cfg(any(feature = "statsd", feature = "prometheus"))]
async fn one_shot_metrics_monitor<T>(context: &AppContext<T>) -> Result<(), HttpError>
where
    T: Clone + Send + Sync + 'static,
{
    let tags = HttpTags::default();

    let mut metric_tags = MetricTags::from(tags.clone()).push("operation".to_string(), "monitor".to_string());
    let stopwatch = start_stopwatch(&context, "outbox_pattern_processor", metric_tags.clone());

    let query = sqlx::query_scalar::<_, i32>(
        r#"
        select coalesce(floor(extract(epoch from (now() - min(process_after)))), 0)::int as delay
        from outbox
        where processed_at is null
        "#,
    );

    let mut conn = context.database().get_connection(true, &tags).await?;

    let result = query.fetch_one(&mut *conn).await.map_err(|error| {
        HttpError::without_body(
            StatusCode::INTERNAL_SERVER_ERROR,
            format!("Failed to one_shot_metrics_monitor: {error}"),
            tags.clone(),
        )
    });

    match result {
        Ok(delay) => {
            current_gauge(&context, "outbox_pattern_processor_delay", metric_tags.clone(), delay as f64);
        },
        Err(error) => {
            error!("Failed when try to get outbox_pattern_processor_delay!");
        }
    }

    stopwatch.record(metric_tags.clone());

    Ok(())
}