在 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")
    }
}

登入後複製