Honor Message return value

- Client -> Send to server - Server -> Send back to player that message targeted. Signed-off-by: Chris Sanders <zidane@outlook.com>
ObsidianBox · Jul 6, 2014 · 15d6f6a · 15d6f6a
1 parent a43ad6d
commit 15d6f6a
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/main/java/org/obsidianbox/obsidian/message/CommonMessagePipeline.java b/src/main/java/org/obsidianbox/obsidian/message/CommonMessagePipeline.java
@@ -99,14 +99,21 @@ public void decodeInto(ChannelHandlerContext ctx, ByteBuf source, Message messag
         if (handler == null) {
             return;
         }
+        Message toSendBack;
         // Handle the message
         switch (game.getSide()) {
             case CLIENT:
-                handler.handle(game, getClientPlayer(), message);
+                toSendBack = handler.handle(game, getClientPlayer(), message);
+                if (toSendBack != null) {
+                    sendToServer(toSendBack);
+                }
                 break;
             case SERVER:
                 INetHandler net = ctx.channel().attr(NetworkRegistry.NET_HANDLER).get();
-                handler.handle(game, ((NetHandlerPlayServer) net).playerEntity, message);
+                toSendBack = handler.handle(game, ((NetHandlerPlayServer) net).playerEntity, message);
+                if (toSendBack != null) {
+                    sendTo(toSendBack, ((NetHandlerPlayServer) net).playerEntity);
+                }
                 break;
             default:
         }