Netty 实现自定义消息格式

最新推荐文章于 2024-03-15 13:54:59 发布

weixin_33738578

最新推荐文章于 2024-03-15 13:54:59 发布

阅读量798

点赞数

文章标签： netty java

原文链接：https://my.oschina.net/grrrr/blog/187450

版权

为什么80%的码农都做不了架构师？>>>

Netty 实现基于google-protocol-buffer自定义消息格式

首先自己定义一个消息格式：

package fhq;

option java_package = "com.wesker.protocol";
option java_outer_classname = "ProtocolTypeMessage";

enum ProtocolType{
	EXCEPTION = 1;
	INFO = 2;
	SHUTDOWN = 3;
	EVALUATE_TO_TYPESET = 4;
	EVALUATE_TO_MATHML = 5;
}

package fhq;

option java_package = "com.wesker.protocol";
option java_outer_classname = "ProtocolMessage";
import "ProtocolTypeMessage";

message Protocol {
	required ProtocolType type = 1;
	required bytes data = 2;
}

这里的Protocol就是我们要处理的消息了。type是消息类型,data是数据。接下来生成java代码：

protoc --java_out=/home/michael/protocol/ ProtocolTypeMessage
protoc --java_out=/home/michael/protocol/ ProtocolMessage

接下来就是定义基于netty的消息转换器了，一个编码：将自定义的消息转换成byte[]。这里用范型指定我们刚才定义好的消息类型。

public class MessageEncoder extends MessageToByteEncoder<Protocol>{

	@Override
	protected void encode(ChannelHandlerContext context, Protocol protocol, ByteBuf out)
			throws Exception {
		byte[] data = protocol.toByteArray();
        out.writeInt(data.length);
        out.writeBytes(data);
	}
}

一个解码：将byte[]转换成自定义的消息

public class MessageDecoder extends ByteToMessageDecoder {

	@Override
	protected void decode(ChannelHandlerContext ctx, ByteBuf in, List<Object> out) throws Exception {
		if (in.readableBytes() < 4) {
			return;
		}
		in.markReaderIndex();
		int dataLength = in.readInt();
		if (in.readableBytes() < dataLength) {
			in.resetReaderIndex();
			return;
		}
		byte[] decoded = new byte[dataLength];
		in.readBytes(decoded);
		out.add(Protocol.parseFrom(decoded));
	}

}

在初始化server的时候将这2个东西放进去，和一个server处理自定义消息的handler

public class MyServerHandler extends SimpleChannelInboundHandler<Protocol>{

	@Override
	protected void messageReceived(ChannelHandlerContext arg0, Protocol arg1) throws Exception {
		//do something.....
	}

}

ServerBootstrap boot = new ServerBootstrap();
		boot.group(core.getBoss(), core.getWorker()).channel(NioServerSocketChannel.class).childHandler(new ChannelInitializer<SocketChannel>(){
			@Override
					protected void initChannel(SocketChannel ch)
							throws Exception {
						ChannelPipeline p = ch.pipeline();
						SSLEngine engine = core.getSslContext().createSSLEngine();
						engine.setUseClientMode(false);
						p.addLast("ssl", new SslHandler(engine));
						p.addLast(new LoggingHandler(LogLevel.INFO));
						p.addLast("encoder", new MessageEncoder());//在这里
						p.addLast("decoder", new MessageDecoder());//在这里
                                                p.addLast("MyServerHandler", new MyServerHandler());//server端处理消息
					}});

同样客户端也要做同样的事情:

public class MyClientHandler extends SimpleChannelInboundHandler<Protocol>{

	@Override
	protected void messageReceived(ChannelHandlerContext arg0, Protocol arg1) throws Exception {
		//do something.....
	}

}

Bootstrap bootstrap = new Bootstrap();
		bootstrap.group(core.getGroup()).channel(NioSocketChannel.class)
				.handler(new ChannelInitializer<SocketChannel>(){
					@Override
					protected void initChannel(SocketChannel ch)
							throws Exception {
						ChannelPipeline p = ch.pipeline();
						SSLEngine engine = core.getSslContext().createSSLEngine();
						engine.setUseClientMode(true);
						p.addLast("ssl", new SslHandler(engine));
						p.addLast("encoder", new MessageEncoder());//这里
						p.addLast("decoder", new MessageDecoder());//这里
						p.addLast("MyClientHandler", new MyClientHandler());//client端处理消息
					}});

到这里你就可以使用自己的消息格式了。

Netty 4以后都自带了对google-protocol-buffer的解析器：