fix: 🐛 Fix for loading memory limits from different cgroups versions

This commit is contained in:
Radek Goláň jr. 2024-08-25 16:59:02 +02:00
parent 379ba3dfaa
commit 26f32ca33a
Signed by: shield
GPG Key ID: D86423BFC31F3591

View File

@ -4,7 +4,15 @@ set -ex
# Caluclate Memory Limit from cgroups/maxmemory
MEMSIZE=$(grep MemTotal /proc/meminfo | awk '{print $2 * 1024}')
MEMLIMIT=$(cat /sys/fs/cgroup/memory/memory.limit_in_bytes)
if [ -f "/sys/fs/cgroup/memory/memory.limit_in_bytes" ]; then
MEMLIMIT=$(cat /sys/fs/cgroup/memory/memory.limit_in_bytes)
elif [ -f "/sys/fs/cgroup/memory.max" ]
MEMLIMIT=$(cat /sys/fs/cgroup/memory.max)
else
MEMLIMIT=$(grep MemTotal /proc/meminfo | awk '{print $2 * 1024}')
fi
MEMLIMIT=$((MEMLIMIT<MEMSIZE ? MEMLIMIT : MEMSIZE))
MEMOFFSET_THRESH=4294967296