code

    @Override
    public void encode(updateBlockStateMessage message, PacketBuffer buffer) {
        buffer.writeInt(message.entity);
        buffer.writeFloat(message.BlockStateID);
    }

    @Override
    public updateBlockStateMessage decode(PacketBuffer buffer) {
        return new updateBlockStateMessage(buffer.readInt(), buffer.readFloat());
    }

    @Override
    public void handle(updateBlockStateMessage message, Supplier<NetworkEvent.Context> contextSupplier) {
        NetworkEvent.Context context = contextSupplier.get();
        Optional<World> world;
        world = LogicalSidedProvider.CLIENTWORLD.get(context.getDirection().getReceptionSide());
        if(context.getDirection() == NetworkDirection.PLAY_TO_CLIENT){
            try {
                context.enqueueWork(() ->
                        ((FlexibleBlockEntity) world.get().getEntityByID(message.entity))
                        .blockState = Block.getStateById((int) message.BlockStateID));
                context.setPacketHandled(true);
            }catch (NullPointerException nullPointerException){
                FlexibleBlockMod.LOGGER.info(nullPointerException);
            }
        }