Сообщения, доставка сообщений, фикс хэндшейков, буферная зона (для синхронмзации)

src/main/java/io/orprotocol/Server.java

@@ -17,6 +17,7 @@ import io.orprotocol.index.ClientIndexer;
 import io.orprotocol.lock.ThreadLocker;
 import io.orprotocol.packet.Packet;
 import io.orprotocol.packet.PacketExecutor;
+import io.orprotocol.packet.PacketFactory;
 import io.orprotocol.packet.PacketManager;
 
 public class Server extends WebSocketServer {
@@ -95,37 +96,13 @@ public class Server extends WebSocketServer {
     public void onMessage(WebSocket socket, ByteBuffer byteBuffer) {
         Client client = socket.getAttachment();
         byte[] bytes = byteBuffer.array();
-        Stream stream = new Stream(bytes);
-        int packetId = stream.readInt16();
-        /**
-         * Обновляем время последнего полученного heartbeat.
-         * Так как клиент отпраивл нам пакет, он живой.
-         */
-        client.updateHeartbeat();
-
-        if(!this.packetManager.hasPacketSupported(packetId)){
-            /**
-             * Если пакет не поддерживается, отключаем клиента с соответствующим кодом ошибки.
-             */
-            client.disconnect(ServerFailures.UNSUPPORTED_PACKET);
-            return;
-        }
-        if(!this.packetManager.hasExecutorDelegated(packetId)){
-            /**
-             * Если для пакета не назначен обработчик, отключаем клиента с соответствующим кодом ошибки.
-             */
-            client.disconnect(ServerFailures.UNSUPPORTED_PACKET);
-            return;
-        }
-
-        Class<? extends Packet> packetClass = this.packetManager.getPacketClass(packetId);
         try {
-            Packet packet = packetClass.getConstructor().newInstance();
-            packet.packetId = packetId;
             /**
-             * Читаем данные пакета из потока.
+             * Создаем пакет из полученных байтов.
              */
-            packet.read(stream);
+            PacketFactory packetFactory = new PacketFactory(bytes, this.packetManager);
+            Packet packet = packetFactory.createPacket();
+            int packetId = packetFactory.getPacketId();
             /**
              * Получаем обработчик пакета и вызываем его метод обработки.
              * 
@@ -161,8 +138,8 @@ public class Server extends WebSocketServer {
                 threadLocker.releaseLock(packet, executor.getClass());
             }
         } catch (Exception e) {
-            System.out.println("Error while processing packet " + packetClass.getName());
             e.printStackTrace();
+            //client.disconnect(ServerFailures.UNSUPPORTED_PACKET);
         }
     }    
 

src/main/java/io/orprotocol/packet/PacketFactory.java

@@ -0,0 +1,52 @@
+package io.orprotocol.packet;
+
+import io.orprotocol.ProtocolException;
+import io.orprotocol.Stream;
+
+/**
+ * Фабрика для создания пакетов из байтового массива. Используется для создания пакетов при получении данных от клиента,
+ * а так же может быть использована приложением
+ */
+public class PacketFactory {
+
+    private byte[] bytes;
+    private PacketManager packetManager;
+
+    /**
+     * Создать фабрику для создания пакетов из байтового массива
+     * @param bytes байтовый массив для создания пакета
+     * @param packetManager менеджер пакетов для получения класса пакета по его id
+     */
+    public PacketFactory(byte[] bytes, PacketManager packetManager) {
+        this.bytes = bytes;
+        this.packetManager = packetManager;
+    }
+
+    /**
+     * Создает пакет из массива байт, сериализует и возвращает его. Если пакет с таким id не поддерживается, выбрасывает ProtocolException
+     * @return созданный пакет
+     * @throws ProtocolException
+     */
+    public Packet createPacket() throws ProtocolException {
+        Stream stream = new Stream(this.bytes);
+        int packetId = stream.readInt16();
+        if(!this.packetManager.hasPacketSupported(packetId)){
+            throw new ProtocolException("Unsupported packet with id " + packetId);
+        }
+        Class<? extends Packet> packetClass = this.packetManager.getPacketClass(packetId);
+        try {
+            Packet packet = packetClass.getConstructor().newInstance();
+            packet.packetId = packetId;
+            packet.read(stream);
+            return packet;
+        } catch (Exception e) {
+            throw new ProtocolException("Failed to create packet with id " + packetId);
+        }
+    }
+
+    public int getPacketId() {
+        Stream stream = new Stream(this.bytes);
+        return stream.readInt16();
+    }
+
+}

src/main/java/io/orprotocol/packet/PacketManager.java

@@ -2,6 +2,8 @@ package io.orprotocol.packet;
 
 import java.util.HashMap;
 
+import io.orprotocol.ProtocolException;
+
 /**
  * Менеджер сетевых пакетов и их обработчиков.
  */
@@ -96,5 +98,24 @@ public class PacketManager {
         return this.packets.size();
     }
 
+    /**
+     * Создает пакет из массива байт, сериализует и возвращает его. Если пакет с таким id не поддерживается, выбрасывает ProtocolException
+     * @param bytes байтовый массив для создания пакета
+     * @return созданный пакет
+     * @throws ProtocolException если пакет с таким id не поддерживается или произошла ошибка при создании пакета
+     */
+    public Packet createPacket(byte[] bytes) throws ProtocolException {
+        PacketFactory packetFactory = new PacketFactory(bytes, this);
+        return packetFactory.createPacket();
+    }
+
+    /**
+     * Создает фабрику для создания пакетов из байтового массива
+     * @param bytes байтовый массив для создания пакета
+     * @return фабрика для создания пакетов из байтового массива
+     */
+    public PacketFactory getPacketFactory(byte[] bytes) {
+        return new PacketFactory(bytes, this);
+    }
 
 }