使用Netty4实现基本的消息分发

示例工程代码

 

可从附件下载

 

具体的说明和用法在后面介绍

 

需求与目的

 

一个游戏服务端需要处理各种业务逻辑，每一种业务逻辑都对应着一个请求消息和一个响应消息。那么服务端需要把这些不同的消息自动分发到对应的业务逻辑中处理。

 

最简单的处理方式就是根据请求消息中的type字段，使用switch case来进行分别处理，但这种方式随着消息的增多，显现了一些坏味道：长长的一大坨不太好看；如果要添加新的消息、新的逻辑，或者去掉新的消息、新的逻辑，在代码上不但要修改这些消息和逻辑，还不得不修改这长长的一坨swtich case，这样的修改显得很多余。

 

所以我们的目的就是把消息分发这块的代码自动化，在增加、修改、删除消息和逻辑的时候不需要再对消息分发的代码再做修改，从而使得修改的代码最小化。

 

实现原理

 

在实现中，使用了注解（annotation）

package com.company.game.dispatcher.annotation;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

/**
 * 修饰消息类和业务逻辑执行类
 * msgType指定对应的类型，从1开始计数
 * @author xingchencheng
 *
 */

@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
public @interface UserMsgAndExecAnnotation {
	short msgType();
}

 

唯一的字段msgType代表了消息类型，这是客户端与服务端的约定，这里我们从1开始计数。

 

当我们要增加一个加法消息时，就使用这个注解来修饰我们的请求消息类：

package com.company.game.dispatcher.msg;

import com.company.game.dispatcher.annotation.UserMsgAndExecAnnotation;

/**
 * 加法请求消息类
 * 
 * @author xingchencheng
 *
 */

@UserMsgAndExecAnnotation(msgType = MsgType.ADD)
public class UserAddRequest extends RequestMsgBase {

	private double leftNumber;
	private double RightNumber;
	
	public UserAddRequest() {
		super(MsgType.ADD);
	}

	public double getLeftNumber() {
		return leftNumber;
	}

	public void setLeftNumber(double leftNumber) {
		this.leftNumber = leftNumber;
	}

	public double getRightNumber() {
		return RightNumber;
	}

	public void setRightNumber(double rightNumber) {
		RightNumber = rightNumber;
	}

}

 

为什么要这样修饰呢？先从服务端的解码（decode）说起，实例代码中，一个请求消息是这样规定的：

 

 0-1字节表示整个消息的长度（单位：字节）

 2-3字节代表消息类型，对应annotation的msgType

 余下的是消息的json字符串（UTF-8编码）

 

我们需要根据2-3字节表示的msgType得到对应请求消息类的class对象，用这个class对象来序列化json字符串，得到具体的请求对象。那么怎么根据msgType得到class对象呢？这就是为什么要使用annotation的原因。

 

在服务端程序启动前，会执行下面的处理：

 

 // msgType->请求、响应类的class对象
	private static Map> typeToMsgClassMap;
	
	// 根据类型得到对应的消息类的class对象
	public static Class getMsgClassByType(short type) {
		return typeToMsgClassMap.get(type);
	}
	
	/**
	 * 初始化typeToMsgClassMap
	 * 遍历包com.company.game.dispatcher.msg
	 * 取得消息类的class文件
	 * 
	 * @throws ClassNotFoundException
	 * @throws IOException
	 */
	public static void initTypeToMsgClassMap() 
			throws ClassNotFoundException, IOException {
		
		Map> tmpMap = new HashMap>();
		
		Set> classSet = getClasses("com.company.game.dispatcher.msg");
		if (classSet != null) {
			for (Class clazz : classSet) {
				if (clazz.isAnnotationPresent(UserMsgAndExecAnnotation.class)) {
					UserMsgAndExecAnnotation annotation = clazz.getAnnotation(UserMsgAndExecAnnotation.class);
					tmpMap.put(annotation.msgType(), clazz);
				}
			}
		}
		
		typeToMsgClassMap = Collections.unmodifiableMap(tmpMap);
	}

 

程序初始化了一个映射，在指定的包找到请求的消息类的class，读取class上的annotation，保存到一个Map中，这样在后续就可以根据这个Map来根据msgType得到class对象了。

 

再给出解码器的实现：

 

package com.company.game.dispatcher.codec;

import java.util.List;

import com.company.game.dispatcher.util.ClassUtil;
import com.company.game.dispatcher.util.GsonUtil;
import com.google.gson.Gson;

import io.netty.buffer.ByteBuf;
import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.codec.ByteToMessageDecoder;

/**
 * 解码器
 * 客户端和服务端均有使用
 * 0-1字节表示整个消息的长度（单位：字节）
 * 2-3字节代表消息类型，对应annotation
 * 余下的是消息的json字符串（UTF-8编码）
 * 
 * @author xingchencheng
 *
 */

public class MsgDecoder extends ByteToMessageDecoder {

	@Override
	protected void decode(ChannelHandlerContext ctx, ByteBuf buf,
			List