Add ToStringMapStruct()

This commit is contained in:
Albert Le Batteux 2018-01-11 13:12:08 +01:00
parent acbeb36b90
commit 60fe4eb110
No known key found for this signature in database
GPG Key ID: 6ECDB1C9555BCE18
3 changed files with 68 additions and 0 deletions

View File

@ -128,6 +128,12 @@ func ToStringMap(i interface{}) map[string]interface{} {
return v return v
} }
// ToStringMapStruct casts an interface to a map[string]struct{} type.
func ToStringMapStruct(i interface{}) map[string]struct{} {
v, _ := ToStringMapStructE(i)
return v
}
// ToSlice casts an interface to a []interface{} type. // ToSlice casts an interface to a []interface{} type.
func ToSlice(i interface{}) []interface{} { func ToSlice(i interface{}) []interface{} {
v, _ := ToSliceE(i) v, _ := ToSliceE(i)

View File

@ -774,6 +774,38 @@ func TestToStringMapE(t *testing.T) {
} }
} }
func TestToStringMapStructE(t *testing.T) {
tests := []struct {
input interface{}
expect map[string]struct{}
iserr bool
}{
{map[string]interface{}{"v1": true, "v2": false}, map[string]struct{}{"v1": struct{}{}, "v2": struct{}{}}, false},
{[]string{"v1", "v2"}, map[string]struct{}{"v1": struct{}{}, "v2": struct{}{}}, false},
{[]interface{}{"v1", "v2"}, map[string]struct{}{"v1": struct{}{}, "v2": struct{}{}}, false},
{map[string]struct{}{"v1": struct{}{}, "v2": struct{}{}}, map[string]struct{}{"v1": struct{}{}, "v2": struct{}{}}, false},
// errors
{nil, nil, true},
{testing.T{}, nil, true},
}
for i, test := range tests {
errmsg := fmt.Sprintf("i = %d", i) // assert helper message
v, err := ToStringMapStructE(test.input)
if test.iserr {
assert.Error(t, err, errmsg)
continue
}
assert.NoError(t, err, errmsg)
assert.Equal(t, test.expect, v, errmsg)
// Non-E test
v = ToStringMapStruct(test.input)
assert.Equal(t, test.expect, v, errmsg)
}
}
func TestToStringMapBoolE(t *testing.T) { func TestToStringMapBoolE(t *testing.T) {
tests := []struct { tests := []struct {
input interface{} input interface{}

View File

@ -977,6 +977,36 @@ func ToStringMapE(i interface{}) (map[string]interface{}, error) {
} }
} }
// ToStringMapStructE casts an interface to a map[string]struct{} type.
func ToStringMapStructE(i interface{}) (map[string]struct{}, error) {
var m = map[string]struct{}{}
fmt.Println("type: ", reflect.TypeOf(i))
switch v := i.(type) {
case map[string]interface{}:
for k := range v {
m[k] = struct{}{}
}
return m, nil
case []string:
fmt.Println("[]string")
for _, s := range v {
m[s] = struct{}{}
}
return m, nil
case []interface{}:
for _, k := range v {
m[ToString(k)] = struct{}{}
}
return m, nil
case map[string]struct{}:
return v, nil
default:
fmt.Println("cast err")
return m, fmt.Errorf("unable to cast %#v of type %T to map[string]interface{}", i, i)
}
}
// ToSliceE casts an interface to a []interface{} type. // ToSliceE casts an interface to a []interface{} type.
func ToSliceE(i interface{}) ([]interface{}, error) { func ToSliceE(i interface{}) ([]interface{}, error) {
var s []interface{} var s []interface{}