mirror of
https://github.com/kataras/iris.git
synced 2025-01-23 18:51:03 +01:00
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:
parent
29835d9a8e
commit
5a3be2ab58
|
@ -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)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -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{}) {
|
||||||
|
|
|
@ -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)
|
||||||
|
|
44
mvc2/mvc.go
44
mvc2/mvc.go
|
@ -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
22
mvc2/mvc_test.go
Normal 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)
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user