mirror of
https://github.com/Kelsidavis/WoWee.git
synced 2026-03-24 08:00:14 +00:00
79 lines
2.4 KiB
C++
79 lines
2.4 KiB
C++
#include "core/memory_monitor.hpp"
|
|
#include "core/logger.hpp"
|
|
#include <fstream>
|
|
#include <string>
|
|
#include <sstream>
|
|
#include <sys/sysinfo.h>
|
|
|
|
namespace wowee {
|
|
namespace core {
|
|
|
|
namespace {
|
|
size_t readMemAvailableBytesFromProc() {
|
|
std::ifstream meminfo("/proc/meminfo");
|
|
if (!meminfo.is_open()) return 0;
|
|
|
|
std::string line;
|
|
while (std::getline(meminfo, line)) {
|
|
// Format: "MemAvailable: 123456789 kB"
|
|
if (line.rfind("MemAvailable:", 0) != 0) continue;
|
|
std::istringstream iss(line.substr(13));
|
|
size_t kb = 0;
|
|
iss >> kb;
|
|
if (kb > 0) return kb * 1024ull;
|
|
break;
|
|
}
|
|
return 0;
|
|
}
|
|
} // namespace
|
|
|
|
MemoryMonitor& MemoryMonitor::getInstance() {
|
|
static MemoryMonitor instance;
|
|
return instance;
|
|
}
|
|
|
|
void MemoryMonitor::initialize() {
|
|
struct sysinfo info;
|
|
if (sysinfo(&info) == 0) {
|
|
totalRAM_ = static_cast<size_t>(info.totalram) * info.mem_unit;
|
|
LOG_INFO("System RAM detected: ", totalRAM_ / (1024 * 1024 * 1024), " GB");
|
|
} else {
|
|
// Fallback: assume 16GB
|
|
totalRAM_ = 16ull * 1024 * 1024 * 1024;
|
|
LOG_WARNING("Could not detect system RAM, assuming 16GB");
|
|
}
|
|
}
|
|
|
|
size_t MemoryMonitor::getAvailableRAM() const {
|
|
// Best source on Linux for reclaimable memory headroom.
|
|
if (size_t memAvailable = readMemAvailableBytesFromProc(); memAvailable > 0) {
|
|
return memAvailable;
|
|
}
|
|
|
|
struct sysinfo info;
|
|
if (sysinfo(&info) == 0) {
|
|
// Fallback approximation if /proc/meminfo is unavailable.
|
|
size_t freeBytes = static_cast<size_t>(info.freeram) * info.mem_unit;
|
|
size_t bufferBytes = static_cast<size_t>(info.bufferram) * info.mem_unit;
|
|
size_t available = freeBytes + bufferBytes;
|
|
return (totalRAM_ > 0 && available > totalRAM_) ? totalRAM_ : available;
|
|
}
|
|
return totalRAM_ / 2; // Fallback: assume 50% available
|
|
}
|
|
|
|
size_t MemoryMonitor::getRecommendedCacheBudget() const {
|
|
size_t available = getAvailableRAM();
|
|
// Use 80% of available RAM for caches (very aggressive), but cap at 90% of total
|
|
size_t budget = available * 80 / 100;
|
|
size_t maxBudget = totalRAM_ * 90 / 100;
|
|
return budget < maxBudget ? budget : maxBudget;
|
|
}
|
|
|
|
bool MemoryMonitor::isMemoryPressure() const {
|
|
size_t available = getAvailableRAM();
|
|
// Memory pressure if < 20% RAM available
|
|
return available < (totalRAM_ * 20 / 100);
|
|
}
|
|
|
|
} // namespace core
|
|
} // namespace wowee
|