Skip to main content

rust_web_server/app/controller/health/
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/// Liveness probe endpoint: `GET /healthz`.
12///
13/// Returns `200 OK` whenever the process is alive.
14/// Register this in your Kubernetes `livenessProbe`.
15pub struct HealthController;
16
17impl Controller for HealthController {
18    fn is_matching(request: &Request, _connection: &ConnectionInfo) -> bool {
19        request.method == METHOD.get && request.request_uri == "/healthz"
20    }
21
22    fn process(_request: &Request, mut response: Response, _connection: &ConnectionInfo) -> Response {
23        response.status_code = *STATUS_CODE_REASON_PHRASE.n200_ok.status_code;
24        response.reason_phrase = STATUS_CODE_REASON_PHRASE.n200_ok.reason_phrase.to_string();
25        response.content_range_list = vec![
26            Range::get_content_range(b"OK".to_vec(), MimeType::TEXT_PLAIN.to_string())
27        ];
28        response
29    }
30}
31
32impl HealthController {
33    pub fn is_matching_request(request: &Request) -> bool {
34        request.method == METHOD.get && request.request_uri == "/healthz"
35    }
36
37    pub fn process_request(_request: &Request, mut response: Response) -> Response {
38        response.status_code = *STATUS_CODE_REASON_PHRASE.n200_ok.status_code;
39        response.reason_phrase = STATUS_CODE_REASON_PHRASE.n200_ok.reason_phrase.to_string();
40        response.content_range_list = vec![
41            Range::get_content_range(b"OK".to_vec(), MimeType::TEXT_PLAIN.to_string())
42        ];
43        response
44    }
45}