From e52ee65bc72fff942b6e3f5a13259f8316089d27 Mon Sep 17 00:00:00 2001
From: arothste-blk <53234250+arothste-blk@users.noreply.github.com>
Date: Thu, 13 Jan 2022 12:23:47 -0500
Subject: [PATCH] add container cpu and memory limits and requests (#38)

---
 kubectl-node_shell | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/kubectl-node_shell b/kubectl-node_shell
index 0c065b9..539ff65 100755
--- a/kubectl-node_shell
+++ b/kubectl-node_shell
@@ -2,7 +2,7 @@
 set -e
 
 kubectl=kubectl
-version=1.5.4
+version=1.5.5
 generator=""
 node=""
 nodefaultctx=0
@@ -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": [