This is an automated email from the ASF dual-hosted git repository.

caigy pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/rocketmq-operator.git


The following commit(s) were added to refs/heads/master by this push:
     new e80e97f  Improve Broker‘s startup script  for Accurate Memory 
Allocation (#210)
e80e97f is described below

commit e80e97fd08ecf006413b752c505acb7d4bc1ca5c
Author: KEN <40203972+mr...@users.noreply.github.com>
AuthorDate: Sat Feb 3 22:43:12 2024 -0800

    Improve Broker‘s startup script  for Accurate Memory Allocation (#210)
    
    This update prevents Out-Of-Memory (OOM) errors in  Broker's pods by 
accurately adhering to their allocated memory, rather than relying on the 
host's total memory.
---
 images/broker/alpine/runbroker-customize.sh | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)

diff --git a/images/broker/alpine/runbroker-customize.sh 
b/images/broker/alpine/runbroker-customize.sh
index 842b9d8..19aa3e0 100755
--- a/images/broker/alpine/runbroker-customize.sh
+++ b/images/broker/alpine/runbroker-customize.sh
@@ -55,7 +55,34 @@ calculate_heap_sizes()
     case "`uname`" in
         Linux)
             system_memory_in_mb=`free -m| sed -n '2p' | awk '{print $2}'`
+            if [ -f /sys/fs/cgroup/memory/memory.limit_in_bytes ]; then
+                system_memory_in_mb_in_docker=$(($(cat 
/sys/fs/cgroup/memory/memory.limit_in_bytes)/1024/1024))
+            elif [ -f /sys/fs/cgroup/memory.max ]; then
+                system_memory_in_mb_in_docker=$(($(cat 
/sys/fs/cgroup/memory.max)/1024/1024))
+            else
+                error_exit "Can not get memory, please check cgroup"
+            fi
+            if [ $system_memory_in_mb_in_docker -lt $system_memory_in_mb ];then
+              system_memory_in_mb=$system_memory_in_mb_in_docker
+            fi
+
             system_cpu_cores=`egrep -c 'processor([[:space:]]+):.*' 
/proc/cpuinfo`
+            if [ -f /sys/fs/cgroup/cpu/cpu.cfs_quota_us ]; then
+                system_cpu_cores_in_docker=$(($(cat 
/sys/fs/cgroup/cpu/cpu.cfs_quota_us)/$(cat 
/sys/fs/cgroup/cpu/cpu.cfs_period_us)))
+            elif [ -f /sys/fs/cgroup/cpu.max ]; then
+                QUOTA=$(cut -d ' ' -f 1 /sys/fs/cgroup/cpu.max)
+                PERIOD=$(cut -d ' ' -f 2 /sys/fs/cgroup/cpu.max)
+                if [ "$QUOTA" == "max" ]; then # no limit, see 
https://docs.kernel.org/admin-guide/cgroup-v2.html#cgroup-v2-cpu
+                  system_cpu_cores_in_docker=$system_cpu_cores
+                else
+                  system_cpu_cores_in_docker=$(($QUOTA/$PERIOD))
+                fi
+            else
+                error_exit "Can not get cpu, please check cgroup"
+            fi
+            if [ $system_cpu_cores_in_docker -lt $system_cpu_cores -a 
$system_cpu_cores_in_docker -ne 0 ];then
+              system_cpu_cores=$system_cpu_cores_in_docker
+            fi
         ;;
         FreeBSD)
             system_memory_in_bytes=`sysctl hw.physmem | awk '{print $2}'`

Reply via email to