Golang笔记

Golang Json/map 合并/update

把两个map 合并,功能与python 的update 功能相同

package main

import (
	"encoding/json"
	"fmt"
	"reflect"
)

var jsonMergeDepth = 32

func main() {
	buf1 := []byte(`{"a":1,"b":2}`)
	buf2 := []byte(`{"c":3,"d":4,"a":"aaa"}`)

	var m1, m2 map[string]interface{}

	json.Unmarshal(buf1, &m1)
	json.Unmarshal(buf2, &m2)

	merged := JsonMerge(m1, m2)

	fmt.Println(merged)
}

func JsonMerge(dst, src map[string]interface{}) map[string]interface{} {
	return jsMerge(dst, src, 0)
}

func jsMerge(dst, src map[string]interface{}, depth int) map[string]interface{} {

	if depth > jsonMergeDepth {
		return dst
		// panic("too deep!")
	}

	for key, srcVal := range src {

		if dstVal, ok := dst[key]; ok {

			srcMap, srcMapOk := jsMapify(srcVal)
			dstMap, dstMapOk := jsMapify(dstVal)

			if srcMapOk && dstMapOk {
				srcVal = jsMerge(dstMap, srcMap, depth+1)
			}
		}

		dst[key] = srcVal
	}

	return dst
}

func jsMapify(i interface{}) (map[string]interface{}, bool) {

	value := reflect.ValueOf(i)

	if value.Kind() == reflect.Map {

		m := map[string]interface{}{}

		for _, k := range value.MapKeys() {
			m[k.String()] = value.MapIndex(k).Interface()
		}

		return m, true
	}

	return map[string]interface{}{}, false
}

注意,若有重复的key,会把原来key 值覆盖,上面代码输出:

map[a:aaa b:2 c:3 d:4]

参考 https://github.com/peterbourgon/mergemap 46

本文网址: https://golangnote.com/topic/209.html (转载注明出处)
关于GolangNote:记录在工作中使用golang 遇到、面临的相关问题及解决方法。如果你在这里获得一些知识或信息,解决你的编程问题,请考虑捐赠给不幸的人或者你喜欢的慈善机构,除捐赠外,种植树木、志愿服务或减少排碳的行为也很有益处。如果你有任何问题可以在下面 留言
Be the first to comment!
Captcha image
Relative Articles