Hunter0x7c7
2022-08-11 a82f9cb69f63aaeba40c024960deda7d75b9fece
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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
// Copyright 2020 Jebbs. All rights reserved.
// Use of this source code is governed by MIT
// license that can be found in the LICENSE file.
 
package merge
 
const (
    priorityKey string = "_priority"
    tagKey      string = "_tag"
)
 
// ApplyRules applies merge rules according to _tag, _priority fields, and remove them
func ApplyRules(m map[string]interface{}) error {
    err := sortMergeSlices(m)
    if err != nil {
        return err
    }
    removeHelperFields(m)
    return nil
}
 
// sortMergeSlices enumerates all slices in a map, to sort by priority and merge by tag
func sortMergeSlices(target map[string]interface{}) error {
    for key, value := range target {
        if slice, ok := value.([]interface{}); ok {
            sortByPriority(slice)
            s, err := mergeSameTag(slice)
            if err != nil {
                return err
            }
            target[key] = s
            for _, item := range s {
                if m, ok := item.(map[string]interface{}); ok {
                    sortMergeSlices(m)
                }
            }
        } else if field, ok := value.(map[string]interface{}); ok {
            sortMergeSlices(field)
        }
    }
    return nil
}
 
func removeHelperFields(target map[string]interface{}) {
    for key, value := range target {
        if key == priorityKey || key == tagKey {
            delete(target, key)
        } else if slice, ok := value.([]interface{}); ok {
            for _, e := range slice {
                if el, ok := e.(map[string]interface{}); ok {
                    removeHelperFields(el)
                }
            }
        } else if field, ok := value.(map[string]interface{}); ok {
            removeHelperFields(field)
        }
    }
}