From 15a18201dbcb112f2eab903993b3faf7ef9c6491 Mon Sep 17 00:00:00 2001
From: "t.decaux" <ebuildy@gmail.com>
Date: Thu, 11 Mar 2021 18:39:20 +0100
Subject: [PATCH] [logstash] Add option loadBalancerIP to service

---
 logstash/templates/service.yaml |  7 +++++++
 logstash/tests/logstash_test.py | 13 +++++++++++++
 logstash/values.yaml            |  1 +
 3 files changed, 21 insertions(+)

diff --git a/logstash/templates/service.yaml b/logstash/templates/service.yaml
index 6540c8c61..d3a61bc7a 100644
--- a/logstash/templates/service.yaml
+++ b/logstash/templates/service.yaml
@@ -13,6 +13,13 @@ metadata:
 {{ toYaml .Values.service.annotations | indent 4 }}
 spec:
   type: {{ .Values.service.type }}
+{{- if .Values.service.loadBalancerIP }}
+  loadBalancerIP: {{ .Values.service.loadBalancerIP }}
+{{- end }}
+{{- with .Values.service.loadBalancerSourceRanges }}
+  loadBalancerSourceRanges:
+{{ toYaml . | indent 4 }}
+{{- end }}
   selector:
     app: "{{ template "logstash.fullname" . }}"
     chart: "{{ .Chart.Name }}"
diff --git a/logstash/tests/logstash_test.py b/logstash/tests/logstash_test.py
index ebb679b98..b68d661b5 100755
--- a/logstash/tests/logstash_test.py
+++ b/logstash/tests/logstash_test.py
@@ -862,6 +862,8 @@ def test_adding_a_service():
         "protocol": "TCP",
         "targetPort": 5044,
     }
+    # Make sure that the default 'loadBalancerIP' string is empty
+    assert "loadBalancerIP" not in s["spec"]
 
 
 def test_setting_fullnameOverride():
@@ -916,3 +918,14 @@ def test_hostaliases():
     r = helm_template(config)
     hostAliases = r["statefulset"][name]["spec"]["template"]["spec"]["hostAliases"]
     assert {"ip": "127.0.0.1", "hostnames": ["foo.local", "bar.local"]} in hostAliases
+
+
+def test_adding_loadBalancerIP():
+    config = """
+    service:
+      loadBalancerIP: 12.5.11.79
+    """
+
+    r = helm_template(config)
+
+    assert r["service"][name]["spec"]["loadBalancerIP"] == "12.5.11.79"
diff --git a/logstash/values.yaml b/logstash/values.yaml
index 43d7eb32d..b70caddec 100755
--- a/logstash/values.yaml
+++ b/logstash/values.yaml
@@ -247,6 +247,7 @@ lifecycle: {}
 service: {}
 #  annotations: {}
 #  type: ClusterIP
+#  loadBalancerIP: ""
 #  ports:
 #    - name: beats
 #      port: 5044