strest 0.1.10

Blazing-fast async HTTP load tester in Rust - lock-free design, real-time stats, distributed runs, and optional chart exports for high-load API testing.
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

use std::sync::Arc;

use tokio::sync::mpsc;
use tokio::task::JoinHandle;

use crate::args::TesterArgs;
use crate::error::AppResult;
use crate::metrics::{LogSink, Metrics};
use crate::shutdown::ShutdownSender;

use super::resolve::resolve_endpoint;
use super::spawner::spawn_transport_sender;
use super::transports::udp_request_once;

pub(super) fn setup_datagram_sender(
    args: &TesterArgs,
    shutdown_tx: &ShutdownSender,
    metrics_tx: &mpsc::Sender<Metrics>,
    log_sink: Option<&Arc<LogSink>>,
    allowed_schemes: &[(&'static str, u16)],
    payload: Vec<u8>,
) -> AppResult<JoinHandle<()>> {
    let endpoint = resolve_endpoint(args, allowed_schemes)?;
    Ok(spawn_transport_sender(
        args,
        shutdown_tx,
        metrics_tx,
        log_sink,
        move |request_timeout, _connect_timeout| {
            let endpoint = endpoint;
            let payload = payload.clone();
            Box::pin(async move { udp_request_once(endpoint, &payload, request_timeout).await })
        },
    ))
}

pub(super) fn datagram_payload(args: &TesterArgs) -> Vec<u8> {
    if args.data.is_empty() {
        vec![0_u8]
    } else {
        args.data.clone().into_bytes()
    }
}