net.mcsproject.daemon.network.PacketEncoder.java Source code

Introduction

Here is the source code for net.mcsproject.daemon.network.PacketEncoder.java

Source

/*
 *     MCS - Minecraft Cloud System
 *     Copyright (C) 2016
 *
 *     This program is free software: you can redistribute it and/or modify
 *     it under the terms of the GNU Affero General Public License as published by
 *     the Free Software Foundation, either version 3 of the License, or
 *     (at your option) any later version.
 *
 *     This program is distributed in the hope that it will be useful,
 *     but WITHOUT ANY WARRANTY; without even the implied warranty of
 *     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *     GNU Affero General Public License for more details.
 *
 *     You should have received a copy of the GNU Affero General Public License
 *     along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package net.mcsproject.daemon.network;

import io.netty.buffer.ByteBuf;
import io.netty.buffer.ByteBufOutputStream;
import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.codec.MessageToByteEncoder;
import lombok.AllArgsConstructor;
import net.mcsproject.daemon.network.packet.Packet;
import net.mcsproject.daemon.network.packet.PacketRegistry;

@AllArgsConstructor
public class PacketEncoder extends MessageToByteEncoder<Packet> {

    private PacketRegistry packetRegistry;

    @Override
    protected void encode(ChannelHandlerContext channelHandlerContext, Packet packet, ByteBuf byteBuf)
            throws Exception {
        byte id = packetRegistry.getIdByPacket(packet.getClass());

        byteBuf.writeByte(id);
        packet.write(new ByteBufOutputStream(byteBuf));
    }

}