Golang如何实现指针引用计数

package main

import (
    "fmt"
    "sync"
)

type RefCounted struct {
    data interface{}
    refs int
    mu   sync.Mutex
    onZero func() // 引用归零时的回调，如释放资源
}

func NewRefCounted(data interface{}, cleanup func()) *RefCounted {
    return &RefCounted{
        data:   data,
        refs:   1,
        onZero: cleanup,
    }
}

// 增加引用
func (r *RefCounted) IncRef() *RefCounted {
    r.mu.Lock()
    r.refs++
    r.mu.Unlock()
    return r
}

// 减少引用
func (r *RefCounted) DecRef() {
    r.mu.Lock()
    r.refs--
    if r.refs == 0 {
        r.mu.Unlock()
        if r.onZero != nil {
            r.onZero()
        }
        return
    }
    r.mu.Unlock()
}

// 获取数据
func (r *RefCounted) Data() interface{} {
    r.mu.Lock()
    defer r.mu.Unlock()
    return r.data
}

func main() {
    cleanup := func() {
        fmt.Println("资源已被释放")
    }

    obj := NewRefCounted("共享字符串", cleanup)

    // 模拟多个引用
    ref1 := obj.IncRef()
    ref2 := obj.IncRef()

    // 使用数据
    fmt.Println("ref1 data:", ref1.Data())
    fmt.Println("ref2 data:", ref2.Data())

    // 释放引用
    obj.DecRef() // 总引用数: 2
    ref1.DecRef() // 总引用数: 1
    ref2.DecRef() // 总引用数: 0 → 触发 cleanup
}

ref1 data: 共享字符串
ref2 data: 共享字符串
资源已被释放