Class TestScheduler

src/examples/java/V1TestFramework.java:44–308

public class V1TestFramework {
  static class TestScheduler implements Scheduler {

    public TestScheduler(
        String master,
        FrameworkInfo framework,
        ExecutorInfo executor) {
      this(master, framework, executor, 5);
    }

    public TestScheduler(
        String master,
        FrameworkInfo framework,
        ExecutorInfo executor,
        int totalTasks) {
      this.framework = framework;
      this.executor = executor;
      this.totalTasks = totalTasks;
      this.state = State.DISCONNECTED;
    }

    // TODO(anand): Synchronize on `state` instead.
    @Override
    public synchronized void connected(final Mesos mesos) {
      System.out.println("Connected");

      state = State.CONNECTED;

      retryTimer = new Timer();
      retryTimer.schedule(new TimerTask() {
        @Override
        public void run() {
          doReliableRegistration(mesos);
        }
      }, 0, 1000); // Repeat every 1 second
    }

    @Override
    public synchronized void disconnected(Mesos mesos) {
      System.out.println("Disconnected");

      state = state.DISCONNECTED;
      cancelRetryTimer();
    }

    @Override
    public synchronized void received(Mesos mesos, Event event) {
      switch (event.getType()) {
        case SUBSCRIBED: {
          frameworkId = event.getSubscribed().getFrameworkId();
          state = State.SUBSCRIBED;

          System.out.println(
              "Subscribed with ID " + frameworkId + " to master " +
              event.getSubscribed().getMasterInfo().getId());
          break;
        }

        case OFFERS: {