Mindustry/kryonet/src/io/anuke/kryonet/ByteSerializer.java

package io.anuke.kryonet;

import com.esotericsoftware.kryonet.FrameworkMessage;
import com.esotericsoftware.kryonet.serialization.Serialization;
import io.anuke.mindustry.net.Packet;
import io.anuke.mindustry.net.Registrator;
import io.anuke.ucore.function.Supplier;
import io.anuke.ucore.util.Pooling;

import java.nio.ByteBuffer;

@SuppressWarnings("unchecked")
public class ByteSerializer implements Serialization{

    @Override
    public void write(ByteBuffer byteBuffer, Object o){
        if(o instanceof FrameworkMessage){
            byteBuffer.put((byte) -2); //code for framework message
            FrameworkSerializer.write(byteBuffer, (FrameworkMessage) o);
        }else{
            if(!(o instanceof Packet))
                throw new RuntimeException("All sent objects must implement be Packets! Class: " + o.getClass());
            byte id = Registrator.getID(o.getClass());
            if(id == -1)
                throw new RuntimeException("Unregistered class: " + o.getClass());
            byteBuffer.put(id);
            ((Packet) o).write(byteBuffer);
        }
    }

    @Override
    public Object read(ByteBuffer byteBuffer){
        byte id = byteBuffer.get();
        if(id == -2){
            return FrameworkSerializer.read(byteBuffer);
        }else{
            Packet packet = Pooling.obtain((Class<Packet>) Registrator.getByID(id).type, (Supplier<Packet>) Registrator.getByID(id).constructor);
            packet.read(byteBuffer);
            return packet;
        }
    }

    @Override
    public int getLengthLength(){
        return 2;
    }

    @Override
    public void writeLength(ByteBuffer byteBuffer, int i){
        byteBuffer.putShort((short) i);
    }

    @Override
    public int readLength(ByteBuffer byteBuffer){
        return byteBuffer.getShort();
    }
}