mirror of
https://github.com/kataras/iris.git
synced 2025-01-23 18:51:03 +01:00
106 lines
3.0 KiB
Go
106 lines
3.0 KiB
Go
package handlerconv
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/kataras/iris/context"
|
|
"github.com/kataras/iris/core/errors"
|
|
)
|
|
|
|
var errHandler = errors.New(`
|
|
Passed argument is not a func(context.Context) neither one of these types:
|
|
- http.Handler
|
|
- func(w http.ResponseWriter, r *http.Request)
|
|
- func(w http.ResponseWriter, r *http.Request, next http.HandlerFunc)
|
|
---------------------------------------------------------------------
|
|
It seems to be a %T points to: %v`)
|
|
|
|
// FromStd converts native http.Handler & http.HandlerFunc to context.Handler.
|
|
//
|
|
// Supported form types:
|
|
// .FromStd(h http.Handler)
|
|
// .FromStd(func(w http.ResponseWriter, r *http.Request))
|
|
// .FromStd(func(w http.ResponseWriter, r *http.Request, next http.HandlerFunc))
|
|
func FromStd(handler interface{}) context.Handler {
|
|
switch handler.(type) {
|
|
case context.Handler:
|
|
{
|
|
//
|
|
//it's already a iris handler
|
|
//
|
|
return handler.(context.Handler)
|
|
}
|
|
|
|
case http.Handler:
|
|
//
|
|
// handlerFunc.ServeHTTP(w,r)
|
|
//
|
|
{
|
|
h := handler.(http.Handler)
|
|
return func(ctx context.Context) {
|
|
h.ServeHTTP(ctx.ResponseWriter(), ctx.Request())
|
|
}
|
|
}
|
|
|
|
case func(http.ResponseWriter, *http.Request):
|
|
{
|
|
//
|
|
// handlerFunc(w,r)
|
|
//
|
|
return FromStd(http.HandlerFunc(handler.(func(http.ResponseWriter, *http.Request))))
|
|
}
|
|
|
|
case func(http.ResponseWriter, *http.Request, http.HandlerFunc):
|
|
{
|
|
//
|
|
// handlerFunc(w,r, http.HandlerFunc)
|
|
//
|
|
return FromStdWithNext(handler.(func(http.ResponseWriter, *http.Request, http.HandlerFunc)))
|
|
}
|
|
|
|
default:
|
|
{
|
|
//
|
|
// No valid handler passed
|
|
//
|
|
panic(errHandler.Format(handler, handler))
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
// FromStdWithNext receives a standar handler - middleware form - and returns a compatible context.Handler wrapper.
|
|
func FromStdWithNext(h func(w http.ResponseWriter, r *http.Request, next http.HandlerFunc)) context.Handler {
|
|
return func(ctx context.Context) {
|
|
// take the next handler in route's chain
|
|
nextIonHandler := ctx.NextHandler()
|
|
if nextIonHandler != nil {
|
|
executed := false // we need to watch this in order to StopExecution from all next handlers
|
|
// if this next handler is not executed by the third-party net/http next-style Handlers.
|
|
nextHandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
nextIonHandler(ctx)
|
|
executed = true
|
|
})
|
|
|
|
h(ctx.ResponseWriter(), ctx.Request(), nextHandler)
|
|
|
|
// after third-party Handlers's job:
|
|
if executed {
|
|
// if next is executed then increment the position manually
|
|
// in order to the next handler not to be executed twice.
|
|
ctx.HandlerIndex(ctx.HandlerIndex(-1) + 1)
|
|
} else {
|
|
// otherwise StopExecution from all next handlers.
|
|
ctx.StopExecution()
|
|
}
|
|
return
|
|
}
|
|
|
|
// if not next handler found then this is not a 'valid' Handlers but
|
|
// some Handlers may don't care about next,
|
|
// so we just execute the handler with an empty net.
|
|
h(ctx.ResponseWriter(), ctx.Request(), http.HandlerFunc(func(http.ResponseWriter, *http.Request) {}))
|
|
}
|
|
}
|