本文共 8384 字,大约阅读时间需要 27 分钟。
Netty 提供了它所谓的 Embedded 传输,用于测试 ChannelHandler。这个传输是一种特殊的 Channel 实现— EmbeddedChannel— 的功能,这个实现提供了通过 ChannelPipeline 传播事件的简便方法。
这个想法是直截了当的:将入站数据或者出站数据写入到 EmbeddedChannel 中,然后检 查是否有任何东西到达了 ChannelPipeline 的尾端。以这种方式,你便可以确定消息是否已 经被编码或者被解码过了,以及是否触发了任何的 ChannelHandler 动作。
入站数据由 ChannelInboundHandler 处理,代表从远程节点读取的数据。出站数据由 ChannelOutboundHandler 处理,代表将要写到远程节点的数据。根据你要测试的 ChannelHandler,你将使用*Inbound()或者*Outbound()
方法对,或者兼而有之。
下图展示了使用 EmbeddedChannel 的方法,数据是如何流经 ChannelPipeline 的。 你可以使用 writeOutbound()方法将消息写到 Channel 中,并通过 ChannelPipeline 沿 着出站的方向传递。随后,你可以使用 readOutbound()方法来读取已被处理过的消息,以确 定结果是否和预期一样。类似地,对于入站数据,你需要使用 writeInbound()和 readInbound() 方法。
测试入站消息:
下图展示了一个简单的ByteToMessageDecoder实现。给定足够的数据,这个实现将 产生固定大小的帧。如果没有足够的数据可供读取,它将等待下一个数据块的到来,并将再次检 查是否能够产生一个新的帧。public class FixedLengthFrameDecoder extends ByteToMessageDecoder { private final int frameLength; public FixedLengthFrameDecoder(int frameLength) { if (frameLength <= 0) { throw new IllegalArgumentException("frameLength must be a positive integer:" + frameLength); } this.frameLength = frameLength; } @Override protected void decode(ChannelHandlerContext ctx, ByteBuf in, List
测试FixedLengthFrameDecoder:
public class FixedLengthFrameDecoderTest { @Test public void testFrameDecoded() { //创建一个ByteBuf,并存储9字节 ByteBuf buf = Unpooled.buffer(); for (int i = 0; i < 9; i++) { buf.writeByte(i); } ByteBuf input = buf.duplicate(); EmbeddedChannel channel = new EmbeddedChannel(new FixedLengthFrameDecoder(3)); //将数据写入EmbeddedChannel System.out.println(channel.writeInbound(input.retain()));//true //标记Channel为已完成状态 System.out.println(channel.finish());//true //读取所生成的消息,并且验证是否有3帧,其中每帧都为3字节 ByteBuf read = channel.readInbound(); System.out.println(buf.readSlice(3).equals(read));//true read = channel.readInbound(); System.out.println(buf.readSlice(3).equals(read));//true read.release(); read = channel.readInbound(); System.out.println(buf.readSlice(3).equals(read));//true read.release(); System.out.println(channel.readInbound() == null);//true buf.release(); } @Test public void testFramesDescode2() { ByteBuf buf = Unpooled.buffer(); for (int i = 0; i < 9; i++) { buf.writeByte(i); } ByteBuf input = buf.duplicate(); EmbeddedChannel channel = new EmbeddedChannel(new FixedLengthFrameDecoder(3)); //返回false,因为没有一个完整的可供读取的帧 System.out.println(channel.writeInbound(input.readBytes(2)));//false System.out.println(channel.writeInbound(input.readBytes(7)));//true System.out.println(channel.finish());//true ByteBuf read = channel.readInbound(); System.out.println(buf.readSlice(3) == read);//false read.release(); read = channel.readInbound(); System.out.println(buf.readSlice(3) == read);//false read.release(); read = channel.readInbound(); System.out.println(buf.readSlice(3) == read);//false read.release(); System.out.println(channel.readInbound() == null);//true buf.release(); }}
该 testFramesDecoded()方法验证了:一个包含 9 个可读字节的 ByteBuf 被解码为 3 个 ByteBuf,每个都包含了 3 字节。需要注意的是,仅通过一次对 writeInbound()方法的调 用,ByteBuf 是如何被填充了 9 个可读字节的。在此之后,通过执行 finish()方法,将 EmbeddedChannel 标记为了已完成状态。最后,通过调用 readInbound()方法,从 Embedded- Channel 中正好读取了 3 个帧和一个 null。
测试出站消息的处理过程和刚才所看到的类似。在下面的例子中,我们将会展示如何使用 EmbeddedChannel 来测试一个编码器形式的 ChannelOutboundHandler,编码器是一种 将一种消息格式转换为另一种的组件。
AbsIntegerEncoder,它是 Netty 的 MessageToMessageEncoder 的一个特殊化的实现,用于将负值整数转换为绝对值。
该示例将会按照下列方式工作:
public class AbsIntegerEncoder extends MessageToMessageEncoder{ @Override protected void encode(ChannelHandlerContext ctx, ByteBuf msg, List
测试AbsIntegerEncoder:
public class AbsIntegerEncoderTest { @Test public void testEncoded() { ByteBuf buf = Unpooled.buffer(); for (int i = 1; i < 10; i++) { buf.writeInt(i * -1); } //创建一个EmbeddedChanel,并安装一个要测试的AbsIntegerEncoder EmbeddedChannel channel = new EmbeddedChannel(new AbsIntegerEncoder()); //写入ByteBuf,调用readOutbound()方法将会产生数据 System.out.println(channel.writeOutbound(buf)); System.out.println(channel.finish()); channel.readOutbound(); for (int i = 1; i < 10; i++) { int temp = channel.readOutbound(); System.out.println(temp); } System.out.println(channel.readOutbound() == null); }}
下面是代码中执行的步骤。
测试异常处理:
应用程序通常需要执行比转换数据更加复杂的任务。例如,你可能需要处理格式不正确的输 入或者过量的数据。在下一个示例中,如果所读取的字节数超出了某个特定的限制,我们将会抛 出一个 TooLongFrameException。这是一种经常用来防范资源被耗尽的方法。
在下图中,最大的帧大小已经被设置为 3 字节。如果一个帧的大小超出了该限制,那么程序将 会丢弃它的字节,并抛出一个 TooLongFrameException。位于 ChannelPipeline 中的其他 ChannelHandler 可以选择在 exceptionCaught()方法中处理该异常或者忽略它:
//扩展ByteToMessageDecoder以将入站字节码为消息public class FrameChunkDecoder extends ByteToMessageDecoder { private final int maxFrameSize; public FrameChunkDecoder(int maxFrameSize) { this.maxFrameSize = maxFrameSize; } @Override protected void decode(ChannelHandlerContext ctx, ByteBuf in, List
测试FrameChunkDecoder:
public class FrameChunkDecoderTest { @Test public void testFramesDecoded() { ByteBuf buf = Unpooled.buffer(); for (int i = 0; i < 9; i++) { buf.writeByte(i); } ByteBuf input = buf.duplicate(); //创建一个EmbeddedChannel,并向其安装一个帧大小为3字节的FixedLengthFrameDecoder EmbeddedChannel channel = new EmbeddedChannel(new FrameChunkDecoder(3)); System.out.println(channel.writeInbound(input.readBytes(2))); try { //写入一个4字节大小的帧,并捕获预期的异常 channel.writeInbound(input.readBytes(4)); } catch (TooLongFrameException e) { e.printStackTrace(); } //写入剩余的2字节,会产生一个有效帧 System.out.println(channel.writeInbound(input.readBytes(3)));//true System.out.println(channel.finish()); //读取产生的消息,并且验证值 ByteBuf read = channel.readInbound(); System.out.println(read.equals(buf.readSlice(2)));//true read.release(); read = channel.readInbound(); System.out.println(read.equals(buf.skipBytes(4).readSlice(3)));//true read.release(); buf.release(); }}
这里使用的try/catch块是EmbeddedChannel的一个特 殊功能。如果其中一个write*方法产生了一个受检查的Exception,那么它将会被包装在一个 RuntimeException中并抛出。这使得可以容易地测试出一个Exception是否在处理数据的 过程中已经被处理了。
转载地址:http://qlpqb.baihongyu.com/