Cron.java

/**
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.apache.aurora.scheduler.http;

import java.util.Map;

import javax.inject.Inject;
import javax.ws.rs.GET;
import javax.ws.rs.Path;
import javax.ws.rs.Produces;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;

import com.google.common.collect.ImmutableMap;

import org.apache.aurora.scheduler.base.JobKeys;
import org.apache.aurora.scheduler.cron.CronJobManager;
import org.apache.aurora.scheduler.cron.CrontabEntry;
import org.apache.aurora.scheduler.storage.entities.IJobKey;

/**
 * HTTP interface to dump state of the internal cron scheduler.
 */
@Path("/cron")
public class Cron {
  private final CronJobManager cronManager;

  @Inject
  Cron(CronJobManager cronManager) {
    this.cronManager = cronManager;
  }

  /**
   * Dumps the state of the cron manager.
   *
   * @return An HTTP response containing the cron manager's state.
   */
  @GET
  @Produces(MediaType.APPLICATION_JSON)
  public Response dumpContents() {
    ImmutableMap.Builder<String, String> scheduled = ImmutableMap.builder();
    for (Map.Entry<IJobKey, CrontabEntry> entry : cronManager.getScheduledJobs().entrySet()) {
      scheduled.put(JobKeys.canonicalString(entry.getKey()), entry.getValue().toString());
    }

    return Response.ok(ImmutableMap.of("scheduled", scheduled.build())).build();
  }
}