From e89f4ac7ee895f00d31ab9bb8fe5c0ddcfe7e0cf Mon Sep 17 00:00:00 2001
From: Aleksandr Didenko <adidenko@mirantis.com>
Date: Fri, 8 Jul 2016 15:04:31 +0200
Subject: [PATCH] Add labeling nodes

---
 ccp/label-nodes.sh        | 28 ++++++++++++++++++++++++++++
 playbooks/ccp-deploy.yaml |  5 +++++
 2 files changed, 33 insertions(+)
 create mode 100755 ccp/label-nodes.sh

diff --git a/ccp/label-nodes.sh b/ccp/label-nodes.sh
new file mode 100755
index 000000000..d8ba6cfbb
--- /dev/null
+++ b/ccp/label-nodes.sh
@@ -0,0 +1,28 @@
+#!/bin/bash
+
+set -e
+
+# FIXME: hardcoded roles
+declare -A nodes
+nodes=( \
+["node2"]="openstack-controller=true"
+["node3"]="openstack-controller=true"
+["node4"]="openstack-controller=true"
+["node5"]="openstack-compute=true"
+["node6"]="openstack-compute=true"
+["node7"]="openstack-compute=true"
+)
+
+label_nodes() {
+  all_label='openstack-compute-controller=true'
+  for i in "${!nodes[@]}"
+  do
+    node=$i
+    label=${nodes[$i]}
+    kubectl get nodes $node --show-labels | grep -q "$label" || kubectl label nodes $node $label
+    kubectl get nodes $node --show-labels | grep -q "$all_label" || kubectl label nodes $node $all_label
+  done
+}
+
+label_nodes
+
diff --git a/playbooks/ccp-deploy.yaml b/playbooks/ccp-deploy.yaml
index 4db12dc81..9e4b5e95b 100644
--- a/playbooks/ccp-deploy.yaml
+++ b/playbooks/ccp-deploy.yaml
@@ -8,6 +8,11 @@
         dest: /root/ccp/
 
   tasks:
+    - name: Label nodes
+      shell: ./label-nodes.sh
+      args:
+        chdir: /root/ccp
+      run_once: true
 
     - name: Get namespaces
       shell: kubectl get namespace