thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-09-20 00:30:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Recursively check cgroups.controllers files for memory controllers

Browse Source
This commit is contained in:
Robert Schulze 2024-04-24 10:01:24 +00:00
parent 0ac60137f0
commit 2510fc85b0
No known key found for this signature in database
GPG Key ID: 26703B55FB13728A
1 changed files with 11 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
base/base/cgroupsv2.cpp
View File

@ -23,18 +23,19 @@ bool cgroupsV2MemoryControllerEnabled()
{ {
#if defined(OS_LINUX) #if defined(OS_LINUX)
chassert(cgroupsV2Enabled()); chassert(cgroupsV2Enabled());
/// According to https://docs.kernel.org/admin-guide/cgroup-v2.html: /// According to https://docs.kernel.org/admin-guide/cgroup-v2.html, file "cgroup.controllers" defines
/// - file 'cgroup.controllers' defines which controllers *can* be enabled /// which controllers are enabled for the current cgroup. Check the bottom-most nested "cgroup.controllers"
/// - file 'cgroup.subtree_control' defines which controllers *are* enabled /// file.
/// Caveat: nested groups may disable controllers. For simplicity, check only the top-level group. std::string cgroup = cgroupV2OfProcess();
std::ifstream subtree_control_file(default_cgroups_mount / "cgroup.subtree_control"); auto cgroup_dir = cgroup.empty() ? default_cgroups_mount : (default_cgroups_mount / cgroup);
if (!subtree_control_file.is_open()) std::ifstream controllers_file(cgroup_dir / "cgroup.controllers");
if (!controllers_file.is_open())
return false; return false;
std::string controllers; std::string controllers;
std::getline(subtree_control_file, controllers); std::getline(controllers_file, controllers);
if (controllers.find("memory") == std::string::npos) if (controllers.find("memory") != std::string::npos)
return false; return true;
return true; return false;
#else #else
return false; return false;
#endif #endif