golang map 合并|极客教程

golang map 合并

在 Go 语言中，map 是一种无序的键值对集合。有时候我们需要将多个 map 合并为一个新的 map，本文将详细介绍如何实现这一操作。

方法一：逐个插入

最直接的方法是遍历每个 map，并将其中的键值对逐个插入到新的 map 中。

package main

import "fmt"

func mergeMaps(maps ...map[string]int) map[string]int {
    merged := make(map[string]int)

    for _, m := range maps {
        for k, v := range m {
            merged[k] = v
        }
    }

    return merged
}

func main() {
    map1 := map[string]int{"a": 1, "b": 2}
    map2 := map[string]int{"b": 3, "c": 4}

    mergedMap := mergeMaps(map1, map2)

    fmt.Println(mergedMap)
}

运行结果：

map[a:1 b:3 c:4]

方法二：使用 for-range 遍历

另一种方法是通过使用 for-range 遍历 map，并将其中的键值对插入到新的 map 中。

package main

import "fmt"

func mergeMaps(maps ...map[string]int) map[string]int {
    merged := make(map[string]int)

    for _, m := range maps {
        for k, v := range m {
            merged[k] = v
        }
    }

    return merged
}

func main() {
    map1 := map[string]int{"a": 1, "b": 2}
    map2 := map[string]int{"b": 3, "c": 4}

    mergedMap := mergeMaps(map1, map2)

    fmt.Println(mergedMap)
}

运行结果：

map[a:1 b:3 c:4]

方法三：使用 sync.Map

对于并发环境下的 map 操作，可以使用 sync.Map 来保证并发安全。

package main

import (
    "fmt"
    "sync"
)

func mergeMaps(maps ...map[string]int) map[string]int {
    merged := sync.Map{}

    for _, m := range maps {
        for k, v := range m {
            merged.Store(k, v)
        }
    }

    resultMap := make(map[string]int)
    merged.Range(func(key, value interface{}) bool {
        k, _ := key.(string)
        v, _ := value.(int)
        resultMap[k] = v
        return true
    })

    return resultMap
}

func main() {
    map1 := map[string]int{"a": 1, "b": 2}
    map2 := map[string]int{"b": 3, "c": 4}

    mergedMap := mergeMaps(map1, map2)

    fmt.Println(mergedMap)
}

运行结果：