Former-commit-id: 129d115937617e4d77b7e6e7efddf3168b15d021
This commit is contained in:
Gerasimos (Makis) Maropoulos 2020-06-18 23:53:53 +03:00
parent 43502ed047
commit 9724592697
3 changed files with 59 additions and 1 deletions

View File

@ -114,6 +114,11 @@ func makeHandler(fn interface{}, c *Container, paramsCount int) context.Handler
inputs[binding.Input.Index] = input
}
// fmt.Printf("For func: %s | valid input deps length(%d)\n", typ.String(), len(inputs))
// for idx, in := range inputs {
// fmt.Printf("[%d] (%s) %#+v\n", idx, in.Type().String(), in.Interface())
// }
outputs := v.Call(inputs)
if err := dispatchFuncResult(ctx, outputs, resultHandler); err != nil {
c.GetErrorHandler(ctx).HandleError(ctx, err)

View File

@ -115,7 +115,8 @@ func (s *Struct) Acquire(ctx context.Context) (reflect.Value, error) {
}
ctrl := ctx.Controller()
if ctrl.Kind() == reflect.Invalid {
if ctrl.Kind() == reflect.Invalid ||
ctrl.Type() != s.ptrType /* in case of changing controller in the same request (see RouteOverlap feature) */ {
ctrl = reflect.New(s.elementType)
ctx.Values().Set(context.ControllerContextKey, ctrl)
elem := ctrl.Elem()

View File

@ -659,3 +659,55 @@ func TestApplicationDependency(t *testing.T) {
e.GET("/").Expect().Status(httptest.StatusOK).Body().Equal("app1")
e.GET("/other").Expect().Status(httptest.StatusOK).Body().Equal("app2")
}
// Authenticated type.
type Authenticated int64
// BasePublicPrivateController base controller between public and private controllers.
type BasePublicPrivateController struct {
CurrentUserID Authenticated
Ctx iris.Context
}
type publicController struct {
Ctx iris.Context
}
// Get desc
// Route / [GET]
func (c *publicController) Get() iris.Map {
return iris.Map{"data": "things"}
}
// privateController serves the "public-private" Customer API.
type privateController struct{ BasePublicPrivateController }
// Get desc
// Route / [GET]
func (c *privateController) Get() iris.Map {
return iris.Map{"id": c.CurrentUserID}
}
func TestControllerOverlapping(t *testing.T) {
app := iris.New()
m := New(app)
m.Router.SetRegisterRule(iris.RouteOverlap)
m.Register(func(ctx iris.Context) Authenticated {
if ctx.URLParam("name") == "kataras" {
return 1
}
ctx.StopWithStatus(iris.StatusForbidden)
return -1
})
// Order matters.
m.Handle(new(privateController))
m.Handle(new(publicController))
e := httptest.New(t, app)
e.GET("/").WithQuery("name", "kataras").Expect().Status(httptest.StatusOK).JSON().Equal(iris.Map{"id": 1})
e.GET("/").Expect().Status(httptest.StatusOK).JSON().Equal(iris.Map{"data": "things"})
}