mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-16 19:32:07 +00:00
53 lines
1.2 KiB
C++
53 lines
1.2 KiB
C++
#include <stdexcept>
|
|
#include <fstream>
|
|
#include <base/getMemoryAmount.h>
|
|
#include <base/getPageSize.h>
|
|
|
|
#include <unistd.h>
|
|
#include <sys/types.h>
|
|
#include <sys/param.h>
|
|
#if defined(BSD)
|
|
#include <sys/sysctl.h>
|
|
#endif
|
|
|
|
|
|
/** Returns the size of physical memory (RAM) in bytes.
|
|
* Returns 0 on unsupported platform
|
|
*/
|
|
uint64_t getMemoryAmountOrZero()
|
|
{
|
|
int64_t num_pages = sysconf(_SC_PHYS_PAGES);
|
|
if (num_pages <= 0)
|
|
return 0;
|
|
|
|
int64_t page_size = getPageSize();
|
|
if (page_size <= 0)
|
|
return 0;
|
|
|
|
uint64_t memory_amount = num_pages * page_size;
|
|
|
|
#if defined(OS_LINUX)
|
|
// Try to lookup at the Cgroup limit
|
|
std::ifstream cgroup_limit("/sys/fs/cgroup/memory/memory.limit_in_bytes");
|
|
if (cgroup_limit.is_open())
|
|
{
|
|
uint64_t memory_limit = 0; // in case of read error
|
|
cgroup_limit >> memory_limit;
|
|
if (memory_limit > 0 && memory_limit < memory_amount)
|
|
memory_amount = memory_limit;
|
|
}
|
|
#endif
|
|
|
|
return memory_amount;
|
|
|
|
}
|
|
|
|
|
|
uint64_t getMemoryAmount()
|
|
{
|
|
auto res = getMemoryAmountOrZero();
|
|
if (!res)
|
|
throw std::runtime_error("Cannot determine memory amount");
|
|
return res;
|
|
}
|