From e6f44577f469e891ed8dab1492a4c53224e0b765 Mon Sep 17 00:00:00 2001
From: Stefan Boberg <stefan.boberg@epicgames.com>
Date: Mon, 2 Dec 2024 12:21:53 +0100
Subject: added support for dynamic LLM tags (#245)

* added FLLMTag which can be used to register memory tags outside of core
* changed `UE_MEMSCOPE` -> `ZEN_MEMSCOPE` for consistency
* instrumented some subsystems with dynamic tags
---
 src/zenhttp/servers/httpsys.cpp | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

(limited to 'src/zenhttp/servers/httpsys.cpp')

diff --git a/src/zenhttp/servers/httpsys.cpp b/src/zenhttp/servers/httpsys.cpp
index ac17d3ba0..9e8044a58 100644
--- a/src/zenhttp/servers/httpsys.cpp
+++ b/src/zenhttp/servers/httpsys.cpp
@@ -9,6 +9,7 @@
 #include <zencore/filesystem.h>
 #include <zencore/fmtutils.h>
 #include <zencore/logging.h>
+#include <zencore/memory/llm.h>
 #include <zencore/scopeguard.h>
 #include <zencore/string.h>
 #include <zencore/timer.h>
@@ -25,6 +26,14 @@
 
 namespace zen {
 
+const FLLMTag&
+GetHttpsysTag()
+{
+	static FLLMTag HttpsysTag("httpsys");
+
+	return HttpsysTag;
+}
+
 /**
  * @brief Windows implementation of HTTP server based on http.sys
  *
@@ -901,6 +910,8 @@ HttpSysServer::HttpSysServer(const HttpSysConfig& InConfig)
 , m_IsAsyncResponseEnabled(InConfig.IsAsyncResponseEnabled)
 , m_InitialConfig(InConfig)
 {
+	ZEN_MEMSCOPE(GetHttpsysTag());
+
 	// Initialize thread pool
 
 	int MinThreadCount;
@@ -976,6 +987,8 @@ HttpSysServer::Close()
 int
 HttpSysServer::InitializeServer(int BasePort)
 {
+	ZEN_MEMSCOPE(GetHttpsysTag());
+
 	using namespace std::literals;
 
 	WideStringBuilder<64> WildcardUrlPath;
@@ -1220,6 +1233,8 @@ HttpSysServer::Cleanup()
 WorkerThreadPool&
 HttpSysServer::WorkPool()
 {
+	ZEN_MEMSCOPE(GetHttpsysTag());
+
 	if (!m_AsyncWorkPool)
 	{
 		RwLock::ExclusiveLockScope _(m_AsyncWorkPoolInitLock);
@@ -1304,6 +1319,8 @@ HttpSysServer::IssueNewRequestMaybe()
 		return;
 	}
 
+	ZEN_MEMSCOPE(GetHttpsysTag());
+
 	std::unique_ptr<HttpSysTransaction> Request = std::make_unique<HttpSysTransaction>(*this);
 
 	std::error_code ErrorCode;
@@ -1327,6 +1344,8 @@ HttpSysServer::IssueNewRequestMaybe()
 void
 HttpSysServer::RegisterService(const char* UrlPath, HttpService& Service)
 {
+	ZEN_MEMSCOPE(GetHttpsysTag());
+
 	if (UrlPath[0] == '/')
 	{
 		++UrlPath;
-- 
cgit v1.2.3