removed tiedot, introduced ledisdb

utils/flat.go

@@ -0,0 +1,130 @@
+// Adapted from https://github.com/nytlabs/gojsonexplode
+
+package utils
+
+import (
+	"strconv"
+	"encoding/json"
+)
+
+const delimiter = "."
+
+func explodeList(l []interface{}, parent string, delimiter string) (map[string]interface{}, error) {
+	var err error
+	var key string
+	j := make(map[string]interface{})
+	for k, i := range l {
+		if len(parent) > 0 {
+			key = parent + delimiter + strconv.Itoa(k)
+		} else {
+			key = strconv.Itoa(k)
+		}
+		switch v := i.(type) {
+		case nil:
+			j[key] = v
+		case int:
+			j[key] = v
+		case float64:
+			j[key] = v
+		case string:
+			j[key] = v
+		case bool:
+			j[key] = v
+		case []interface{}:
+			out := make(map[string]interface{})
+			out, err = explodeList(v, key, delimiter)
+			if err != nil {
+				return nil, err
+			}
+			for newkey, value := range out {
+				j[newkey] = value
+			}
+		case map[string]interface{}:
+			out := make(map[string]interface{})
+			out, err = explodeMap(v, key, delimiter)
+			if err != nil {
+				return nil, err
+			}
+			for newkey, value := range out {
+				j[newkey] = value
+			}
+		default:
+		// do nothing
+		}
+	}
+	return j, nil
+}
+
+func explodeMap(m map[string]interface{}, parent string, delimiter string) (map[string]interface{}, error) {
+	var err error
+	j := make(map[string]interface{})
+	for k, i := range m {
+		if len(parent) > 0 {
+			k = parent + delimiter + k
+		}
+		switch v := i.(type) {
+		case nil:
+			j[k] = v
+		case int:
+			j[k] = v
+		case float64:
+			j[k] = v
+		case string:
+			j[k] = v
+		case bool:
+			j[k] = v
+		case []interface{}:
+			out := make(map[string]interface{})
+			out, err = explodeList(v, k, delimiter)
+			if err != nil {
+				return nil, err
+			}
+			for key, value := range out {
+				j[key] = value
+			}
+		case map[string]interface{}:
+			out := make(map[string]interface{})
+			out, err = explodeMap(v, k, delimiter)
+			if err != nil {
+				return nil, err
+			}
+			for key, value := range out {
+				j[key] = value
+			}
+		default:
+		//nothing
+		}
+	}
+	return j, nil
+}
+
+func FlattenMap(input map[string]interface{}) (map[string]interface{}, error) {
+	var flattened map[string]interface{}
+	var err error
+	flattened, err = explodeMap(input, "", delimiter)
+	if err != nil {
+		return nil, err
+	}
+	return flattened, nil
+}
+
+func marshal(rec interface{}) (map[string]interface{}, error) {
+	// Convert to JSON...
+	b, err := json.Marshal(rec);
+	if err != nil {
+		return nil, err
+	}
+
+	// ... then convert to map
+	var m map[string]interface{}
+	err = json.Unmarshal(b, &m)
+	return m, err
+}
+
+func Flatten(input interface{}) (map[string]interface{}, error) {
+	m, err := marshal(input)
+	if err != nil {
+		return nil, err
+	}
+	return FlattenMap(m)
+}