postgres/typesense
1
0
Fork 0
mirror of https://github.com/typesense/typesense.git synced 2025-05-26 00:36:22 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Add health with rusage.

Browse Source
This commit is contained in:
Kishore Nallan 2024-03-03 17:46:21 +05:30
parent ff9eb6b8ea
commit 51525108c2
3 changed files with 32 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
include/core_api.h
View File

@ -113,6 +113,8 @@ bool get_debug(const std::shared_ptr<http_req>& req, const std::shared_ptr<http_
bool get_health(const std::shared_ptr<http_req>& req, const std::shared_ptr<http_res>& res);
bool get_health_with_resource_usage(const std::shared_ptr<http_req>& req, const std::shared_ptr<http_res>& res);
bool post_health(const std::shared_ptr<http_req>& req, const std::shared_ptr<http_res>& res);
bool get_metrics_json(const std::shared_ptr<http_req>& req, const std::shared_ptr<http_res>& res);

29
src/core_api.cpp
View File

@ -311,6 +311,35 @@ bool get_debug(const std::shared_ptr<http_req>& req, const std::shared_ptr<http_
return true;
}
bool get_health_with_resource_usage(const std::shared_ptr<http_req>& req, const std::shared_ptr<http_res>& res) {
nlohmann::json result;
bool alive = server->is_alive();
auto resource_error = cached_resource_stat_t::get_instance().get_out_of_resource_error();
if (resource_error != cached_resource_stat_t::resource_check_t::OK) {
result["resource_error"] = std::string(magic_enum::enum_name(resource_error));
}
if(req->params.count("cpu_threshold") != 0 && StringUtils::is_float(req->params["cpu_threshold"])) {
float cpu_threshold = std::stof(req->params["cpu_threshold"]);
SystemMetrics sys_metrics;
std::vector<cpu_stat_t> cpu_stats = sys_metrics.get_cpu_stats();
if(!cpu_stats.empty() && StringUtils::is_float(cpu_stats[0].active)) {
alive = alive && (std::stof(cpu_stats[0].active) < cpu_threshold);
}
}
result["ok"] = alive;
if(alive) {
res->set_body(200, result.dump());
} else {
res->set_body(503, result.dump());
}
return alive;
}
bool get_health(const std::shared_ptr<http_req>& req, const std::shared_ptr<http_res>& res) {
nlohmann::json result;
bool alive = server->is_alive();

2
src/main/typesense_server.cpp
View File

@ -86,6 +86,7 @@ void master_server_routes() {
server->get("/stats.json", get_stats_json);
server->get("/debug", get_debug);
server->get("/health", get_health);
server->get("/health_with_rusage", get_health_with_resource_usage);
server->post("/health", post_health);
server->get("/status", get_status);
@ -100,7 +101,6 @@ void master_server_routes() {
server->get("/conversations/models/:id", get_conversation_model);
server->del("/conversations/models/:id", del_conversation_model);
server->get("/conversations", get_conversations);
server->get("/conversations/:id", get_conversation);
server->del("/conversations/:id", del_conversation);