From 626e8a3e9fdf4a736ca48793caf41a0b772cbbd1 Mon Sep 17 00:00:00 2001
From: Jonathan Lange <jml@mumak.net>
Date: Mon, 12 Dec 2016 17:40:15 +0000
Subject: [PATCH] Neater way of doing it

---
 prog/app.go | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/prog/app.go b/prog/app.go
index 454f3b909..dfa108d84 100644
--- a/prog/app.go
+++ b/prog/app.go
@@ -270,11 +270,10 @@ func appMain(flags appFlags) {
 
 	handler := router(collector, controlRouter, pipeRouter, flags.externalUI)
 	if flags.logHTTP {
-		if flags.logHTTPHeaders {
-			handler = middleware.Log{LogRequestHeaders: true}.Wrap(handler)
-		} else {
-			handler = middleware.LogFailed.Wrap(handler)
-		}
+		handler = middleware.Log{
+			LogRequestHeaders: flags.logHTTPHeaders,
+			LogSuccess:        false,
+		}.Wrap(handler)
 	}
 
 	server := &graceful.Server{