Go语言中存储指针指向的地址

第一段引用上面的摘要：

本文介绍了如何在Go语言中存储由new()函数返回的指针指向的内存地址。通常情况下，直接使用指针类型作为map的键会引发类型错误。本文将探讨如何使用unsafe包中的Pointer类型以及reflect包中的UnsafeAddr函数来解决此问题，并提供使用示例和注意事项，帮助开发者安全有效地管理内存地址。

在Go语言中，当你使用new()函数为一个类型分配内存时，它会返回一个指向该内存的指针。如果你想将这个指针指向的地址存储在一个map中，通常会遇到类型不匹配的问题。这是因为Go是一种强类型语言，直接将指针类型作为map的键是不允许的。解决这个问题，可以使用unsafe包中的Pointer类型或者reflect包中的UnsafeAddr函数。

使用 unsafe.Pointer

unsafe包提供了一种绕过Go类型系统的方式，允许你将任何指针转换为unsafe.Pointer类型。unsafe.Pointer可以转换为uintptr，而uintptr可以作为map的键。 但是，使用 unsafe 包需要格外小心，因为它可能会破坏Go的内存安全保证。

立即学习“go语言免费学习笔记（深入）”；

以下是一个示例：

云雀语言模型

云雀是一款由字节跳动研发的语言模型，通过便捷的自然语言交互，能够高效的完成互动对话

54

查看详情

package main

import (
    "fmt"
    "unsafe"
)

type T struct {
    a, b int
}

func main() {
    memmap := make(map[uintptr]int64) // 使用 uintptr 作为键
    var t int64 = 123456789 // 模拟时间戳

    var ptr *T = new(T)
    ptr.a = 1
    ptr.b = 2

    // 将指针转换为 unsafe.Pointer，再转换为 uintptr
    address := uintptr(unsafe.Pointer(ptr))
    memmap[address] = t

    fmt.Printf("Address: %v, Time: %v\n", address, memmap[address])

    // 注意：使用完指针后，确保内存仍然有效，否则读取地址可能会导致错误。
}

注意事项：

• unsafe.Pointer 的使用需要谨慎，因为它绕过了Go的类型安全检查。
• 确保在使用地址之前，指针指向的内存仍然有效。如果内存被释放，访问该地址会导致程序崩溃。
• 避免过度依赖unsafe包，尽量使用更安全的Go语言特性。

使用 reflect.Value.Pointer() (Go 1.17 及更早版本) 或 reflect.Value.UnsafePointer() (Go 1.17 及更高版本)

reflect 包提供了一种更安全的方式来获取指针的地址，它使用 reflect.Value 来操作变量。 从 Go 1.17 开始，推荐使用 UnsafePointer() 方法来获取 unsafe.Pointer。 在 Go 1.17 之前的版本，可以使用 Pointer() 方法来获取 uintptr。

package main

import (
    "fmt"
    "reflect"
    "unsafe"
)

type T struct {
    a, b int
}

func main() {
    memmap := make(map[uintptr]int64)
    var t int64 = 123456789

    var ptr *T = new(T)
    ptr.a = 1
    ptr.b = 2

    // 使用 reflect.ValueOf 获取指针的 Value
    value := reflect.ValueOf(ptr)

    // 获取指针的地址 (Go 1.17 及更高版本)
    address := uintptr(value.UnsafePointer())

    // 获取指针的地址 (Go 1.17 之前版本)
    // address := value.Pointer()

    memmap[address] = t

    fmt.Printf("Address: %v, Time: %v\n", address, memmap[address])
}

注意事项：

• reflect 包提供了更类型安全的方式来操作指针，但性能可能会略有下降。
• 确保在使用地址之前，指针指向的内存仍然有效。

总结

在Go语言中存储指针指向的地址，可以使用unsafe.Pointer或者reflect包。unsafe.Pointer提供了最大的灵活性，但也带来了安全风险。reflect包提供了更安全的方式，但性能可能会略有下降。选择哪种方式取决于你的具体需求和对安全性的考虑。 建议优先考虑使用 reflect 包提供的功能，除非性能是关键因素。 并且，无论选择哪种方式，都需要谨慎处理指针，确保在使用地址之前，指针指向的内存仍然有效。

以上就是Go语言中存储指针指向的地址的详细内容，更多请关注php中文网其它相关文章！