annotation on pods for kubernetes/autoscaler

topfreegames · Nov 8, 2017 · d61701e · d61701e
1 parent 665b749
commit d61701e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/models/pod.go b/models/pod.go
@@ -31,6 +31,8 @@ metadata:
   namespace: {{.Namespace}}
   labels:
     app: {{.Name}}
+  annotations:
+    cluster-autoscaler.kubernetes.io/safe-to-evict: "true"
 spec:
   terminationGracePeriodSeconds: {{.ShutdownTimeout}}
 {{- if .NodeToleration }}
@@ -157,10 +159,12 @@ func NewPod(
 	return pod, err
 }
 
+//SetAffinity sets kubernetes Affinity on pod
 func (p *Pod) SetAffinity(affinity string) {
 	p.NodeAffinity = affinity
 }
 
+//SetToleration sets kubernetes Toleration on pod
 func (p *Pod) SetToleration(toleration string) {
 	p.NodeToleration = toleration
 }

diff --git a/models/pod_test.go b/models/pod_test.go
@@ -217,6 +217,8 @@ var _ = Describe("Pod", func() {
 			}
 			Expect(podv1.Spec.Containers[0].Command).To(HaveLen(3))
 			Expect(podv1.Spec.Containers[0].Command).To(Equal(command))
+
+			Expect(podv1.GetAnnotations()).To(HaveKeyWithValue("cluster-autoscaler.kubernetes.io/safe-to-evict", "true"))
 		})
 
 		It("should create pod without requests and limits", func() {