andreacavalli/netty5
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

NETTY-406 ProtobufEncoder should also support Message.Builder

Browse Source 
* Added support for Message.Builder
This commit is contained in:
Trustin Lee 2011-08-01 03:46:22 +09:00
parent e2655f3c38
commit b9ef785b5b
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/main/java/org/jboss/netty/handler/codec/protobuf/ProtobufEncoder.java
View File

@ -78,9 +78,12 @@ public class ProtobufEncoder extends OneToOneEncoder {
@Override
protected Object encode(
ChannelHandlerContext ctx, Channel channel, Object msg) throws Exception {
if (!(msg instanceof MessageLite)) {
return msg;
}
if (msg instanceof MessageLite) {
return wrappedBuffer(((MessageLite) msg).toByteArray());
}
if (msg instanceof MessageLite.Builder) {
return wrappedBuffer(((MessageLite.Builder) msg).build().toByteArray());
}
return msg;
}
}