首页 > 代码库 > Akka边学边写(4)-- MiniRPG
Akka边学边写(4)-- MiniRPG
前面几篇文章用Akka写了HelloWorld和EchoServer,为了更进一步学习Akka,本文将会实现一个很小的RPG游戏服务器:MiniRPG。
游戏逻辑
因为是迷你RPG,所以逻辑很简单。服务器可以处理四种操作:创建玩家、给玩家加经验、升级、查询玩家信息。下面是Player类的代码(Getters和Setters省略):
public class Player { private int id; private String name; private int exp; private int level; // Getters & Setters ... public void addExp(int val) { exp += val; } public void levelUp() { if (exp > 100) { exp -= 100; level++; } } }
消息协议
MiniRPG底层使用TCP协议,消息使用JSON格式。完整的消息格式如下图所示:
前八个字节可以认为是消息头,其中前四个字节是消息ID,后四个字节是JSON字符串长度。其余字节是消息体,也就是UTF8格式编码的JSON字符串。
消息接口
MiniRPG设计了三个接口来表示游戏消息,这三个接口都是Marker接口,里面没有定义任何方法,如下图所示:
MsgRegistry
MiniRPG使用GSON来编码和解码JSON字符串,为了把JSON解析为相应的消息对象,需要一个消息ID和class之间的映射关系。MsgRegistry类便是要建立起这样一个映射关系,下面是它的完整代码:
public class MsgRegistry { private static final Map<Integer, Class<?>> msgById = new HashMap<>(); private static final Map<Class<?>, Integer> idByMsg = new HashMap<>(); static { register(1, CreatePlayerRequest.class); register(2, CreatePlayerResponse.class); register(3, AddExpRequest.class); register(4, AddExpResponse.class); register(5, LevelUpRequest.class); register(6, LevelUpResponse.class); register(7, GetPlayerInfoRequest.class); register(8, GetPlayerInfoResponse.class); } private static void register(int msgId, Class<?> msgClass) { msgById.put(msgId, msgClass); idByMsg.put(msgClass, msgId); } public static Class<?> getMsgClass(int msgId) { return msgById.get(msgId); } public static int getMsgId(Class<?> msgClass) { return idByMsg.get(msgClass); } public static int getMsgId(Object msg) { return getMsgId(msg.getClass()); } }
服务器Actor系统设计
MiniRPG服务器的Actor系统如下图所示:
TcpServer负责监听TCP连接,连接建立之后,交给Codec处理。Codec将收到的字节编码成消息对象,然后交给MsgHandler处理。对于每条请求消息,MsgHandler都会产生一条响应消息,响应消息被Codec编码之后发送到客户端。下面详细介绍整个Actor系统是如何实现的。
TcpServer
TcpServer是一个UntypedActor,实例化TcpServer时,我们把MsgHandler引用传给它:
public class TcpServer extends UntypedActor { private final ActorRef msgHandler; public TcpServer(ActorRef msgHandler) { this.msgHandler = msgHandler; } }TcpServer只关心四种消息,下面是onReceive()方法实现:
@Override public void onReceive(Object msg) throws Exception { if (msg instanceof Integer) { final int port = (Integer) msg; startServer(port); } else if (msg instanceof Bound) { getSender().tell(msg, getSelf()); } else if (msg instanceof CommandFailed) { getContext().stop(getSelf()); } else if (msg instanceof Connected) { final Connected conn = (Connected) msg; getSender().tell(conn, getSelf()); registerCodec(getSender()); } }
Integer消息通知TcpServer绑定到某个端口,准备接收客户端连接。如果收到Bound消息,则端口绑定成功,服务器正常启动。如果是CommandFailed消息,则服务器启动失败:
private void startServer(int port) { final InetSocketAddress endpoint = new InetSocketAddress("localhost", port); final Object bindCmd = TcpMessage.bind(getSelf(), endpoint, 100); Tcp.get(getContext().system()).getManager() .tell(bindCmd, getSelf()); }
如果是Connected消息,说明有客户端连接已经建立,TcpServer创建一个子Actor(也就是Codec)来处理客户端连接:
private void registerCodec(ActorRef connection) { final Props codecProps = Props.create(MsgCodec.class, connection, msgHandler); final ActorRef codec = getContext().actorOf(codecProps); connection.tell(TcpMessage.register(codec), getSelf()); }
MsgCodec
MsgCodec主要负责消息的编码和解码,为此,MsgCodec内部使用了一个ByteString来缓存接收到的字节:
public class MsgCodec extends UntypedActor { private static final Gson GSON = new Gson(); private final ActorRef connection; private final ActorRef msgHandler; private ByteString buf = ByteString.empty(); public MsgCodec(ActorRef connection, ActorRef msgHandler) { this.connection = connection; this.msgHandler = msgHandler; } }如果MsgCodec收到的是Received消息,说明有数据到达,MsgCodec尝试解码出一个消息对象。如果收到的是GameMessage消息,MsgCodec将其编码为byte[]然后发送给客户端。如果收到的是ConnectionClosed,说明连接已经断开了:
@Override public void onReceive(Object msg) throws Exception { if (msg instanceof Received) { final ByteString data = http://www.mamicode.com/((Received) msg).data();>每当有数据到达时,decodeMsg()方法都会被调用。decodeMsg()先确定是否可以把消息头解码出来,如果不能,就继续等待更多的字节到达。如果消息头完整到达,decodeMsg()就可以知道消息体的长度,然后等到消息体完整到达。之后根据消息ID和JSON字符串解码消息对象,然后通知msgHandler:private void decodeMsg() { while (buf.length() > 8) { final ByteIterator it = buf.iterator(); final int msgId = it.getInt(ByteOrder.BIG_ENDIAN); final int jsonLength = it.getInt(ByteOrder.BIG_ENDIAN); if (buf.length() >= 8 + jsonLength) { final Object msg = decodeMsg(msgId, buf.slice(8, 8 + jsonLength)); buf = buf.drop(8 + jsonLength); msgHandler.tell(msg, getSelf()); } } } private Object decodeMsg(int msgId, ByteString jsonData) { final Class<?> msgClass = MsgRegistry.getMsgClass(msgId); final Reader reader = new InputStreamReader( jsonData.iterator().asInputStream(), StandardCharsets.UTF_8); return GSON.fromJson(reader, msgClass); }消息的编码就简单多了,代码如下所示:private ByteString encodeMsg(Object msg) { final int msgId = MsgRegistry.getMsgId(msg); final byte[] jsonBytes = GSON.toJson(msg) .getBytes(StandardCharsets.UTF_8); final ByteStringBuilder bsb = new ByteStringBuilder(); bsb.putInt(msgId, ByteOrder.BIG_ENDIAN); bsb.putInt(jsonBytes.length, ByteOrder.BIG_ENDIAN); bsb.putBytes(jsonBytes); return bsb.result(); }MsgHandler
游戏逻辑由MsgHandler来处理。因为只是个demo,所以MsgHandler内部使用HashMap来模拟数据库。下面是MsgHandler的完整代码:
public class MsgHandler extends UntypedActor { private final List<Player> players = new ArrayList<>(); @Override public void onReceive(Object msg) throws Exception { if (msg instanceof CreatePlayerRequest) { int newPlayerId = createPlayer((CreatePlayerRequest) msg); getSender().tell(new CreatePlayerResponse(newPlayerId), getSelf()); } else if (msg instanceof AddExpRequest) { int newExp = addExpToPlayer((AddExpRequest) msg); getSender().tell(new AddExpResponse(newExp), getSelf()); } else if (msg instanceof LevelUpRequest) { int newLevel = levelUpPlayer((LevelUpRequest) msg); getSender().tell(new LevelUpResponse(newLevel), getSelf()); } else if (msg instanceof GetPlayerInfoRequest) { PlayerInfo playerInfo = getPlayerInfo((GetPlayerInfoRequest) msg); getSender().tell(new GetPlayerInfoResponse(playerInfo), getSelf()); } } private int createPlayer(CreatePlayerRequest req) { int playerId = players.size() + 1; Player newPlayer = new Player(); newPlayer.setId(playerId); newPlayer.setLevel(1); newPlayer.setName(req.getPlayerName()); players.add(newPlayer); return playerId; } private int addExpToPlayer(AddExpRequest req) { Player player = players.get(req.getPlayerId()); player.addExp(req.getExp()); return player.getExp(); } private int levelUpPlayer(LevelUpRequest req) { Player player = players.get(req.getPlayerId()); player.levelUp(); return player.getLevel(); } private PlayerInfo getPlayerInfo(GetPlayerInfoRequest req) { Player player = players.get(req.getPlayerId()); return new PlayerInfo(player.getId(), player.getName(), player.getExp(), player.getLevel()); } }ServerApp
ServerApp是MiniRPG游戏服务器主类,main()方法建立好整个Actor系统,然后通知tcpServer绑定到端口12345,让服务器运转起来:
public class ServerApp { public static void main(String[] args) { ActorSystem mySystem = ActorSystem.create("rpgServer"); ActorRef msgHandler = mySystem.actorOf(Props.create(MsgHandler.class)); ActorRef tcpServer = mySystem.actorOf(Props.create(TcpServer.class, msgHandler)); tcpServer.tell(12345, ActorRef.noSender()); } }客户端
为了测试MiniServer,我写了个简单的客户端程序,具体实现就不在这里介绍了。
Akka边学边写(4)-- MiniRPG