Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
Former-commit-id: 129d115937617e4d77b7e6e7efddf3168b15d021
  • Loading branch information
kataras committed Jun 18, 2020
1 parent 43502ed commit 9724592
Show file tree
Hide file tree
Showing 3 changed files with 59 additions and 1 deletion.
5 changes: 5 additions & 0 deletions hero/handler.go
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down
3 changes: 2 additions & 1 deletion hero/struct.go
Original file line number Diff line number Diff line change
Expand Up @@ -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()
Expand Down
52 changes: 52 additions & 0 deletions mvc/controller_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -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"})
}

0 comments on commit 9724592

Please sign in to comment.