简介
为什么世界上有这么多JAVA的程序员呢?其中一个很重要的原因就是JAVA相对于C++而言,不需要考虑对象的释放,一切都是由垃圾回收器来完成的。在崇尚简单的现代编程世界中,会C++的高手越来越少,会JAVA的程序员越来越多。
JVM的垃圾回收器中一个很重要的概念就是Reference count,也就是对象的引用计数,用来控制对象是否还被引用,是否可以被垃圾回收。
netty也是运行在JVM中的,所以JVM中的对象引用计数也适用于netty中的对象。这里我们说的对象引用指的是netty中特定的某些对象,通过对象的引用计数来判断这些对象是否还被使用,如果不再被使用的话就可以把它们(或它们的共享资源)返回到对象池(或对象分配器)。
这就叫做netty的对象引用计数技术,其中一个最关键的对象就是ByteBuf。
ByteBuf和ReferenceCounted
netty中的对象引用计数是从4.X版本开始的,ByteBuf是其中最终要的一个应用,它利用引用计数来提高分配和释放性能.
先来看一下ByteBuf的定义:
public abstract class ByteBuf implements ReferenceCounted, Comparable<ByteBuf>
可以看到ByteBuf是一个抽象类,它实现了ReferenceCounted的接口。
ReferenceCounted就是netty中对象引用的基础,它定义了下面几个非常重要的方法,如下所示:
int refCnt();
ReferenceCounted retain();
ReferenceCounted retain(int increment);
boolean release();
boolean release(int decrement);
其中refCnt返回的是当前引用个数,retain用来增加引用,而release用来释放引用。
ByteBuf的基本使用
刚分配情况下ByteBuf的引用个数是1:
ByteBuf buf = ctx.alloc().directBuffer();
assert buf.refCnt() == 1;
当调用他的release方法之后,refCnt就变成了0:
boolean destroyed = buf.release();
assert destroyed;
assert buf.refCnt() == 0;
当调用它的retain方法,refCnt就会加一:
ByteBuf buf = ctx.alloc().directBuffer();
assert buf.refCnt() == 1;
buf.retain();
assert buf.refCnt() == 2;
要注意的是,如果ByteBuf的refCnt已经是0了,就表示这个ByteBuf准备被回收了,如果再调用其retain方法,则会抛出IllegalReferenceCountException:refCnt: 0, increment: 1
所以我们必须在ByteBuf还未被回收之前调用retain方法。
既然refCnt=0的情况下,不能调用retain()方法,那么其他的方法能够调用吗?
我们来尝试调用一下writeByte方法:
try {
buf.writeByte(10);
} catch (IllegalReferenceCountException e) {
log.error(e.getMessage(),e);
}
可以看到,如果refCnt=0的时候,调用它的writeByte方法会抛出IllegalReferenceCountException异常。
这样看来,只要refCnt=0,说明这个对象已经被回收了,不能够再使用了。
ByteBuf的回收
既然ByteBuf中保存的有refCnt,那么谁来负责ByteBuf的回收呢?
netty的原则是谁消费ByteBuf,谁就负责ByteBuf的回收工作。
在实际的工作中,ByteBuf会在channel中进行传输,根据谁消费谁负责销毁的原则,接收ByteBuf的一方,如果消费了ByteBuf,则需要将其回收。
这里的回收指的是调用ByteBuf的release()方法。
ByteBuf的衍生方法
ByteBuf可以从一个parent buff中衍生出很多子buff。这些子buff并没有自己的reference count,它们的引用计数是和parent buff共享的,这些提供衍生buff的方法有:ByteBuf.duplicate(), ByteBuf.slice() 和 ByteBuf.order(ByteOrder)。
buf = directBuffer();
ByteBuf derived = buf.duplicate();
assert buf.refCnt() == 1;
assert derived.refCnt() == 1;
因为衍生的byteBuf和parent buff共享引用计数,所以如果要将衍生的byteBuf传给其他的流程进行处理的话,需要调用retain()方法:
ByteBuf parent = ctx.alloc().directBuffer(512);
parent.writeBytes(...);
try {
while (parent.isReadable(16)) {
ByteBuf derived = parent.readSlice(16);
derived.retain();
process(derived);
}
} finally {
parent.release();
}
...
public void process(ByteBuf buf) {
...
buf.release();
}
ChannelHandler中的引用计数
netty根据是读消息还是写消息,可以分为InboundChannelHandler和OutboundChannelHandler,分别用来读消息和写消息。
根据谁消费,谁释放的原则,对Inbound消息来说,读取完毕之后,需要调用ByteBuf的release方法:
public void channelRead(ChannelHandlerContext ctx, Object msg) {
ByteBuf buf = (ByteBuf) msg;
try {
...
} finally {
buf.release();
}
}
但是如果你只是将byteBuf重发到channel中供其他的步骤进行处理,则不需要release:
public void channelRead(ChannelHandlerContext ctx, Object msg) {
ByteBuf buf = (ByteBuf) msg;
...
ctx.fireChannelRead(buf);
}
同样的在Outbound中,如果只是简单的重发,则不需要release:
public void write(ChannelHandlerContext ctx, Object message, ChannelPromise promise) {
System.err.println("Writing: " + message);
ctx.write(message, promise);
}
如果是处理了消息,则需要release:
public void write(ChannelHandlerContext ctx, Object message, ChannelPromise promise) {
if (message instanceof HttpContent) {
// Transform HttpContent to ByteBuf.
HttpContent content = (HttpContent) message;
try {
ByteBuf transformed = ctx.alloc().buffer();
....
ctx.write(transformed, promise);
} finally {
content.release();
}
} else {
// Pass non-HttpContent through.
ctx.write(message, promise);
}
}
内存泄露
因为reference count是netty自身来进行维护的,需要在程序中手动进行release,这样会带来一个问题就是内存泄露。因为所有的reference都是由程序自己来控制的,而不是由JVM来控制,所以可能因为程序员个人的原因导致某些对象reference count无法清零。
为了解决这个问题,默认情况下,netty会选择1%的buffer allocations样本来检测他们是否存在内存泄露的情况.
如果发生泄露,则会得到下面的日志:
LEAK: ByteBuf.release() was not called before it's garbage-collected. Enable advanced leak reporting to find out where the leak occurred. To enable advanced leak reporting, specify the JVM option '-Dio.netty.leakDetectionLevel=advanced' or call ResourceLeakDetector.setLevel()
上面提到了一个检测内存泄露的level,netty提供了4种level,分别是:
- DISABLED---禁用泄露检测
- SIMPLE --默认的检测方式,占用1% 的buff。
- ADVANCED - 也是1%的buff进行检测,不过这个选项会展示更多的泄露信息。
- PARANOID - 检测所有的buff。
具体的检测选项如下:
java -Dio.netty.leakDetection.level=advanced ...
总结
掌握了netty中的引用计数,就掌握了netty的财富密码!
本文的例子可以参考:learn-netty4
本文已收录于 http://www.flydean.com/43-netty-reference-cound/
最通俗的解读,最深刻的干货,最简洁的教程,众多你不知道的小技巧等你来发现!
欢迎关注我的公众号:「程序那些事」,懂技术,更懂你!