gfwleak/tango-tfe
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

TSG-23842 请tfe提供健康状态查询的HTTP服务

Browse Source
This commit is contained in:
luwenpeng
2024-11-20 14:53:45 +08:00
parent ff436e3f26
commit 6a21dd8848
5 changed files with 103 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

66
common/src/http_healthcheck.cpp Normal file
View File

@@ -0,0 +1,66 @@
#include <string.h>
#include <limits.h>
#include <pthread.h>
#include "event2/event.h"
#include "event2/buffer.h"
#include "event2/http.h"
#include "http_healthcheck.h"
#include <MESA/MESA_prof_load.h>
struct http_status_ctx
{
pthread_t pthread;
struct event_base *base;
unsigned int enable;
unsigned int port;
char addr[PATH_MAX];
char path[PATH_MAX];
uint64_t is_error;
} g_ctx = {0};
static void http_callback(struct evhttp_request *req, void *arg)
{
evhttp_send_reply(req, HTTP_OK, "", NULL);
}
static void *http_healthcheck_thread(void *arg)
{
printf("http_healthcheck_thread running (%s:%d/%s)\n", g_ctx.addr, g_ctx.port, g_ctx.path);
g_ctx.base = event_base_new();
struct evhttp *http = evhttp_new(g_ctx.base);
evhttp_set_cb(http, g_ctx.path, http_callback, NULL);
evhttp_bind_socket(http, g_ctx.addr, g_ctx.port);
event_base_dispatch(g_ctx.base);
evhttp_free(http);
event_base_free(g_ctx.base);
printf("http_healthcheck_thread exit\n");
return NULL;
}
void http_healthcheck_server_start(const char *profile)
{
MESA_load_profile_uint_def(profile, "http_healthcheck_server", "enable", &g_ctx.enable, 0);
MESA_load_profile_uint_def(profile, "http_healthcheck_server", "port", &g_ctx.port, 8080);
MESA_load_profile_string_def(profile, "http_healthcheck_server", "addr", g_ctx.addr, sizeof(g_ctx.addr), "127.0.0.1");
MESA_load_profile_string_def(profile, "http_healthcheck_server", "path", g_ctx.path, sizeof(g_ctx.path), "/status");
if (!g_ctx.enable)
{
return;
}
pthread_create(&g_ctx.pthread, NULL, http_healthcheck_thread, (void *)&g_ctx);
}
void http_healthcheck_server_stop()
{
if (g_ctx.enable)
{
event_base_loopbreak(g_ctx.base);
pthread_join(g_ctx.pthread, NULL);
}
}