rust_web_server/app/controller/ready/
mod.rs

1#[cfg(test)]
2mod tests;
3
4use crate::controller::Controller;
5use crate::mime_type::MimeType;
6use crate::range::Range;
7use crate::request::{METHOD, Request};
8use crate::response::{Response, STATUS_CODE_REASON_PHRASE};
9use crate::server::ConnectionInfo;
10
11/// Readiness probe endpoint: `GET /readyz`.
12///
13/// Returns `200 OK` when [`crate::metrics::SERVER_READY`] is `true`
14/// (set after [`crate::server::Server::setup`] completes and cleared on shutdown).
15/// Returns `503 Service Unavailable` during startup or graceful drain.
16/// Register this in your Kubernetes `readinessProbe`.
17pub struct ReadyController;
18
19impl Controller for ReadyController {
20    fn is_matching(request: &Request, _connection: &ConnectionInfo) -> bool {
21        request.method == METHOD.get && request.request_uri == "/readyz"
22    }
23
24    fn process(_request: &Request, mut response: Response, _connection: &ConnectionInfo) -> Response {
25        ReadyController::fill_response(&mut response);
26        response
27    }
28}
29
30impl ReadyController {
31    fn fill_response(response: &mut Response) {
32        use std::sync::atomic::Ordering;
33        if crate::metrics::SERVER_READY.load(Ordering::Relaxed) {
34            response.status_code = *STATUS_CODE_REASON_PHRASE.n200_ok.status_code;
35            response.reason_phrase = STATUS_CODE_REASON_PHRASE.n200_ok.reason_phrase.to_string();
36            response.content_range_list = vec![
37                Range::get_content_range(b"OK".to_vec(), MimeType::TEXT_PLAIN.to_string())
38            ];
39        } else {
40            response.status_code = *STATUS_CODE_REASON_PHRASE.n503_service_unavailable.status_code;
41            response.reason_phrase = STATUS_CODE_REASON_PHRASE.n503_service_unavailable.reason_phrase.to_string();
42            response.content_range_list = vec![
43                Range::get_content_range(b"not ready".to_vec(), MimeType::TEXT_PLAIN.to_string())
44            ];
45        }
46    }
47
48    pub fn is_matching_request(request: &Request) -> bool {
49        request.method == METHOD.get && request.request_uri == "/readyz"
50    }
51
52    pub fn process_request(_request: &Request, mut response: Response) -> Response {
53        ReadyController::fill_response(&mut response);
54        response
55    }
56}
rust_web_server/app/controller/ready/mod.rs

rust_web_server/app/controller/ready/
mod.rs
