-
Notifications
You must be signed in to change notification settings - Fork 0
/
flatten.go
39 lines (33 loc) · 922 Bytes
/
flatten.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
// Copyright © 2023 Ory Corp
// SPDX-License-Identifier: Apache-2.0
package jsonx
import (
"encoding/json"
"strconv"
"strings"
"github.com/tidwall/gjson"
)
// Flatten flattens a JSON object using dot notation.
func Flatten(raw json.RawMessage) map[string]interface{} {
parsed := gjson.ParseBytes(raw)
if !parsed.IsObject() {
return nil
}
flattened := make(map[string]interface{})
flatten(parsed, nil, flattened)
return flattened
}
func flatten(parsed gjson.Result, parents []string, flattened map[string]interface{}) {
if parsed.IsObject() {
parsed.ForEach(func(k, v gjson.Result) bool {
flatten(v, append(parents, strings.ReplaceAll(k.String(), ".", "\\.")), flattened)
return true
})
} else if parsed.IsArray() {
for kk, vv := range parsed.Array() {
flatten(vv, append(parents, strconv.Itoa(kk)), flattened)
}
} else {
flattened[strings.Join(parents, ".")] = parsed.Value()
}
}