92 lines
2.2 KiB
Go
92 lines
2.2 KiB
Go
|
package merge
|
||
|
|
|
||
|
|
import (
|
||
|
|
"reflect"
|
||
|
|
|
||
|
|
"github.com/imdario/mergo"
|
||
|
|
"github.com/pkg/errors"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
ErrNonPointerDst = errors.New("dst is not a pointer")
|
||
|
|
ErrUnsupportedMerge = errors.New("unsupported merge")
|
||
|
|
ErrUnexpectedFailedCast = errors.New("unexpected failed cast")
|
||
|
|
)
|
||
|
|
|
||
|
|
func Merge(dst interface{}, sources ...interface{}) error {
|
||
|
|
if reflect.TypeOf(dst).Kind() != reflect.Ptr {
|
||
|
|
return errors.WithStack(ErrNonPointerDst)
|
||
|
|
}
|
||
|
|
|
||
|
|
dstPointedKind := reflect.Indirect(reflect.ValueOf(dst)).Elem().Kind()
|
||
|
|
|
||
|
|
for _, src := range sources {
|
||
|
|
srcKind := reflect.ValueOf(src).Kind()
|
||
|
|
|
||
|
|
switch dstPointedKind {
|
||
|
|
case reflect.Map:
|
||
|
|
if srcKind != dstPointedKind {
|
||
|
|
return errors.WithStack(unsupportedMergeError(dstPointedKind, srcKind))
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := mergeMaps(dst, src); err != nil {
|
||
|
|
return errors.WithStack(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
case reflect.Slice:
|
||
|
|
if srcKind != dstPointedKind {
|
||
|
|
return errors.WithStack(unsupportedMergeError(dstPointedKind, srcKind))
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := mergeSlices(dst, src); err != nil {
|
||
|
|
return errors.WithStack(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
default:
|
||
|
|
return errors.WithStack(unsupportedMergeError(dstPointedKind, srcKind))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func unsupportedMergeError(dstKind reflect.Kind, defaultsKind reflect.Kind) error {
|
||
|
|
return errors.Wrapf(ErrUnsupportedMerge, "could not merge '%s' with defaults '%s'", dstKind, defaultsKind)
|
||
|
|
}
|
||
|
|
|
||
|
|
func mergeMaps(dst interface{}, defaults interface{}) error {
|
||
|
|
dstMap, ok := reflect.Indirect(reflect.ValueOf(dst)).Elem().Interface().(map[string]interface{})
|
||
|
|
if !ok {
|
||
|
|
return errors.WithStack(ErrUnexpectedFailedCast)
|
||
|
|
}
|
||
|
|
|
||
|
|
defaultsMap, ok := defaults.(map[string]interface{})
|
||
|
|
if !ok {
|
||
|
|
return errors.WithStack(ErrUnexpectedFailedCast)
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := mergo.Merge(&dstMap, defaultsMap, mergo.WithOverride); err != nil {
|
||
|
|
return errors.WithStack(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func mergeSlices(dst interface{}, defaults interface{}) error {
|
||
|
|
dstSlice, ok := reflect.Indirect(reflect.ValueOf(dst)).Elem().Interface().([]interface{})
|
||
|
|
if !ok {
|
||
|
|
return errors.WithStack(ErrUnexpectedFailedCast)
|
||
|
|
}
|
||
|
|
|
||
|
|
defaultsSlice, ok := defaults.([]interface{})
|
||
|
|
if !ok {
|
||
|
|
return errors.WithStack(ErrUnexpectedFailedCast)
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := mergo.Merge(&dstSlice, defaultsSlice, mergo.WithOverride); err != nil {
|
||
|
|
return errors.WithStack(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|