在netty消息中添加固定或可变长度的标题

Question

我想为netty聊天程序的所有消息预先添加一个固定或可变长度的标题。

我正在建立一个简单的聊天；所有消息（当前）都使用LengthFieldPrepender和LengthFieldBasedFrameDecoder进行长度分隔，长度部分为2字节，目前为止效果良好：

 +-------------+-------------------+  
 | <length>    | <message>         |  
 |    13       | 'Yo, waz up?'     | 
 +-------------+-------------------+ 

..管道呢

//inbound
pipeline.addLast(new LengthFieldBasedFrameDecoder (1024,0,2,0,2));
pipeline.addLast(new StringDecoder(CharsetUtil.UTF_8));
//outbound
pipeline.addLast(new LengthFieldPrepender(2));
pipeline.addLast(new StringDecoder(CharsetUtil.UTF_8));

目标是添加一个包含应用程序名称版本的标头（固定或变量），以过滤无效消息；本示例使用标头'nchat v1.0'，10字节固定长度：

 --------------+--------------+-------------------+  
 | <length>    | <header>     | <message>         |  
 |    23       | 'nChat v1.0' | 'Yo, waz up?'     | 
 --------------+--------------+-------------------+ 

调整入站管道以容纳标头应该不是问题，尽管我不希望标头被剥离，所以我将initialBytesToStrip留在2：

pipeline.addLast(new LengthFieldBasedFrameDecoder (1024,0,2,0,2));
pipeline.addLast(new StringDecoder(CharsetUtil.UTF_8));

我不确定向消息添加标题的最佳方法是什么。我假设需要一个自定义处理程序；我想到了两个想法：

• 子类StringEncoder，将报头放在传出消息的前面，像往常一样编码。
• 创建新的处理程序，类似于LengthFieldPrepender，使用out.writeBytes（"nChatv1.0". getBytes（））

我怀疑后一个选项是正确的，因为标题在原则上与LengthFieldPrepender在所做的事情上没有什么不同。

Answer 1

这是我想出来的；它支持可变长度的标题。我仍然00b，所以欢迎评论：-D

   public class HeaderFieldEncoder extends MessageToByteEncoder<ByteBuf> { 
      public HeaderFieldEncoder(String header){
         this.header = header;
      }     
      @Override
      protected void encode(
         ChannelHandlerContext ctx,
         ByteBuf msg,
         ByteBuf out
      )throws Exception {                           
         //header = "nChat v1.0"
         ByteBuf headerBuf = Unpooled.copiedBuffer(
            header.getBytes(charset));                                    
         out.writeBytes(headerBuf);
         out.writeBytes(msg, msg.readerIndex(), msg.readableBytes());      
      }           
   }


   public class HeaderFieldDecoder extends MessageToMessageDecoder<ByteBuf> {
      public HeaderFieldDecoder(String header){
         this.header = header;
      }
      @Override
      protected void decode(
         ChannelHandlerContext ctx,
         ByteBuf msg,
         List<Object> out
      )throws Exception {
         String extractedHeader = extractHeader(msg);
         //if desired, validate header
         //if(!header.equals(extractedHeader)
         String content = msg.toString(charset);
         out.add(content);         
      }

      private String extractHeader(ByteBuf msg){
         String headerField = null;
         if(msg.readableBytes() >= header.length()){
            byte[] headerBytes = new byte[header.length()];
            msg.readBytes(headerBytes, 0, header.length());
            headerField = new String(headerBytes, charset);
         }
         return headerField;
      }
   }

工作管道：

String header = "nChat v1.0";
//INBOUND 
pipeline.addLast(new LengthFieldBasedFrameDecoder(1024,0,2,0,2));
pipeline.addLast(new HeaderFieldDecoder(header));
pipeline.addLast(new StringDecoder());

//OUTBOUND
pipeline.addLast(new LengthFieldPrepender(2)); 
pipeline.addLast(new HeaderFieldEncoder(header));
pipeline.addLast(new StringEncoder()); 

你怎么看；这有效吗？我清理资源正确吗？