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
use afire::{extension::RateLimiter, Content, Method, Middleware, Response, Server, Status};
use crate::Example;
// You can run this example with `cargo run --example basic -- rate_limit`
// Use some of afire's built-in middleware to log requests.
pub struct RateLimit;
impl Example for RateLimit {
fn name(&self) -> &'static str {
"rate_limit"
}
fn exec(&self) {
// Create a new Server instance on localhost port 8080
let mut server = Server::<()>::new("localhost", 8080);
// Define a handler for GET "/"
server.route(Method::GET, "/", |_req| {
Response::new().text("Hello World!").content(Content::TXT)
});
// For this example, we'll limit requests to 1 every 2 seconds
// Make a new Ratelimater
// Default Limit is 10
// Default Timeout is 60 sec
RateLimiter::new()
// Override the Limit to 1
.limit(1)
// Override the timeout to 2
.timeout(2)
// Override the Handler
.handler(Box::new(|_req| {
Some(
Response::new()
.status(Status::TooManyRequests)
.text("AHHHH!!! Too Many Requests")
.content(Content::TXT),
)
}))
// Attach to the server
.attach(&mut server);
// Now if you goto http://localhost:8080/ and reload a bunch of times,
// you'll see the rate limiter kicking in.
// Start the server
// This will block the current thread
server.start().unwrap();
}
}