mirror of
https://github.com/labstack/echo.git
synced 2024-11-28 08:38:39 +02:00
Key auth validator now accepts context
Signed-off-by: Vishal Rana <vr@labstack.com>
This commit is contained in:
parent
97c680662d
commit
ee2ac3b9a2
@ -32,7 +32,7 @@ type (
|
||||
}
|
||||
|
||||
// KeyAuthValidator defines a function to validate KeyAuth credentials.
|
||||
KeyAuthValidator func(string) bool
|
||||
KeyAuthValidator func(string, echo.Context) bool
|
||||
|
||||
keyExtractor func(echo.Context) (string, error)
|
||||
)
|
||||
@ -94,7 +94,7 @@ func KeyAuthWithConfig(config KeyAuthConfig) echo.MiddlewareFunc {
|
||||
if err != nil {
|
||||
return echo.NewHTTPError(http.StatusBadRequest, err.Error())
|
||||
}
|
||||
if config.Validator(key) {
|
||||
if config.Validator(key, c) {
|
||||
return next(c)
|
||||
}
|
||||
|
||||
|
@ -15,7 +15,7 @@ func TestKeyAuth(t *testing.T) {
|
||||
res := httptest.NewRecorder()
|
||||
c := e.NewContext(req, res)
|
||||
config := KeyAuthConfig{
|
||||
Validator: func(key string) bool {
|
||||
Validator: func(key string, c echo.Context) bool {
|
||||
return key == "valid-key"
|
||||
},
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user