Support for cluster autoscaler in GCE Trusty and GCI images

kubernetes · May 27, 2016 · 5f2695b · andyzheng0831 · Jun 1, 2016 · andyzheng0831
1 parent 73a5a13
commit 5f2695b
Show file tree

Hide file tree

Showing 5 changed files with 35 additions and 0 deletions.
diff --git a/build/common.sh b/build/common.sh
@@ -911,6 +911,7 @@ function kube::release::package_kube_manifests_tarball() {
   mkdir -p "${dst_dir}"
 
   local salt_dir="${KUBE_ROOT}/cluster/saltbase/salt"
+  cp "${salt_dir}/cluster-autoscaler/cluster-autoscaler.manifest" "${dst_dir}/"
   cp "${salt_dir}/fluentd-es/fluentd-es.yaml" "${release_stage}/"
   cp "${salt_dir}/fluentd-gcp/fluentd-gcp.yaml" "${release_stage}/"
   cp "${salt_dir}/kube-registry-proxy/kube-registry-proxy.yaml" "${release_stage}/"

diff --git a/cluster/gce/gci/configure-helper.sh b/cluster/gce/gci/configure-helper.sh
@@ -622,6 +622,22 @@ function start-kube-scheduler {
   cp "${src_file}" /etc/kubernetes/manifests
 }
 
+# Starts cluster autoscaler.
+function start-cluster-autoscaler {
+  if [ "${ENABLE_NODE_AUTOSCALER:-}" = "true" ]; then
+    touch /etc/kubernetes/start-cluster-autoscaler-enabled
+
+     # Remove salt comments and replace variables with values
+    src_file="${KUBE_HOME}/kube-manifests/kubernetes/gci-trusty/cluster-autoscaler.manifest"
+    remove-salt-config-comments "${src_file}"
+
+    local params=`sed 's/^/"/;s/ /","/g;s/$/",/' <<< "${AUTOSCALER_MIG_CONFIG}"`
+    sed -i -e "s@\"{{param}}\",@${params}@g" "${src_file}"
+    sed -i -e "s@{%.*%}@@g" "${src_file}"
+    cp "${src_file}" /etc/kubernetes/manifests
+  fi
+}
+
 # A helper function for copying addon manifests and set dir/files
 # permissions.
 #
@@ -812,6 +828,7 @@ if [[ "${KUBERNETES_MASTER:-}" == "true" ]]; then
   start-kube-controller-manager
   start-kube-scheduler
   start-kube-addons
+  start-cluster-autoscaler
 else
   start-kube-proxy
   # Kube-registry-proxy.

diff --git a/cluster/gce/trusty/configure-helper.sh b/cluster/gce/trusty/configure-helper.sh
@@ -614,6 +614,20 @@ start_kube_scheduler() {
   cp "${src_file}" /etc/kubernetes/manifests
 }
 
+# Starts k8s cluster autoscaler.
+start_cluster_autoscaler() {
+  if [ "${ENABLE_NODE_AUTOSCALER:-}" = "true" ]; then
+     # Remove salt comments and replace variables with values
+    src_file="${kube_home}/kube-manifests/kubernetes/gci-trusty/cluster-autoscaler.manifest"
+    remove_salt_config_comments "${src_file}"
+
+    local params=`sed 's/^/"/;s/ /","/g;s/$/",/' <<< "${AUTOSCALER_MIG_CONFIG}"`
+    sed -i -e "s@\"{{param}}\",@${params}@g" "${src_file}"
+    sed -i -e "s@{%.*%}@@g" "${src_file}"
+    cp "${src_file}" /etc/kubernetes/manifests
+  fi
+}
+
 # Starts a fluentd static pod for logging.
 start_fluentd() {
   if [ "${ENABLE_NODE_LOGGING:-}" = "true" ]; then

diff --git a/cluster/gce/trusty/master.yaml b/cluster/gce/trusty/master.yaml
@@ -191,6 +191,7 @@ script
 	start_kube_controller_manager
 	start_kube_scheduler
 	start_kube_addons
+	start_cluster_autoscaler
 } 2>&1 | logger --priority daemon.info -t ${UPSTART_JOB}
 end script
 

diff --git a/cluster/saltbase/salt/cluster-autoscaler/cluster-autoscaler.manifest b/cluster/saltbase/salt/cluster-autoscaler/cluster-autoscaler.manifest
@@ -1,3 +1,4 @@
+{% if pillar.get('enable_node_autoscaler', '').lower() == 'true' %}
 {% set params = pillar['autoscaler_mig_config'] -%}
 {
     "kind": "Pod",
@@ -56,3 +57,4 @@
         "restartPolicy": "Always"
     }
 }
+{% endif %}