Server.java

Go to the documentation of this file.

package rlpark.plugin.robot.internal.disco;

import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
import java.nio.ByteOrder;

import rlpark.plugin.robot.internal.disco.io.DiscoSocket;

import zephyr.plugin.core.api.signals.Signal;

public class Server {
  public Signal<DiscoSocket> onAccept = new Signal<DiscoSocket>();
  private final ServerSocket socket;
  public final int port;
  protected Thread acceptThread = null;
  private final ByteOrder byteOrder;

  public Server() throws IOException {
    this(0);
  }

  public Server(int port) throws IOException {
    this(port, ByteOrder.BIG_ENDIAN);
  }

  public Server(int port, ByteOrder byteOrder) throws IOException {
    socket = new ServerSocket(port);
    this.port = socket.getLocalPort();
    this.byteOrder = byteOrder;
  }

  public DiscoSocket accept() throws IOException {
    Socket clientSocket = socket.accept();
    return new DiscoSocket(clientSocket, byteOrder);
  }

  public void runAcceptThread() {
    assert acceptThread == null;
    acceptThread = new Thread() {
      @Override
      public void run() {
        while (acceptThread != null)
          try {
            onAccept.fire(accept());
          } catch (IOException e) {
            e.printStackTrace();
            break;
          }
        acceptThread = null;
      }
    };
    acceptThread.start();
  }

  public void close() throws IOException {
    acceptThread = null;
    socket.close();
  }
}