net.lenni0451.mcping.pings.AUDPPing Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of MCPing Show documentation
A simple library to ping minecraft servers
The newest version!
package net.lenni0451.mcping.pings;

import net.lenni0451.mcping.ServerAddress;
import net.lenni0451.mcping.pings.sockets.factories.IUDPSocketFactory;
import net.lenni0451.mcping.pings.sockets.types.IUDPSocket;

import java.io.*;

/**
 * The abstract class used to implement UDP based pings.
 */
public abstract class AUDPPing extends APing {

    private IUDPSocketFactory socketFactory;
    private final int readTimeout;
    private IUDPSocket socket;

    public AUDPPing(final IUDPSocketFactory socketFactory, final int readTimeout) {
        this.socketFactory = socketFactory;
        this.readTimeout = readTimeout;
    }

    /**
     * Create a new datagram socket.
     *
     * @param serverAddress The server address
     * @return The created socket
     * @throws IOException If an I/O error occurs
     */
    protected final IUDPSocket connect(final ServerAddress serverAddress) throws IOException {
        this.socket = this.socketFactory.create(serverAddress, this.readTimeout);
        this.socket.connect();
        return this.socket;
    }

    @Override
    public void close() throws Exception {
        if (this.socket != null) this.socket.close();
    }

    /**
     * Write a packet and send it to the given server address.
     *
     * @param s            The UDP socket
     * @param packetWriter The packet writer
     * @throws IOException If an I/O error occurs
     */
    protected void writePacket(final IUDPSocket s, final PacketWriter packetWriter) throws IOException {
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        DataOutputStream os = new DataOutputStream(baos);
        packetWriter.write(os);
        s.send(baos.toByteArray());
    }

    /**
     * Read a packet from the given socket.
     *
     * @param s            The UDP socket
     * @param bufferSize   The buffer size
     * @param packetReader The packet reader
     * @throws IOException If an I/O error occurs
     */
    protected void readPacket(final IUDPSocket s, final int bufferSize, PacketReader packetReader) throws IOException {
        byte[] data = s.receive(bufferSize);
        DataInputStream is = new DataInputStream(new ByteArrayInputStream(data));
        packetReader.read(is);
    }


    @FunctionalInterface
    protected interface PacketWriter {
        void write(final DataOutputStream os) throws IOException;
    }

    @FunctionalInterface
    protected interface PacketReader {
        void read(final DataInputStream is) throws IOException;
    }

}