概述

RpmTimer (每分钟请求计时器) 是一个工具，可以将您的处理速度限制为每分钟的请求数量（例如请求）。

它旨在与任何速率限制的API一起使用。

文档

存储库

入门指南

首先，您必须将此依赖项添加到您的 Cargo.toml

[dev-dependencies]
rpm-timer = "0.0.3"

然后，在您的 lib.rs 或 main.rs 文件中包含 rpm_timer crate

extern crate rpm_timer;

最后，在您想要限制处理速度的模块中使用 RpmTimer

use rpm_timer::RpmTimer;

示例

为了避免不必要的内存分配，run 函数有两种版本。

1. run_slices 接受切片并将子切片传递给处理函数

extern crate rpm_timer;

use rpm_timer::RpmTimer;

fn send_slice(requests: Vec<String>) {
    RpmTimer::default()
        .rpm_limit(100.0)
        .run_slice(&requests, send_http_requests);
}

fn send_http_requests(requests: &[&String]) {
    // process all requests here
}

2. run_iter 接受任何迭代器，收集项目并将每个部分作为 Vec 传递给处理函数：processing

extern crate rpm_timer;

use rpm_timer::RpmTimer;

fn send_slice(reader: BufReader) {
    let lines = reader.lines();

    RpmTimer::default()
        .rpm_limit(100.0)
        .run_iter(lines, send_http_requests);
}

fn send_http_requests(requests: Vec<Result<String, io::Error>>) {
    // process all requests here
}

请检查 examples 目录以获取更详细的工作示例。

描述

RpmTimer 在滴答间隔内工作。您可以使用 tick 方法调整滴答长度。每个滴答它检查是否有任何空闲的工作线程（线程的数量可以用 max_threads 调整）以及应该处理多少项目以实现请求的速度。如果有需要处理的项目，RpmTimer 将它们收集到切片（在不分配版本中）或 Vec（在分配版本中），并在并行中触发处理函数。

可视化

假设有2个工作线程和500毫秒的滴答时间。此外，假设在第2秒和第3秒之间存在延迟（例如，CPU忙于其他进程）

60 RPM = 1 RPS = 1 每秒1个请求

Time                     0   0.5   1   1.5   2   2.5   3   3.5
Main Thread:             |....X....X....X....X.........X....X..
Number of items 'ready'  1   0.5   1   0.5   1         2   0.5
Worker #1                1**********************.......2******.
Worker #2                ..........1**************.............
                                             ^         ^
                                             |         |- 2 items sent to the thread
                                             |- an item is ready but no worker threads available

Time                     0   0.5   1   1.5   2   2.5   3   3.5
Main Thread:             |....X....X....X..............X....X..
Number of items 'ready'  1  0.25  0.5  0.75  1  0.25  0.5  0.75
Worker #1                1***********..........................
Worker #2                ....................1************.....