From caa454c1237123d0b0f9d12dc72d5519d5c3f15f Mon Sep 17 00:00:00 2001
From: Chris Fordham <chris@fordham-nagy.id.au>
Date: Tue, 25 Jul 2017 22:12:06 +1000
Subject: [PATCH] Parameterise the host LB host label used for affinity
 scheduling.

---
 templates/drone/4/docker-compose.yml.tpl | 2 +-
 templates/drone/4/rancher-compose.yml    | 6 ++++++
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/templates/drone/4/docker-compose.yml.tpl b/templates/drone/4/docker-compose.yml.tpl
index b822ae1..b2ed55b 100644
--- a/templates/drone/4/docker-compose.yml.tpl
+++ b/templates/drone/4/docker-compose.yml.tpl
@@ -113,4 +113,4 @@ services:
       - ${host_port}:${host_port}
     labels:
       io.rancher.scheduler.global: 'true'
-      io.rancher.scheduler.affinity:host_label_soft: drone_lb=true
+      io.rancher.scheduler.affinity:host_label_soft: ${drone_lb_host_label}
diff --git a/templates/drone/4/rancher-compose.yml b/templates/drone/4/rancher-compose.yml
index 2f7dc05..bac6c11 100644
--- a/templates/drone/4/rancher-compose.yml
+++ b/templates/drone/4/rancher-compose.yml
@@ -148,6 +148,12 @@ catalog:
         - 'true'
         - 'false'
       default: 'false'
+    - variable: drone_lb_host_label
+      label: Drone LB Host Label
+      description: Host Label (soft affinity) for scheduling of the load balancer service.
+      required: true
+      default: "drone_lb=true"
+      type: string
 services:
   agent:
     scale: ${agent_scale}