2024-01-23 17:51:52 +01:00
|
|
|
package reflectext
|
|
|
|
|
2024-03-18 11:19:01 +01:00
|
|
|
import (
|
|
|
|
"encoding/json"
|
|
|
|
"reflect"
|
|
|
|
)
|
2024-01-23 17:51:52 +01:00
|
|
|
|
2024-03-18 11:19:01 +01:00
|
|
|
type ConvertStructToMapOpt struct {
|
|
|
|
KeepJsonMarshalTypes bool
|
2024-01-23 17:51:52 +01:00
|
|
|
}
|
|
|
|
|
2024-03-18 11:19:01 +01:00
|
|
|
func ConvertStructToMap(v any, opts ...ConvertStructToMapOpt) any {
|
|
|
|
opt := ConvertStructToMapOpt{}
|
|
|
|
if len(opts) > 0 {
|
|
|
|
opt = opts[0]
|
|
|
|
}
|
|
|
|
return reflectToMap(reflect.ValueOf(v), opt)
|
|
|
|
}
|
|
|
|
|
|
|
|
func reflectToMap(fv reflect.Value, opt ConvertStructToMapOpt) any {
|
2024-01-23 17:51:52 +01:00
|
|
|
|
|
|
|
if fv.Kind() == reflect.Ptr {
|
|
|
|
|
|
|
|
if fv.IsNil() {
|
|
|
|
return nil
|
|
|
|
} else {
|
2024-03-18 11:19:01 +01:00
|
|
|
return reflectToMap(fv.Elem(), opt)
|
2024-01-23 17:51:52 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
if fv.Kind() == reflect.Func {
|
|
|
|
|
|
|
|
// skip
|
|
|
|
return nil
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
if fv.Kind() == reflect.Array {
|
|
|
|
|
|
|
|
arrlen := fv.Len()
|
|
|
|
arr := make([]any, arrlen)
|
|
|
|
for i := 0; i < arrlen; i++ {
|
2024-03-18 11:19:01 +01:00
|
|
|
arr[i] = reflectToMap(fv.Index(i), opt)
|
2024-01-23 17:51:52 +01:00
|
|
|
}
|
|
|
|
return arr
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
if fv.Kind() == reflect.Slice {
|
|
|
|
|
|
|
|
arrlen := fv.Len()
|
|
|
|
arr := make([]any, arrlen)
|
|
|
|
for i := 0; i < arrlen; i++ {
|
2024-03-18 11:19:01 +01:00
|
|
|
arr[i] = reflectToMap(fv.Index(i), opt)
|
2024-01-23 17:51:52 +01:00
|
|
|
}
|
|
|
|
return arr
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
if fv.Kind() == reflect.Chan {
|
|
|
|
|
|
|
|
// skip
|
|
|
|
return nil
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
if fv.Kind() == reflect.Struct {
|
|
|
|
|
2024-03-18 11:19:01 +01:00
|
|
|
if opt.KeepJsonMarshalTypes && fv.Type().Implements(reflect.TypeFor[json.Marshaler]()) {
|
|
|
|
return fv.Interface()
|
|
|
|
}
|
|
|
|
|
2024-01-23 17:51:52 +01:00
|
|
|
res := make(map[string]any)
|
|
|
|
|
|
|
|
for i := 0; i < fv.NumField(); i++ {
|
|
|
|
if fv.Type().Field(i).IsExported() {
|
2024-03-18 11:19:01 +01:00
|
|
|
res[fv.Type().Field(i).Name] = reflectToMap(fv.Field(i), opt)
|
2024-01-23 17:51:52 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return res
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
return fv.Interface()
|
|
|
|
}
|