de.sanandrew.core.manpack.network.PacketProcessor.java Source code

Introduction

Here is the source code for de.sanandrew.core.manpack.network.PacketProcessor.java

Source

/*******************************************************************************************************************
 * Authors:   SanAndreasP
 * Copyright: SanAndreasP
 * License:   Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International
 *                http://creativecommons.org/licenses/by-nc-sa/4.0/
 *******************************************************************************************************************/
package de.sanandrew.core.manpack.network;

import com.google.common.collect.Maps;
import de.sanandrew.core.manpack.init.ManPackLoadingPlugin;
import io.netty.buffer.ByteBuf;
import io.netty.buffer.ByteBufInputStream;
import net.minecraft.network.INetHandler;
import org.apache.logging.log4j.Level;

import java.io.IOException;
import java.lang.reflect.InvocationTargetException;
import java.util.Map;

public final class PacketProcessor {
    private final String modId;
    private final Map<Short, Class<? extends IPacket>> packetMap = Maps.newHashMap();

    public PacketProcessor(String modId) {
        this.modId = modId;
    }

    public void processPacket(ByteBuf data, INetHandler handler) {
        short packetId = -1;
        try (ByteBufInputStream bbis = new ByteBufInputStream(data)) {
            packetId = bbis.readShort();
            if (this.packetMap.containsKey(packetId)) {
                this.packetMap.get(packetId).getConstructor().newInstance().process(bbis, data, handler);
            }
        } catch (IOException ioe) {
            ManPackLoadingPlugin.MOD_LOG.log(Level.ERROR, "The packet with the ID %d from %s cannot be processed!",
                    packetId, this.modId);
            ioe.printStackTrace();
        } catch (IllegalAccessException | InstantiationException rex) {
            ManPackLoadingPlugin.MOD_LOG.log(Level.ERROR,
                    "The packet with the ID %d from %s cannot be instantiated!", packetId, this.modId);
            rex.printStackTrace();
        } catch (NoSuchMethodException | InvocationTargetException e) {
            e.printStackTrace();
        }
    }

    public Class<? extends IPacket> getPacketCls(short packetId) {
        return this.packetMap.get(packetId);
    }

    public void addPacketCls(int packetId, Class<? extends IPacket> packetCls) {
        this.packetMap.put((short) packetId, packetCls);
    }
}