From 60a45c22f9c173c0edb7ee10257d1749ce01e6a1 Mon Sep 17 00:00:00 2001
From: Daniel Wright <danielwright@bitgo.com>
Date: Mon, 11 Sep 2023 13:22:43 +1000
Subject: [PATCH] feat: add http header logging as option to rollouts demo
 application

Introduce an environment variable DISPLAY_HEADERS to toggle header
logging.

Signed-off-by: Daniel Wright <danielwright@bitgo.com>
---
 main.go | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/main.go b/main.go
index c1866e301..344977b8b 100644
--- a/main.go
+++ b/main.go
@@ -37,8 +37,9 @@ var (
 		"blue",
 		"purple",
 	}
-	envLatency   float64
-	envErrorRate int
+	envLatency         float64
+	envErrorRate       int
+	displayHttpHeaders bool
 )
 
 func init() {
@@ -57,6 +58,8 @@ func init() {
 			panic(fmt.Sprintf("failed to parse ERROR_RATE: %s", envErrorRateStr))
 		}
 	}
+	displayHttpHeadersEnv := os.Getenv("DISPLAY_HEADERS")
+	displayHttpHeaders = displayHttpHeadersEnv == "true" || displayHttpHeadersEnv == "1"
 }
 
 func main() {
@@ -187,8 +190,13 @@ func getColor(w http.ResponseWriter, r *http.Request) {
 	} else if envErrorRate > 0 && rand.Intn(100) >= envErrorRate {
 		statusCode = http.StatusInternalServerError
 	}
+	headersLog := ""
+	if displayHttpHeaders {
+		headersLog = fmt.Sprintf(", Headers: %v", r.Header)
+	}
+
 	printColor(colorToReturn, w, statusCode)
-	log.Printf("%d - %s%s\n", statusCode, colorToReturn, delayLengthStr)
+	log.Printf("%d - %s%s%s\n", statusCode, colorToReturn, delayLengthStr, headersLog)
 }
 
 func printColor(colorToPrint string, w http.ResponseWriter, statusCode int) {