From 6ee83f5578ea6100189d06535c51811f0715a9e7 Mon Sep 17 00:00:00 2001
From: Manuel de Brito Fontes <aledbf@gmail.com>
Date: Mon, 12 Feb 2018 18:39:30 -0300
Subject: [PATCH] Fix opentracing configuration when multiple options are
 configured

---
 .../ingress/controller/template/template.go   | 52 +++++++++++++++++++
 rootfs/etc/nginx/template/nginx.tmpl          | 25 ++-------
 2 files changed, 55 insertions(+), 22 deletions(-)

diff --git a/internal/ingress/controller/template/template.go b/internal/ingress/controller/template/template.go
index 70d95c9835..a61fdd3ca2 100644
--- a/internal/ingress/controller/template/template.go
+++ b/internal/ingress/controller/template/template.go
@@ -17,6 +17,7 @@ limitations under the License.
 package template
 
 import (
+	"bytes"
 	"encoding/base64"
 	"encoding/json"
 	"fmt"
@@ -145,6 +146,8 @@ var (
 		"isValidClientBodyBufferSize": isValidClientBodyBufferSize,
 		"buildForwardedFor":           buildForwardedFor,
 		"buildAuthSignURL":            buildAuthSignURL,
+		"buildOpentracingLoad":        buildOpentracingLoad,
+		"buildOpentracing":            buildOpentracing,
 	}
 )
 
@@ -714,3 +717,52 @@ func randomString() string {
 
 	return string(b)
 }
+
+func buildOpentracingLoad(input interface{}) string {
+	cfg, ok := input.(config.Configuration)
+	if !ok {
+		glog.Errorf("expected a 'config.Configuration' type but %T was returned", input)
+		return ""
+	}
+
+	if !cfg.EnableOpentracing {
+		return ""
+	}
+
+	buf := bytes.NewBufferString("load_module/etc/nginx/modules/ngx_http_opentracing_module.so;")
+
+	if cfg.ZipkinCollectorHost != "" {
+		buf.WriteString("load_module/etc/nginx/modules/ngx_http_zipkin_module.so;")
+	} else if cfg.JaegerCollectorHost != "" {
+		buf.WriteString("load_module/etc/nginx/modules/ngx_http_jaeger_module.so;")
+	}
+
+	return buf.String()
+}
+
+func buildOpentracing(input interface{}) string {
+	cfg, ok := input.(config.Configuration)
+	if !ok {
+		glog.Errorf("expected a 'config.Configuration' type but %T was returned", input)
+		return ""
+	}
+
+	if !cfg.EnableOpentracing {
+		return ""
+	}
+
+	buf := bytes.NewBufferString("")
+
+	if cfg.ZipkinCollectorHost != "" {
+		buf.WriteString(fmt.Sprintf("zipkin_collector_host                   %v;", cfg.ZipkinCollectorHost))
+		buf.WriteString(fmt.Sprintf("zipkin_collector_port                   %v;", cfg.ZipkinCollectorPort))
+		buf.WriteString(fmt.Sprintf("zipkin_service_name                     %v;", cfg.ZipkinServiceName))
+	} else if cfg.JaegerCollectorHost != "" {
+		buf.WriteString(fmt.Sprintf("jaeger_reporter_local_agent_host_port   %v:%v;", cfg.JaegerCollectorHost, cfg.JaegerCollectorPort))
+		buf.WriteString(fmt.Sprintf("jaeger_service_name                     %v;", cfg.JaegerServiceName))
+		buf.WriteString(fmt.Sprintf("jaeger_sampler_type                     %v;", cfg.JaegerSamplerType))
+		buf.WriteString(fmt.Sprintf("jaeger_sampler_param                    %v;", cfg.JaegerSamplerParam))
+	}
+
+	return buf.String()
+}
diff --git a/rootfs/etc/nginx/template/nginx.tmpl b/rootfs/etc/nginx/template/nginx.tmpl
index 2227398ecc..58b9759338 100644
--- a/rootfs/etc/nginx/template/nginx.tmpl
+++ b/rootfs/etc/nginx/template/nginx.tmpl
@@ -11,17 +11,7 @@
 load_module /etc/nginx/modules/ngx_http_modsecurity_module.so;
 {{ end }}
 
-{{ if $cfg.EnableOpentracing }}
-load_module /etc/nginx/modules/ngx_http_opentracing_module.so;
-{{ end }}
-
-{{ if (and $cfg.EnableOpentracing (ne $cfg.ZipkinCollectorHost "")) }}
-load_module /etc/nginx/modules/ngx_http_zipkin_module.so;
-{{ end }}
-
-{{ if (and $cfg.EnableOpentracing (ne $cfg.JaegerCollectorHost "")) }}
-load_module /etc/nginx/modules/ngx_http_jaeger_module.so;
-{{ end }}
+{{ buildOpentracingLoad $cfg }}
 
 daemon off;
 
@@ -107,17 +97,8 @@ http {
     {{ if $cfg.EnableOpentracing }}
     opentracing on;
     {{ end }}
-
-    {{ if (and $cfg.EnableOpentracing (ne $cfg.ZipkinCollectorHost "")) }}
-    zipkin_collector_host                   {{ $cfg.ZipkinCollectorHost }};
-    zipkin_collector_port                   {{ $cfg.ZipkinCollectorPort }};
-    zipkin_service_name                     {{ $cfg.ZipkinServiceName }};
-    {{ else if (and $cfg.EnableOpentracing (ne $cfg.JaegerCollectorHost "")) }}
-    jaeger_reporter_local_agent_host_port   {{ $cfg.JaegerCollectorHost }}:{{ $cfg.JaegerCollectorPort }};
-    jaeger_service_name                     {{ $cfg.JaegerServiceName }};
-    jaeger_sampler_type                     {{ $cfg.JaegerSamplerType }};
-    jaeger_sampler_param                    {{ $cfg.JaegerSamplerParam }};
-    {{ end }}
+    
+    {{ buildOpentracing $cfg }}
 
     include /etc/nginx/mime.types;
     default_type text/html;