Go语言中合并两个Map的惯用方法与实践

package main

import "fmt"

// SomeObject 示例结构体，代表文件信息
type SomeObject struct {
    Path string
    Size int
}

func main() {
    // 假设 map a 已经包含了一些数据
    a := map[string]*SomeObject{
        "/path/to/file1.txt": {Path: "/path/to/file1.txt", Size: 100},
        "/path/to/dir/fileA.log": {Path: "/path/to/dir/fileA.log", Size: 200},
    }

    // 假设 map b 是递归调用或其他操作的结果
    b := map[string]*SomeObject{
        "/path/to/file2.txt": {Path: "/path/to/file2.txt", Size: 150},
        "/path/to/dir/fileA.log": {Path: "/path/to/dir/fileA.log", Size: 250}, // 注意：键冲突
        "/path/to/newfile.dat": {Path: "/path/to/newfile.dat", Size: 300},
    }

    fmt.Println("Original map 'a':")
    for k, v := range a {
        fmt.Printf("  %s: {Path: %s, Size: %d}\n", k, v.Path, v.Size)
    }
    fmt.Println("Original map 'b':")
    for k, v := range b {
        fmt.Printf("  %s: {Path: %s, Size: %d}\n", k, v.Path, v.Size)
    }

    // 合并操作：将 b 中的所有键值对赋值到 a 中
    for k, v := range b {
        a[k] = v
    }

    fmt.Println("\nMerged map 'a' (after merging 'b' into 'a'):")
    for k, v := range a {
        fmt.Printf("  %s: {Path: %s, Size: %d}\n", k, v.Path, v.Size)
    }
}

// 策略一：如果键已存在，则保留a中的值，不覆盖
for k, v := range b {
    if _, exists := a[k]; !exists {
        a[k] = v
    }
}

// 策略二：如果键已存在，合并值（假设SomeObject有合并逻辑）
// 这种情况下需要SomeObject类型提供一个合并方法或自定义逻辑
/*
for k, v := range b {
    if existingVal, exists := a[k]; exists {
        // 假设SomeObject有一个Merge方法
        a[k] = existingVal.Merge(v)
    } else {
        a[k] = v
    }
}
*/

func MergeMaps(map1, map2 map[string]*SomeObject) map[string]*SomeObject {
    // 预估合并后的容量，可以减少内存重新分配的开销
    merged := make(map[string]*SomeObject, len(map1) + len(map2))

    // 先将第一个map的内容复制到新map
    for k, v := range map1 {
        merged[k] = v
    }

    // 再将第二个map的内容复制到新map
    // 此时，如果map2中有与map1相同的键，map2的值将覆盖map1的值
    for k, v := range map2 {
        merged[k] = v
    }
    return merged
}

// 使用示例
// mergedMap := MergeMaps(a, b)

// 假设SomeObject有一个Copy方法，可以进行深拷贝
func (s *SomeObject) Copy() *SomeObject {
    if s == nil {
        return nil
    }
    return &SomeObject{
        Path: s.Path,
        Size: s.Size,
        // 复制所有其他字段...
    }
}

// 合并时进行深拷贝的示例
func MergeMapsDeepCopy(map1, map2 map[string]*SomeObject) map[string]*SomeObject {
    merged := make(map[string]*SomeObject, len(map1) + len(map2))

    for k, v := range map1 {
        merged[k] = v.Copy() // 深拷贝
    }

    for k, v := range map2 {
        // 这里可以根据需要决定是深拷贝还是直接赋值
        // 如果map2中的值应该覆盖map1中的深拷贝，则直接覆盖
        // 如果希望合并内部字段，则需要更复杂的逻辑
        merged[k] = v.Copy() // 深拷贝
    }
    return merged
}