Closes #44392

cluster/juju/layers/kubernetes-worker/actions.yaml

@@ -1,6 +1,17 @@
 pause:
     description: |
       Cordon the unit, draining all active workloads.
+    params:
+      delete-local-data:
+        type: boolean
+        description: Force deletion of local storage to enable a drain
+        default: False
+      force:
+        type: boolean
+        description: |
+            Continue even if there are pods not managed by a RC, RS, Job, DS or SS
+        default: False
+
 resume:
     description: |
       UnCordon the unit, enabling workload scheduling.

cluster/juju/layers/kubernetes-worker/actions/pause

@@ -4,6 +4,25 @@ set -ex
 
 export PATH=$PATH:/snap/bin
 
+DELETE_LOCAL_DATA=$(action-get delete-local-data)
+FORCE=$(action-get force)
+
+# placeholder for additional flags to the command
+export EXTRA_FLAGS=""
+
+# Determine if we have extra flags
+if [[ "${DELETE_LOCAL_DATA}" == "True" || "${DELETE_LOCAL_DATA}" == "true" ]]; then
+  EXTRA_FLAGS="${EXTRA_FLAGS} --delete-local-data=true"
+fi
+
+if [[ "${FORCE}" == "True" || "${FORCE}" == "true" ]]; then
+  EXTRA_FLAGS="${EXTRA_FLAGS} --force"
+fi
+
+
+# Cordon and drain the unit
 kubectl --kubeconfig=/root/cdk/kubeconfig cordon $(hostname)
-kubectl --kubeconfig=/root/cdk/kubeconfig drain $(hostname) --force
+kubectl --kubeconfig=/root/cdk/kubeconfig drain $(hostname) ${EXTRA_FLAGS}
+
+# Set status to indicate the unit is paused and under maintenance.
 status-set 'waiting' 'Kubernetes unit paused'