Added FlattenMap

xload/loader.go

@@ -25,20 +25,6 @@ func PrefixLoader(prefix string, loader Loader) Loader {
 	})
 }
 
-// MapLoader loads values from a map.
-// Mostly used for testing.
-type MapLoader map[string]string
-
-// Load fetches the value from the map.
-func (m MapLoader) Load(ctx context.Context, key string) (string, error) {
-	value, ok := m[key]
-	if !ok {
-		return "", nil
-	}
-
-	return value, nil
-}
-
 // OSLoader loads values from the OS environment.
 func OSLoader() Loader {
 	return LoaderFunc(func(ctx context.Context, key string) (string, error) {

xload/maps.go

@@ -0,0 +1,46 @@
+package xload
+
+import (
+	"context"
+
+	"github.com/spf13/cast"
+)
+
+// MapLoader loads values from a map.
+//
+// Can be used with xload.FlattenMap as an intermediate format
+// when loading from various sources.
+type MapLoader map[string]string
+
+// Load fetches the value from the map.
+func (m MapLoader) Load(ctx context.Context, key string) (string, error) {
+	value, ok := m[key]
+	if !ok {
+		return "", nil
+	}
+
+	return value, nil
+}
+
+// FlattenMap flattens a map[string]interface{} into a map[string]string.
+// Nested maps are flattened using given separator.
+func FlattenMap(m map[string]interface{}, sep string) map[string]string {
+	return flatten(m, "", sep)
+}
+
+func flatten(m map[string]interface{}, prefix string, sep string) map[string]string {
+	flattened := make(map[string]string)
+
+	for key, value := range m {
+		switch value := value.(type) {
+		case map[string]interface{}:
+			for k, v := range flatten(value, key+sep, sep) {
+				flattened[prefix+k] = v
+			}
+		default:
+			flattened[prefix+key] = cast.ToString(value)
+		}
+	}
+
+	return flattened
+}

xload/maps_test.go

@@ -0,0 +1,32 @@
+package xload
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestFlattenMap(t *testing.T) {
+	input := map[string]interface{}{
+		"NAME":    "xload",
+		"VERSION": 1.1,
+		"AUTHOR": map[string]interface{}{
+			"NAME":  "gojek",
+			"EMAIL": "[email protected]",
+			"ADDRESS": map[string]interface{}{
+				"CITY": "Bombay",
+			},
+		},
+	}
+
+	want := map[string]string{
+		"NAME":                "xload",
+		"VERSION":             "1.1",
+		"AUTHOR_NAME":         "gojek",
+		"AUTHOR_EMAIL":        "[email protected]",
+		"AUTHOR_ADDRESS_CITY": "Bombay",
+	}
+
+	got := FlattenMap(input, "_")
+	assert.EqualValues(t, want, got)
+}

xload/providers/yaml/go.mod

@@ -9,12 +9,10 @@ require (
 	gopkg.in/yaml.v3 v3.0.1
 )
 
-require (
-	github.com/spf13/cast v1.5.1
-	github.com/stretchr/testify v1.8.4
-)
+require github.com/stretchr/testify v1.8.4
 
 require (
 	github.com/davecgh/go-spew v1.1.1 // indirect
 	github.com/pmezard/go-difflib v1.0.0 // indirect
+	github.com/spf13/cast v1.5.1 // indirect
 )

xload/providers/yaml/yaml.go

@@ -4,14 +4,13 @@ package yaml
 import (
 	"io"
 
-	"github.com/spf13/cast"
 	"gopkg.in/yaml.v3"
 
 	"github.com/gojekfarm/xtools/xload"
 )
 
 // NewLoader reads YAML from the given io.Reader and returns a xload.Loader
-func NewLoader(r io.Reader, delim string) (xload.Loader, error) {
+func NewLoader(r io.Reader, sep string) (xload.Loader, error) {
 	b, err := io.ReadAll(r)
 	if err != nil {
 		return nil, err
@@ -24,22 +23,5 @@ func NewLoader(r io.Reader, delim string) (xload.Loader, error) {
 		return nil, err
 	}
 
-	return xload.MapLoader(flatten("", delim, out)), nil
-}
-
-func flatten(prefix, delim string, data map[string]interface{}) map[string]string {
-	flattened := make(map[string]string)
-
-	for key, value := range data {
-		switch value := value.(type) {
-		case map[string]interface{}:
-			for k, v := range flatten(key+delim, delim, value) {
-				flattened[prefix+k] = v
-			}
-		default:
-			flattened[prefix+key] = cast.ToString(value)
-		}
-	}
-
-	return flattened
+	return xload.MapLoader(xload.FlattenMap(out, sep)), nil
 }