AbstractReferenceCountedByteBuf源码分析

最新推荐文章于 2024-12-04 15:01:22 发布

HryReal

最新推荐文章于 2024-12-04 15:01:22 发布

阅读量1.5w

点赞数 1

CC 4.0 BY-SA版权

分类专栏： Netty源码分析(ByteBuf) 文章标签： ReferenceC 自旋操作 netty 计数器

本文链接：https://blog.youkuaiyun.com/qq_33394088/article/details/72730689

Netty源码分析(ByteBuf) 专栏收录该内容

8 篇文章

订阅专栏

本文深入分析了Netty的AbstractReferenceCountedByteBuf类，探讨了其如何利用AtomicIntegerFieldUpdater确保原子性，并详细解释了retain()、release()等引用计数方法的实现，尤其是通过CAS操作进行的自旋操作确保线程安全。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1.成员变量

 private static final AtomicIntegerFieldUpdater<AbstractReferenceCountedByteBuf> refCntUpdater;

    static {
        AtomicIntegerFieldUpdater<AbstractReferenceCountedByteBuf> updater =
                PlatformDependent.newAtomicIntegerFieldUpdater(AbstractReferenceCountedByteBuf.class, "refCnt");
        if (updater == null) {
            updater = AtomicIntegerFieldUpdater.newUpdater(AbstractReferenceCountedByteBuf.class, "refCnt");
        }
        refCntUpdater = updater;
    }

    private volatile int refCnt = 1；

AtomicIntegerFieldUpdater是什么鬼？先研究一下atomic包下的两个重要的类
1.AtomicInteger的使用
AtomicInteger ℹ=new AtimicInteger(0);
这样变量i就有了原子性

2.AtomicintegerFieldUpdater的使用
public static AtomicIntegerFieldUpdaternewUpdater(Class tclass,
String fieldName)
这里就不详细分析他的源码了，其实很简单，他让tclass的成员fieldName具有了原子性，是不是很简单～

研究完再回来看他的成员变量，实在不能太简单。
AbstractReferenceCountedByteBuf源码实现，该类主要是实现引用计算的常规方法，充分利用voliate内存可见性与CAS操作完成refCnt变量的维护。

2.实现的函数

在这个类的函数中，看到了很熟悉的东西release(),retain(),refcnt(),touch()这些方法，天呐，这不就是ByteBuf类，AbstarctByteBuf类中没有实现的方法吗？原来都拿到这里来实现了。上代码！

 @Override
    public ByteBuf retain() {
        for (;;) {
            int refCnt = this.refCnt;
            if (refCnt == 0) {
                throw new IllegalReferenceCountException(0, 1);
            }
            if (refCnt == Integer.MAX_VALUE) {
                throw new IllegalReferenceCountException(Integer.MAX_VALUE, 1);
            }
            if (refCntUpdater.compareAndSet(this, refCnt, refCnt + 1)) {
                break;
            }
        }
        return this;
    }

每调用一次retain方法，计数器加一
compareAndSet方法用来获取自己的值和期望的值进行比较，如果其间被其他线程修改了，那么比对失败，进行自旋操作，重新获得计数器重新比较
compareAndSet这个方法是CAS操作，由操作系统层面提供。

 @Override
    public final boolean release() {
        for (;;) {
            int refCnt = this.refCnt;
            if (refCnt == 0) {
                throw new IllegalReferenceCountException(0, -1);
            }

            if (refCntUpdater.compareAndSet(this, refCnt, refCnt - 1)) {
                if (refCnt == 1) {
                    deallocate();
                    return true;
                }
                return false;
            }
        }
    }