在 Go 中使用弱指针-Golang-PHP中文网

在 Go 中使用弱指针

DDD

发布： 2024-12-06 01:22:13

原创

962 人浏览过

Using Weak Pointers in Go

弱指针是 Go 的新增功能（在版本 1.24 中提供），它允许您引用内存中的对象，而不会阻止它们被垃圾收集。这篇博文将介绍弱指针，解释它们的用处，并提供使用它们构建内存高效缓存的具体示例。

什么是弱指针？

弱指针是对内存中对象的一种特殊引用。与强引用不同，如果不存在强引用，弱指针不会阻止垃圾收集器回收引用的对象。这使得弱指针成为您想要引用对象但又不想干扰 Go 的自动内存管理的场景的绝佳工具。

在Go 1.24中，弱指针将成为新的weak包的一部分。他们的工作方式如下：

使用weak.Make创建一个弱指针。
您可以使用 Value 方法访问引用的对象（如果它仍然存在）。
如果垃圾收集器已回收该对象，则 Value 返回 nil。

为什么使用弱指针？

弱指针在内存效率至关重要的情况下大放异彩。例如：

缓存：避免保留未使用的对象超过必要的时间。
观察者：跟踪对象而不阻止其清理。
参考文献：降低长时间运行的程序中内存泄漏的风险。

示例：使用弱指针构建缓存

假设您正在为存储经常访问的数据的 Web 服务器构建缓存。您希望缓存暂时保存数据，但让垃圾收集器清理其他地方不再使用的对象。

以下是使用弱指针的方法：

package main

import (
    "fmt"
    "runtime"
    "sync"
    "time"
    "weak"
)

// Cache represents a thread-safe cache with weak pointers.
type Cache[K comparable, V any] struct {
    mu    sync.Mutex
    items map[K]weak.Pointer[V] // Weak pointers to cached objects
}

// NewCache creates a new generic Cache instance.
func NewCache[K comparable, V any]() *Cache[K, V] {
    return &Cache[K, V]{
        items: make(map[K]weak.Pointer[V]),
    }
}

// Get retrieves an item from the cache, if it's still alive.
func (c *Cache[K, V]) Get(key K) (*V, bool) {
    c.mu.Lock()
    defer c.mu.Unlock()

    // Retrieve the weak pointer for the given key
    ptr, exists := c.items[key]
    if !exists {
        return nil, false
    }

    // Attempt to dereference the weak pointer
    val := ptr.Value()
    if val == nil {
        // Object has been reclaimed by the garbage collector
        delete(c.items, key)
        return nil, false
    }

    return val, true
}

// Set adds an item to the cache.
func (c *Cache[K, V]) Set(key K, value V) {
    c.mu.Lock()
    defer c.mu.Unlock()

    // Create a weak pointer to the value
    c.items[key] = weak.Make(&value)
}

func main() {
    // Create a cache with string keys and string values
    cache := NewCache[string, string]()

    // Add an object to the cache
    data := "cached data"
    cache.Set("key1", data)

    // Retrieve it
    if val, ok := cache.Get("key1"); ok {
        fmt.Println("Cache hit:", *val)
    } else {
        fmt.Println("Cache miss")
    }

    // Simulate losing the strong reference
    data = ""
    runtime.GC() // Force garbage collection

    // Try to retrieve it again
    time.Sleep(1 * time.Second)
    if val, ok := cache.Get("key1"); ok {
        fmt.Println("Cache hit:", *val)
    } else {
        fmt.Println("Cache miss")
    }
}

登录后复制