From 4fd8dd78ef9b8edc15b1d9cdb1d5aedbf07febe4 Mon Sep 17 00:00:00 2001
From: Ryan Cavicchioni <ryan@cavi.cc>
Date: Wed, 30 Apr 2025 21:51:49 -0500
Subject: [PATCH] use jemalloc

---
 dockerfiles/alpine.Dockerfile   | 2 +-
 dockerfiles/bookworm.Dockerfile | 1 +
 dockerfiles/entrypoint.sh       | 5 +++++
 3 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/dockerfiles/alpine.Dockerfile b/dockerfiles/alpine.Dockerfile
index 9082200..297fd1c 100644
--- a/dockerfiles/alpine.Dockerfile
+++ b/dockerfiles/alpine.Dockerfile
@@ -5,7 +5,7 @@ WORKDIR /kubernaut
 
 RUN <<EOT
   apk update -q
-  apk add bash
+  apk add bash jemalloc
   rm -rf /var/cache/apk
   gem update --system --no-document
   gem install -N bundler
diff --git a/dockerfiles/bookworm.Dockerfile b/dockerfiles/bookworm.Dockerfile
index 3905fe9..9997ec4 100644
--- a/dockerfiles/bookworm.Dockerfile
+++ b/dockerfiles/bookworm.Dockerfile
@@ -5,6 +5,7 @@ WORKDIR /kubernaut
 
 RUN <<EOT
   apt-get update -qq
+  apt-get install --yes --no-install-recommends libjemalloc2
   rm -rf /var/lib/apt/lists /var/cache/apt/archives
   gem update --system --no-document
   gem install -N bundler
diff --git a/dockerfiles/entrypoint.sh b/dockerfiles/entrypoint.sh
index 58caad3..2f538b7 100755
--- a/dockerfiles/entrypoint.sh
+++ b/dockerfiles/entrypoint.sh
@@ -7,4 +7,9 @@ ruby --version
 printf "rubygems %s\n" "$(gem --version)"
 bundle version
 
+if [ -z "${LD_PRELOAD+x}" ]; then
+    LD_PRELOAD="$(find /usr/lib -name libjemalloc.so.2 -print -quit)"
+    export LD_PRELOAD
+fi
+
 exec "${@}"