-
Notifications
You must be signed in to change notification settings - Fork 1
/
util.go
80 lines (69 loc) · 1.99 KB
/
util.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
package apikit
import (
"reflect"
"fmt"
)
var _ = fmt.Println
func implementsRESTObject(obj interface{}) bool {
t := reflect.TypeOf(obj)
return t.Implements(reflect.TypeOf((*RESTObject)(nil)).Elem())
}
func implementsUser(obj interface{}) bool {
t := reflect.TypeOf(obj)
return t.Implements(reflect.TypeOf((*User)(nil)).Elem())
}
func implementsModelProvider(obj interface{}) bool {
var v reflect.Value = reflect.ValueOf(obj)
return v.Type().Implements(reflect.TypeOf((*RESTController)(nil)).Elem())
}
func embedsRESTController(obj interface{}) bool {
return getEmbeddedRESTController(obj) != nil
}
func getEmbeddedRESTController(obj interface{}) *GenericRESTController {
var theStruct reflect.Value
t := reflect.TypeOf(obj)
if t.Kind() == reflect.Struct {
theStruct = reflect.ValueOf(obj)
} else if t.Kind() == reflect.Ptr {
dereferenced := reflect.Indirect(reflect.ValueOf(obj))
if dereferenced.Kind() == reflect.Struct {
theStruct = dereferenced
} else {
return nil
}
} else {
return nil
}
for fieldIdx := 0; fieldIdx < theStruct.NumField(); fieldIdx ++ {
field := theStruct.Type().Field(fieldIdx)
if field.Name == RESTControllerName && field.Anonymous {
ctrl := theStruct.FieldByIndex(field.Index).Interface().(GenericRESTController)
return &ctrl
}
}
return nil
}
func setEmbeddedRESTController(obj interface{}, ctrl GenericRESTController) (ok bool) {
var theStruct reflect.Value
t := reflect.TypeOf(obj)
if t.Kind() == reflect.Struct {
theStruct = reflect.ValueOf(obj)
} else if t.Kind() == reflect.Ptr {
dereferenced := reflect.Indirect(reflect.ValueOf(obj))
if dereferenced.Kind() == reflect.Struct {
theStruct = dereferenced
} else {
return false
}
} else {
return false
}
for fieldIdx := 0; fieldIdx < theStruct.NumField(); fieldIdx ++ {
field := theStruct.Type().Field(fieldIdx)
if field.Name == RESTControllerName && field.Anonymous {
theStruct.FieldByIndex(field.Index).Set(reflect.ValueOf(ctrl))
return true
}
}
return false
}