Passing unmarshal of array table into interfaces

This commit is contained in:
Thomas Pelletier
2021-03-18 08:47:50 -04:00
parent 939f889666
commit cb678e6221
4 changed files with 70 additions and 45 deletions
+61 -36
View File
@@ -58,6 +58,35 @@ func (t valueTarget) setFloat64(v float64) error {
return nil return nil
} }
// interfaceTarget wraps an other target to dereference on get.
type interfaceTarget struct {
x target
}
func (t interfaceTarget) get() reflect.Value {
return t.x.get().Elem()
}
func (t interfaceTarget) set(v reflect.Value) error {
return t.x.set(v)
}
func (t interfaceTarget) setString(v string) error {
return t.x.setString(v)
}
func (t interfaceTarget) setBool(v bool) error {
return t.x.setBool(v)
}
func (t interfaceTarget) setInt64(v int64) error {
return t.x.setInt64(v)
}
func (t interfaceTarget) setFloat64(v float64) error {
return t.x.setFloat64(v)
}
// mapTarget targets a specific key of a map. // mapTarget targets a specific key of a map.
type mapTarget struct { type mapTarget struct {
v reflect.Value v reflect.Value
@@ -199,66 +228,62 @@ func pushNew(t target) (target, error) {
func scopeTableTarget(append bool, t target, name string) (target, error) { func scopeTableTarget(append bool, t target, name string) (target, error) {
x := t.get() x := t.get()
if x.Kind() == reflect.Interface { switch x.Kind() {
t, err := initInterface(append, t) case reflect.Interface:
t, err := scopeInterface(append, t)
if err != nil { if err != nil {
return t, err return t, err
} }
x = t.get() return scopeTableTarget(append, t, name)
case reflect.Struct:
return scopeStruct(x, name)
case reflect.Map:
return scopeMap(x, name)
case reflect.Slice:
return scopeSlice(append, t)
default:
panic(fmt.Errorf("can't scope on a %s", x.Kind()))
} }
return t, nil
}
if x.Kind() == reflect.Slice { func scopeInterface(append bool, t target) (target, error) {
return scopeSlice(t, append) err := initInterface(append, t)
}
t, err := scope(x, name)
if err != nil { if err != nil {
return t, err return t, err
} }
return t, nil return interfaceTarget{t}, nil
} }
// initInterface makes sure that the interface pointed at by the target is not // initInterface makes sure that the interface pointed at by the target is not
// nil. // nil.
// Returns the target to the initialized value of the target. // Returns the target to the initialized value of the target.
func initInterface(append bool, t target) (target, error) { func initInterface(append bool, t target) error {
x := t.get() x := t.get()
if x.Kind() != reflect.Interface { if x.Kind() != reflect.Interface {
panic("this should only be called on interfaces") panic("this should only be called on interfaces")
} }
if x.IsNil() { if !x.IsNil() {
var newElement reflect.Value return nil
if append {
newElement = reflect.MakeSlice(reflect.TypeOf([]interface{}{}), 0, 0)
} else {
newElement = reflect.MakeMap(reflect.TypeOf(map[string]interface{}{}))
}
err := t.set(newElement)
if err != nil {
return t, err
}
x = t.get()
} }
x = x.Elem() var newElement reflect.Value
t = valueTarget(x) if append {
return t, nil newElement = reflect.MakeSlice(reflect.TypeOf([]interface{}{}), 0, 0)
} } else {
newElement = reflect.MakeMap(reflect.TypeOf(map[string]interface{}{}))
func scope(v reflect.Value, name string) (target, error) {
switch v.Kind() {
case reflect.Struct:
return scopeStruct(v, name)
case reflect.Map:
return scopeMap(v, name)
default:
panic(fmt.Errorf("can't scope on a %s", v.Kind()))
} }
err := t.set(newElement)
if err != nil {
return err
}
return nil
} }
func scopeSlice(t target, append bool) (target, error) { func scopeSlice(append bool, t target) (target, error) {
v := t.get() v := t.get()
if append { if append {
+5 -5
View File
@@ -39,7 +39,7 @@ func TestStructTarget_Ensure(t *testing.T) {
for _, e := range examples { for _, e := range examples {
t.Run(e.desc, func(t *testing.T) { t.Run(e.desc, func(t *testing.T) {
target, err := scope(e.input, e.name) target, err := scopeTableTarget(false, valueTarget(e.input), e.name)
require.NoError(t, err) require.NoError(t, err)
err = ensureSlice(target) err = ensureSlice(target)
v := target.get() v := target.get()
@@ -86,7 +86,7 @@ func TestStructTarget_SetString(t *testing.T) {
for _, e := range examples { for _, e := range examples {
t.Run(e.desc, func(t *testing.T) { t.Run(e.desc, func(t *testing.T) {
target, err := scope(e.input, e.name) target, err := scopeTableTarget(false, valueTarget(e.input), e.name)
require.NoError(t, err) require.NoError(t, err)
err = setString(target, str) err = setString(target, str)
v := target.get() v := target.get()
@@ -102,7 +102,7 @@ func TestPushNew(t *testing.T) {
} }
d := Doc{} d := Doc{}
x, err := scope(reflect.ValueOf(&d).Elem(), "A") x, err := scopeTableTarget(false, valueTarget(reflect.ValueOf(&d).Elem()), "A")
require.NoError(t, err) require.NoError(t, err)
n, err := pushNew(x) n, err := pushNew(x)
@@ -122,7 +122,7 @@ func TestPushNew(t *testing.T) {
} }
d := Doc{} d := Doc{}
x, err := scope(reflect.ValueOf(&d).Elem(), "A") x, err := scopeTableTarget(false, valueTarget(reflect.ValueOf(&d).Elem()), "A")
require.NoError(t, err) require.NoError(t, err)
n, err := pushNew(x) n, err := pushNew(x)
@@ -161,7 +161,7 @@ func TestScope_Struct(t *testing.T) {
for _, e := range examples { for _, e := range examples {
t.Run(e.desc, func(t *testing.T) { t.Run(e.desc, func(t *testing.T) {
x, err := scope(e.input, e.name) x, err := scopeTableTarget(false, valueTarget(e.input), e.name)
if e.err { if e.err {
require.Error(t, err) require.Error(t, err)
} else { } else {
+2 -2
View File
@@ -95,7 +95,7 @@ func scopeWithArrayTable(x target, key []ast.Node) (target, error) {
v := x.get() v := x.get()
if v.Kind() == reflect.Interface { if v.Kind() == reflect.Interface {
x, err = initInterface(true, x) x, err = scopeInterface(true, x)
if err != nil { if err != nil {
return x, err return x, err
} }
@@ -103,7 +103,7 @@ func scopeWithArrayTable(x target, key []ast.Node) (target, error) {
} }
if v.Kind() == reflect.Slice { if v.Kind() == reflect.Slice {
return scopeSlice(x, true) return scopeSlice(true, x)
} }
return x, err return x, err
+2 -2
View File
@@ -439,12 +439,12 @@ B = "data"`,
"Products": []interface{}{ "Products": []interface{}{
map[string]interface{}{ map[string]interface{}{
"Name": "Hammer", "Name": "Hammer",
"Sku": 738594937, "Sku": int64(738594937),
}, },
nil, nil,
map[string]interface{}{ map[string]interface{}{
"Name": "Nail", "Name": "Nail",
"Sku": 284758393, "Sku": int64(284758393),
"Color": "gray", "Color": "gray",
}, },
}, },