fix: add ResponseWriter to return value

54m · 54m · commit e07d65a0ba69 · 2020-07-10T07:47:28.000+09:00
diff --git a/middleware.go b/middleware.go
@@ -109,7 +109,7 @@ func RequestLoggingWithEcho(config *Config) echo.MiddlewareFunc {
 }
 
 // SetContext for CloudFunctions
-func SetContext(config *Config, w http.ResponseWriter, r *http.Request) *http.Request {
+func SetContext(config *Config, w http.ResponseWriter, r *http.Request) (http.ResponseWriter, *http.Request) {
 	before := time.Now()
 
 	traceId := getTraceId(r)
@@ -143,7 +143,8 @@ func SetContext(config *Config, w http.ResponseWriter, r *http.Request) *http.Re
 			_, _ = fmt.Fprintln(os.Stderr, err.Error())
 		}
 	}()
-	return r
+
+	return wrw, r
 }
 
 func getTraceId(r *http.Request) string {

Original file line number	Diff line number	Diff line change
`@@ -109,7 +109,7 @@ func RequestLoggingWithEcho(config *Config) echo.MiddlewareFunc {`
`109`	`109`	`}`
`110`	`110`
`111`	`111`	`// SetContext for CloudFunctions`
`112`		`-func SetContext(config Config, w http.ResponseWriter, r http.Request) *http.Request {`
	`112`	`+func SetContext(config Config, w http.ResponseWriter, r http.Request) (http.ResponseWriter, *http.Request) {`
`113`	`113`	`before := time.Now()`
`114`	`114`
`115`	`115`	`traceId := getTraceId(r)`
`@@ -143,7 +143,8 @@ func SetContext(config Config, w http.ResponseWriter, r http.Request) *http.Re`
`143`	`143`	`_, _ = fmt.Fprintln(os.Stderr, err.Error())`
`144`	`144`	`}`
`145`	`145`	`}()`
`146`		`- return r`
	`146`	`+`
	`147`	`+ return wrw, r`
`147`	`148`	`}`
`148`	`149`
`149`	`150`	`func getTraceId(r *http.Request) string {`