mirror of
https://github.com/kataras/iris.git
synced 2025-01-24 19:21:03 +01:00
55 lines
1.3 KiB
Go
55 lines
1.3 KiB
Go
|
package activator
|
||
|
|
|
||
|
|
import (
|
||
|
|
"reflect"
|
||
|
|
|
||
|
|
"github.com/kataras/iris/context"
|
||
|
|
)
|
||
|
|
|
||
|
|
type modelControl struct {
|
||
|
|
fields []field
|
||
|
|
}
|
||
|
|
|
||
|
|
func (mc *modelControl) Load(t *TController) error {
|
||
|
|
fields := lookupFields(t, func(f reflect.StructField) bool {
|
||
|
|
if tag, ok := f.Tag.Lookup("iris"); ok {
|
||
|
|
if tag == "model" {
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return false
|
||
|
|
})
|
||
|
|
|
||
|
|
if len(fields) == 0 {
|
||
|
|
// first is the `Controller` so we need to
|
||
|
|
// check the second and after that.
|
||
|
|
return ErrControlSkip
|
||
|
|
}
|
||
|
|
|
||
|
|
mc.fields = fields
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (mc *modelControl) Handle(ctx context.Context, c reflect.Value, methodFunc func()) {
|
||
|
|
elem := c.Elem() // controller should always be a pointer at this state
|
||
|
|
|
||
|
|
for _, f := range mc.fields {
|
||
|
|
elemField := elem.Field(f.Index)
|
||
|
|
// check if current controller's element field
|
||
|
|
// is valid, is not nil and it's type is the same (should be but make that check to be sure).
|
||
|
|
if !elemField.IsValid() || (elemField.Kind() == reflect.Ptr && elemField.IsNil()) || elemField.Type() != f.Type {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
fieldValue := elemField.Interface()
|
||
|
|
// fmt.Printf("setting %s to %#v", f.Name, fieldValue)
|
||
|
|
ctx.ViewData(f.Name, fieldValue)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// ModelControl returns a TControl which is responsible
|
||
|
|
// to load and handle the `Model(s)` inside a controller struct
|
||
|
|
// via the `iris:"model"` tag field.
|
||
|
|
func ModelControl() TControl {
|
||
|
|
return &modelControl{}
|
||
|
|
}
|