mirror of
https://github.com/kataras/iris.git
synced 2025-01-23 18:51:03 +01:00
59 lines
1.7 KiB
Go
59 lines
1.7 KiB
Go
package handlerconv
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"github.com/kataras/iris/v12/context"
|
|
)
|
|
|
|
// 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 h := handler.(type) {
|
|
case context.Handler:
|
|
return h
|
|
case func(*context.Context):
|
|
return h
|
|
case http.Handler:
|
|
// handlerFunc.ServeHTTP(w,r)
|
|
return func(ctx *context.Context) {
|
|
h.ServeHTTP(ctx.ResponseWriter(), ctx.Request())
|
|
}
|
|
case func(http.ResponseWriter, *http.Request):
|
|
// handlerFunc(w,r)
|
|
return FromStd(http.HandlerFunc(h))
|
|
case func(http.ResponseWriter, *http.Request, http.HandlerFunc):
|
|
// handlerFunc(w,r, http.HandlerFunc)
|
|
//
|
|
return FromStdWithNext(h)
|
|
default:
|
|
// No valid handler passed
|
|
panic(fmt.Errorf(`
|
|
Passed argument is not a func(iris.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`, 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) {
|
|
next := func(w http.ResponseWriter, r *http.Request) {
|
|
ctx.ResetRequest(r)
|
|
ctx.Next()
|
|
}
|
|
|
|
h(ctx.ResponseWriter(), ctx.Request(), next)
|
|
}
|
|
}
|