Check for MESOS_TCP (#509)

d2iq-archive · Feb 7, 2018 · 58654fb · 58654fb
1 parent 53ddd80
commit 58654fb
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/marathon_lb.py b/marathon_lb.py
@@ -520,7 +520,8 @@ def config(apps, groups, bind_http_https, ssl_certs, templater,
                 if app.healthCheck:
                     template_backend_health_check = None
                     if app.mode == 'tcp' \
-                            or app.healthCheck['protocol'] == 'TCP':
+                            or app.healthCheck['protocol'] == 'TCP' \
+                            or app.healthCheck['protocol'] == 'MESOS_TCP':
                         template_backend_health_check = templater \
                             .haproxy_backend_tcp_healthcheck_options(app)
                     elif app.mode == 'http':
@@ -563,7 +564,9 @@ def config(apps, groups, bind_http_https, ssl_certs, templater,
             server_health_check_options = None
             if app.healthCheck:
                 template_server_healthcheck_options = None
-                if app.mode == 'tcp' or app.healthCheck['protocol'] == 'TCP':
+                if app.mode == 'tcp' \
+                        or app.healthCheck['protocol'] == 'TCP' \
+                        or app.healthCheck['protocol'] == 'MESOS_TCP':
                     template_server_healthcheck_options = templater \
                         .haproxy_backend_server_tcp_healthcheck_options(app)
                 elif app.mode == 'http':