mirror of
https://github.com/kataras/iris.git
synced 2026-01-09 04:51:56 +00:00
reorganization of _examples and add some new examples such as iris+groupcache+mysql+docker
Former-commit-id: ed635ee95de7160cde11eaabc0c1dcb0e460a620
This commit is contained in:
@@ -10,7 +10,7 @@ func init() {
|
||||
context.SetHandlerName("iris/middleware/requestid.*", "iris.request.id")
|
||||
}
|
||||
|
||||
const xRequestIDHeaderValue = "X-Request-ID"
|
||||
const xRequestIDHeaderValue = "X-Request-Id"
|
||||
|
||||
// Generator defines the function which should extract or generate
|
||||
// a Request ID. See `DefaultGenerator` and `New` package-level functions.
|
||||
@@ -40,13 +40,14 @@ var DefaultGenerator Generator = func(ctx context.Context) string {
|
||||
}
|
||||
|
||||
// New returns a new request id middleware.
|
||||
// It accepts an ID Generator.
|
||||
// It optionally accepts an ID Generator.
|
||||
// The Generator can stop the handlers chain with an error or
|
||||
// return a valid ID (string).
|
||||
// If it's nil then the `DefaultGenerator` will be used instead.
|
||||
func New(gen Generator) context.Handler {
|
||||
if gen == nil {
|
||||
gen = DefaultGenerator
|
||||
func New(generator ...Generator) context.Handler {
|
||||
gen := DefaultGenerator
|
||||
if len(generator) > 0 {
|
||||
gen = generator[0]
|
||||
}
|
||||
|
||||
return func(ctx context.Context) {
|
||||
|
||||
@@ -17,7 +17,7 @@ func TestRequestID(t *testing.T) {
|
||||
|
||||
def := app.Party("/default")
|
||||
{
|
||||
def.Use(requestid.New(nil))
|
||||
def.Use(requestid.New())
|
||||
def.Get("/", h)
|
||||
}
|
||||
|
||||
@@ -51,7 +51,7 @@ func TestRequestID(t *testing.T) {
|
||||
ctx.SetID(expectedCustomIDFromOtherMiddleware)
|
||||
ctx.Next()
|
||||
})
|
||||
changeID.Use(requestid.New(nil))
|
||||
changeID.Use(requestid.New())
|
||||
changeID.Get("/", h)
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user