Go语言中自定义类型作为Map键的陷阱与解决方案：指针与值语义的辨析

package main

import "fmt"

type Point struct {
    row int
    col int
}

// NewPoint 返回 *Point 指针
func NewPoint(r, c int) *Point {
    return &Point{r, c}
}

func (p *Point) String() string {
    return fmt.Sprintf("{%d, %d}", p.row, p.col)
}

func main() {
    set := make(map[*Point]bool)

    // 添加两个逻辑上相等的点，但它们是不同的指针实例
    p1 := NewPoint(0, 0)
    p2 := NewPoint(0, 2)
    set[p1] = true // p1 指向地址 A
    set[p2] = true // p2 指向地址 B

    fmt.Println("Set after adding p1 and p2:")
    for k, v := range set {
        fmt.Printf("Key: %s (地址: %p), Value: %t\n", k, k, v)
    }

    // 尝试查找一个逻辑上等于p2的新点实例
    // NewPoint(0, 2) 会创建另一个新的 Point 实例，指向地址 C
    _, ok := set[NewPoint(0, 2)]
    if ok {
        fmt.Println("错误: 使用新的指针找到了 NewPoint(0, 2)。")
    } else {
        fmt.Println("正确: 未能使用新的指针找到 NewPoint(0, 2) (因为指针地址不同)。")
    }

    // 尝试查找原始 p2 指针
    _, ok = set[p2]
    if ok {
        fmt.Println("正确: 找到了原始 p2 指针。")
    } else {
        fmt.Println("错误: 未能找到原始 p2 指针。")
    }
}

package main

import "fmt"

type Point struct {
    row int
    col int
}

// NewPoint 函数现在返回 Point 值，而不是指针
func NewPoint(r, c int) Point {
    return Point{r, c}
}

func (p Point) String() string { // 接收者改为值类型
    return fmt.Sprintf("{%d, %d}", p.row, p.col)
}

func main() {
    // 键类型改为 Point 结构体本身
    set := make(map[Point]bool)

    // 直接使用 Point 值作为键
    set[NewPoint(0, 0)] = true
    set[NewPoint(0, 2)] = true // 这里的 NewPoint(0,2) 是一个新的值，但会与现有键比较值

    fmt.Println("Set after adding points (using Point as key):")
    for k, v := range set {
        fmt.Printf("Key: %s, Value: %t\n", k, v)
    }

    // 尝试查找一个逻辑上等于 NewPoint(0, 2) 的新点实例
    // 此时，map 会根据 Point 的值进行比较
    _, ok := set[NewPoint(0, 2)]
    if ok {
        fmt.Println("正确: 使用值相等性找到了 NewPoint(0, 2)。")
    } else {
        fmt.Println("错误: 未能找到 NewPoint(0, 2)。")
    }

    // 尝试查找一个不存在的元素
    _, ok = set[NewPoint(3, 3)]
    if !ok {
        fmt.Println("正确: 未能找到 NewPoint(3, 3)。")
    } else {
        fmt.Println("错误: 找到了 NewPoint(3, 3)。")
    }
}