From 8c6179c4efc071ede46b4a8c0dd0acf556978df1 Mon Sep 17 00:00:00 2001
From: Drew Rothstein <arothste@blackrock.com>
Date: Thu, 13 Jan 2022 09:14:56 -0500
Subject: [PATCH] add container cpu and memory limits and requests

---
 kubectl-node_shell | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/kubectl-node_shell b/kubectl-node_shell
index 0c065b9..b13598e 100755
--- a/kubectl-node_shell
+++ b/kubectl-node_shell
@@ -95,6 +95,8 @@ pod="nsenter-$(env LC_ALL=C tr -dc a-z0-9 </dev/urandom | head -c 6)"
 # Check the node
 $kubectl get node "$node" >/dev/null || exit 1
 
+container_cpu="${KUBECTL_NODE_SHELL_POD_CPU:-100m}"
+container_memory="${KUBECTL_NODE_SHELL_POD_MEMORY:-256Mi}"
 overrides="$(
   cat <<EOT
 {
@@ -112,7 +114,17 @@ overrides="$(
         "stdin": true,
         "stdinOnce": true,
         "tty": $tty,
-        "command": $cmd
+        "command": $cmd,
+        "resources": {
+          "limits": {
+            "cpu": "${container_cpu}",
+            "memory": "${container_memory}"
+          },
+          "requests": {
+            "cpu": "${container_cpu}",
+            "memory": "${container_memory}"
+          }
+        }
       }
     ],
     "tolerations": [