TSG-23844 请packet adapter提供健康状态查询的HTTP服务v2.1.1-20241120

3 files changed, 93 insertions, 1 deletions

diff --git a/common/CMakeLists.txt b/common/CMakeLists.txt
index a14f32a..c80dfb8 100644
--- a/common/CMakeLists.txt
+++ b/common/CMakeLists.txt
@@ -2,10 +2,12 @@ add_library(common
     src/log.cpp
     src/packet_io.cpp
     src/packet_parser.cpp
-    src/packet_inject.cpp)
+    src/packet_inject.cpp
+    src/http_healthcheck.cpp)
 target_include_directories(common PUBLIC ${CMAKE_CURRENT_LIST_DIR}/include)
 
 target_link_libraries(common PUBLIC mrzcpd)
 target_link_libraries(common PUBLIC MESA_prof_load)
 target_link_libraries(common PUBLIC MESA_handle_logger)
+target_link_libraries(common PUBLIC libevent-static)
 add_subdirectory(test)
\ No newline at end of file
diff --git a/common/include/http_healthcheck.h b/common/include/http_healthcheck.h
new file mode 100644
index 0000000..a883984
--- /dev/null
+++ b/common/include/http_healthcheck.h
@@ -0,0 +1,24 @@
+#ifndef _HTTP_HEALTHCHECK_H
+#define _HTTP_HEALTHCHECK_H
+
+#ifdef __cplusplus
+extern "C"
+{
+#endif
+
+/*
+ * [http_healthcheck_server]
+ * enable=1
+ * port=8080
+ * addr="0.0.0.0"
+ * path="/status"
+ */
+
+void http_healthcheck_server_start(const char *profile);
+void http_healthcheck_server_stop();
+
+#ifdef __cplusplus
+}
+#endif
+
+#endif
diff --git a/common/src/http_healthcheck.cpp b/common/src/http_healthcheck.cpp
new file mode 100644
index 0000000..30d1fd5
--- /dev/null
+++ b/common/src/http_healthcheck.cpp
@@ -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);
+	}
+}