From 59500f7c36fbc1dd11f66b805b7bb476b7be9c46 Mon Sep 17 00:00:00 2001
From: Adam Zielinski <zielinskia@wit.edu>
Date: Sat, 23 Feb 2019 18:12:12 -0500
Subject: [PATCH] make context available as LogFormatterParams

---
 logger.go      | 3 +++
 logger_test.go | 4 ++++
 2 files changed, 7 insertions(+)

diff --git a/logger.go b/logger.go
index dc639975..e8237bf4 100644
--- a/logger.go
+++ b/logger.go
@@ -66,6 +66,8 @@ type LogFormatterParams struct {
 	IsTerm bool
 	// BodySize is the size of the Response Body
 	BodySize int
+	// Context is the Context associated with the request.
+	Context *Context
 }
 
 // StatusCodeColor is the ANSI color for appropriately logging http status code to a terminal.
@@ -227,6 +229,7 @@ func LoggerWithConfig(conf LoggerConfig) HandlerFunc {
 			param := LogFormatterParams{
 				Request: c.Request,
 				IsTerm:  isTerm,
+				Context: c,
 			}
 
 			// Stop timer
diff --git a/logger_test.go b/logger_test.go
index c551677a..dae72f76 100644
--- a/logger_test.go
+++ b/logger_test.go
@@ -181,6 +181,7 @@ func TestLoggerWithFormatter(t *testing.T) {
 
 func TestLoggerWithConfigFormatting(t *testing.T) {
 	var gotParam LogFormatterParams
+	var gotCtx *Context
 	buffer := new(bytes.Buffer)
 
 	router := New()
@@ -204,6 +205,7 @@ func TestLoggerWithConfigFormatting(t *testing.T) {
 	router.GET("/example", func(c *Context) {
 		// set dummy ClientIP
 		c.Request.Header.Set("X-Forwarded-For", "20.20.20.20")
+		gotCtx = c
 	})
 	performRequest(router, "GET", "/example?a=100")
 
@@ -223,6 +225,8 @@ func TestLoggerWithConfigFormatting(t *testing.T) {
 	assert.Equal(t, "GET", gotParam.Method)
 	assert.Equal(t, "/example?a=100", gotParam.Path)
 	assert.Empty(t, gotParam.ErrorMessage)
+	assert.Empty(t, gotParam.ErrorMessage)
+	assert.Equal(t, gotCtx, gotParam.Context)
 
 }