go/echo
go/echo
1
0
Fork 0
mirror of https://github.com/labstack/echo.git synced 2025-04-21 12:17:04 +02:00
Code Issues Releases Activity

Enable custom request id generator (#881)

Browse Source
This commit is contained in:
nownabe 2017-03-18 01:22:58 +09:00 committed by Vishal Rana
parent b2430fc4a8
commit cb6f0a5f80
2 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
middleware/request_id.go
View File

@ -50,7 +50,7 @@ func RequestIDWithConfig(config RequestIDConfig) echo.MiddlewareFunc {
res := c.Response() res := c.Response()
rid := req.Header.Get(echo.HeaderXRequestID) rid := req.Header.Get(echo.HeaderXRequestID)
if rid == "" { if rid == "" {
rid = random.String(32) rid = config.Generator()
} }
res.Header().Set(echo.HeaderXRequestID, rid) res.Header().Set(echo.HeaderXRequestID, rid)

16
middleware/request_id_test.go
View File

@ -14,10 +14,20 @@ func TestRequestID(t *testing.T) {
req, _ := http.NewRequest(echo.GET, "/", nil) req, _ := http.NewRequest(echo.GET, "/", nil)
rec := httptest.NewRecorder() rec := httptest.NewRecorder()
c := e.NewContext(req, rec) c := e.NewContext(req, rec)
rid := RequestIDWithConfig(RequestIDConfig{}) handler := func(c echo.Context) error {
h := rid(func(c echo.Context) error {
return c.String(http.StatusOK, "test") return c.String(http.StatusOK, "test")
}) }
rid := RequestIDWithConfig(RequestIDConfig{})
h := rid(handler)
h(c) h(c)
assert.Len(t, rec.Header().Get(echo.HeaderXRequestID), 32) assert.Len(t, rec.Header().Get(echo.HeaderXRequestID), 32)
// Custom generator
rid = RequestIDWithConfig(RequestIDConfig{
Generator: func() string { return "customGenerator" },
})
h = rid(handler)
h(c)
assert.Equal(t, rec.Header().Get(echo.HeaderXRequestID), "customGenerator")
} }