simplify by join the bind registration(ctx-transformer-to-something-func-binder and service one, which just sets the struct as it's) to one named 'In' and create a 'Child' which will return a new mvc instance with binders inheritanced from the parent one and add a simple test to the mvc_test.go - will have more later on

Former-commit-id: 81ae99390c683a61e1b0bac58725a04b9a3eebbb
This commit is contained in:
kataras 2017-11-24 17:34:35 +02:00
parent 29835d9a8e
commit 5a3be2ab58
5 changed files with 111 additions and 56 deletions

View File

@ -68,29 +68,41 @@ func MustMakeFuncInputBinder(binder interface{}) *InputBinder {
return b return b
} }
type binderType uint32
const (
functionType binderType = iota
serviceType
invalidType
)
func resolveBinderType(binder interface{}) binderType {
if binder == nil {
return invalidType
}
switch indirectTyp(reflect.TypeOf(binder)).Kind() {
case reflect.Func:
return functionType
case reflect.Struct:
return serviceType
}
return invalidType
}
// MakeFuncInputBinder takes a binder function or a struct which contains a "Bind" // MakeFuncInputBinder takes a binder function or a struct which contains a "Bind"
// function and returns an `InputBinder`, which Iris uses to // function and returns an `InputBinder`, which Iris uses to
// resolve and set the input parameters when a handler is executed. // resolve and set the input parameters when a handler is executed.
// //
// The "binder" can have the following form: // The "binder" can have the following form:
// `func(iris.Context) UserViewModel` // `func(iris.Context) UserViewModel`.
// and a struct which contains a "Bind" method
// of the same binder form that was described above.
// //
// The return type of the "binder" should be a value instance, not a pointer, for your own protection. // The return type of the "binder" should be a value instance, not a pointer, for your own protection.
// The binder function should return only one value and // The binder function should return only one value and
// it can accept only one input argument, the Iris' Context (`context.Context` or `iris.Context`). // it can accept only one input argument, the Iris' Context (`context.Context` or `iris.Context`).
func MakeFuncInputBinder(binder interface{}) (*InputBinder, error) { func MakeFuncInputBinder(binder interface{}) (*InputBinder, error) {
v := reflect.ValueOf(binder) v := reflect.ValueOf(binder)
// check if it's a struct or a pointer to a struct
// and contains a "Bind" method, if yes use that as the binder func.
if indirectTyp(v.Type()).Kind() == reflect.Struct {
if m := v.MethodByName("Bind"); m.IsValid() && m.CanInterface() {
v = m
}
}
return makeFuncInputBinder(v) return makeFuncInputBinder(v)
} }

View File

@ -22,15 +22,8 @@ func testBinderFunc(ctx context.Context) testUserStruct {
} }
} }
type testBinderStruct struct{}
func (t *testBinderStruct) Bind(ctx context.Context) testUserStruct {
return testBinderFunc(ctx)
}
func TestMakeFuncInputBinder(t *testing.T) { func TestMakeFuncInputBinder(t *testing.T) {
testMakeFuncInputBinder(t, testBinderFunc) testMakeFuncInputBinder(t, testBinderFunc)
testMakeFuncInputBinder(t, new(testBinderStruct))
} }
func testMakeFuncInputBinder(t *testing.T, binder interface{}) { func testMakeFuncInputBinder(t *testing.T, binder interface{}) {

View File

@ -40,39 +40,51 @@ func (s *testServiceImpl) Say(message string) string {
return s.prefix + " " + message return s.prefix + " " + message
} }
var (
// binders, as user-defined
testBinderFuncUserStruct = testBinderFunc
testBinderService = &testServiceImpl{prefix: "say"}
testBinderFuncParam = func(ctx iris.Context) string {
return ctx.Params().Get("param")
}
// consumers
// a context as first input arg, which is not needed to be binded manually,
// and a user struct which is binded to the input arg by the #1 func(ctx) any binder.
testConsumeUserHandler = func(ctx iris.Context, user testUserStruct) {
ctx.JSON(user)
}
// just one input arg, the service which is binded by the #2 service binder.
testConsumeServiceHandler = func(service testService) string {
return service.Say("something")
}
// just one input arg, a standar string which is binded by the #3 func(ctx) any binder.
testConsumeParamHandler = func(myParam string) string {
return "param is: " + myParam
}
)
func TestMakeHandler(t *testing.T) { func TestMakeHandler(t *testing.T) {
binders := []*InputBinder{ binders := []*InputBinder{
// #1 // #1
MustMakeFuncInputBinder(testBinderFunc), MustMakeFuncInputBinder(testBinderFuncUserStruct),
// #2 // #2
MustMakeServiceInputBinder(&testServiceImpl{prefix: "say"}), MustMakeServiceInputBinder(testBinderService),
// #3 // #3
MustMakeFuncInputBinder(func(ctx iris.Context) string { MustMakeFuncInputBinder(testBinderFuncParam),
return ctx.Params().Get("param")
}),
} }
var ( var (
// a context as first input arg, which is not needed to be binded manually, h1 = MustMakeHandler(testConsumeUserHandler, binders)
// and a user struct which is binded to the input arg by the #1 func(ctx) any binder. h2 = MustMakeHandler(testConsumeServiceHandler, binders)
consumeUserHandler = func(ctx iris.Context, user testUserStruct) { h3 = MustMakeHandler(testConsumeParamHandler, binders)
ctx.JSON(user)
}
h1 = MustMakeHandler(consumeUserHandler, binders)
// just one input arg, the service which is binded by the #2 service binder.
consumeServiceHandler = func(service testService) string {
return service.Say("something")
}
h2 = MustMakeHandler(consumeServiceHandler, binders)
// just one input arg, a standar string which is binded by the #3 func(ctx) any binder.
consumeParamHandler = func(myParam string) string {
return "param is: " + myParam
}
h3 = MustMakeHandler(consumeParamHandler, binders)
) )
testAppWithMvcHandlers(t, h1, h2, h3)
}
func testAppWithMvcHandlers(t *testing.T, h1, h2, h3 iris.Handler) {
app := iris.New() app := iris.New()
app.Get("/{id:long}/{username:string}", h1) app.Get("/{id:long}/{username:string}", h1)
app.Get("/service", h2) app.Get("/service", h2)

View File

@ -20,28 +20,44 @@ func New() *Mvc {
return new(Mvc) return new(Mvc)
} }
func (m *Mvc) RegisterBinder(binders ...interface{}) error { func (m *Mvc) Child() *Mvc {
for _, binder := range binders { child := New()
b, err := MakeFuncInputBinder(binder)
if err != nil { // copy the current parent's ctx func binders and services to this new child.
return err if len(m.binders) > 0 {
binders := make([]*InputBinder, len(m.binders), len(m.binders))
for i, v := range m.binders {
binders[i] = v
} }
m.binders = append(m.binders, b) child.binders = binders
} }
return nil return child
} }
func (m *Mvc) RegisterService(services ...interface{}) error { func (m *Mvc) In(binders ...interface{}) {
for _, service := range services { for _, binder := range binders {
b, err := MakeServiceInputBinder(service) typ := resolveBinderType(binder)
if err != nil {
return err var (
b *InputBinder
err error
)
if typ == functionType {
b, err = MakeFuncInputBinder(binder)
} else if typ == serviceType {
b, err = MakeServiceInputBinder(binder)
} else {
err = errBad
} }
if err != nil {
continue
}
m.binders = append(m.binders, b) m.binders = append(m.binders, b)
} }
return nil
} }
func (m *Mvc) Handler(handler interface{}) context.Handler { func (m *Mvc) Handler(handler interface{}) context.Handler {

22
mvc2/mvc_test.go Normal file
View File

@ -0,0 +1,22 @@
package mvc2_test
// black-box in combination with the handler_test
import (
"testing"
. "github.com/kataras/iris/mvc2"
)
func TestMvcInAndHandler(t *testing.T) {
m := New()
m.In(testBinderFuncUserStruct, testBinderService, testBinderFuncParam)
var (
h1 = m.Handler(testConsumeUserHandler)
h2 = m.Handler(testConsumeServiceHandler)
h3 = m.Handler(testConsumeParamHandler)
)
testAppWithMvcHandlers(t, h1, h2, h3)
}