Adding upstream version 1.1.1.upstream/1.1.1

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 47 insertions, 0 deletions

diff --git a/pkg/flatten/flatten.go b/pkg/flatten/flatten.go
new file mode 100644
index 0000000..94a6e7e
--- /dev/null
+++ b/pkg/flatten/flatten.go
@@ -0,0 +1,47 @@
+package flatten
+
+import (
+	"database/sql"
+	"fmt"
+	"github.com/icinga/icingadb/pkg/types"
+	"strconv"
+)
+
+// Flatten creates flat, one-dimensional maps from arbitrarily nested values, e.g. JSON.
+func Flatten(value interface{}, prefix string) map[string]types.String {
+	var flatten func(string, interface{})
+	flattened := make(map[string]types.String)
+
+	flatten = func(key string, value interface{}) {
+		switch value := value.(type) {
+		case map[string]interface{}:
+			if len(value) == 0 {
+				flattened[key] = types.String{}
+				break
+			}
+
+			for k, v := range value {
+				flatten(key+"."+k, v)
+			}
+		case []interface{}:
+			if len(value) == 0 {
+				flattened[key] = types.String{}
+				break
+			}
+
+			for i, v := range value {
+				flatten(key+"["+strconv.Itoa(i)+"]", v)
+			}
+		default:
+			val := "null"
+			if value != nil {
+				val = fmt.Sprintf("%v", value)
+			}
+			flattened[key] = types.String{NullString: sql.NullString{String: val, Valid: true}}
+		}
+	}
+
+	flatten(prefix, value)
+
+	return flattened
+}