reqwest-lb 0.3.1

The reqwest load balancer middleware
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

#![allow(unused)]

use reqwest::{Client, Url};
use reqwest_lb::supplier::LoadBalancer;
use reqwest_lb::LoadBalancerMiddleware;
use reqwest_lb::LoadBalancerPolicy;
use reqwest_lb::LoadBalancerRegistry;
use reqwest_middleware::{ClientBuilder, ClientWithMiddleware};

pub fn create<const N: usize>(
    ports: [u16; N],
    policy: LoadBalancerPolicy<Url>,
) -> ClientWithMiddleware {
    // create load balancer factory
    let mut registry = LoadBalancerRegistry::default();

    // create url list
    let mut urls = Vec::with_capacity(N);
    for port in ports {
        urls.push(Url::parse(format!("http://127.0.0.1:{}", port).as_str()).unwrap());
    }

    // create load balancer
    let load_balancer = LoadBalancer::new(urls, policy);

    // register load balancer for the host: example-server
    registry.add("example-server", load_balancer);

    // create reqwest client
    let middleware = LoadBalancerMiddleware::new(registry);
    ClientBuilder::new(Client::builder().no_proxy().build().unwrap())
        .with(middleware)
        .build()
}

#[tokio::main]
async fn main() {
    // use round robin policy
    let client = create([3001, 3002], LoadBalancerPolicy::RoundRobin);
    // http://127.0.0.1:3001
    let response = client.get("lb://example-server/").send().await.unwrap();
    // http://127.0.0.1:3002
    let response = client.get("lb://example-server/").send().await.unwrap();
}