Source code

001/*
002 * Licensed to the Apache Software Foundation (ASF) under one
003 * or more contributor license agreements.  See the NOTICE file
004 * distributed with this work for additional information
005 * regarding copyright ownership.  The ASF licenses this file
006 * to you under the Apache License, Version 2.0 (the
007 * "License"); you may not use this file except in compliance
008 * with the License.  You may obtain a copy of the License at
009 *
010 *   http://www.apache.org/licenses/LICENSE-2.0
011 *
012 * Unless required by applicable law or agreed to in writing,
013 * software distributed under the License is distributed on an
014 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
015 * KIND, either express or implied.  See the License for the
016 * specific language governing permissions and limitations
017 * under the License.
018 */
019package org.apache.reef.io.watcher;
020
021import org.apache.reef.annotations.Unstable;
022import org.apache.reef.driver.parameters.*;
023import org.apache.reef.io.watcher.param.EventStreams;
024import org.apache.reef.tang.formats.*;
025import org.apache.reef.wake.time.Clock;
026
027/**
028 * ConfigurationModule for Watcher.
029 */
030@Unstable
031public final class WatcherConfiguration extends ConfigurationModuleBuilder {
032
033  /**
034   * Types of EventStream where events subscribed by Watcher will be reported.
035   */
036  public static final OptionalImpl<EventStream> EVENT_STREAMS = new OptionalImpl<>();
037
038  public static final ConfigurationModule CONF = new WatcherConfiguration()
039      .bindSetEntry(EventStreams.class, EVENT_STREAMS)
040      .bindSetEntry(Clock.RuntimeStartHandler.class, Watcher.DriverRuntimeStartHandler.class)
041      .bindSetEntry(Clock.StartHandler.class, Watcher.DriverStartHandler.class)
042      .bindSetEntry(Clock.StopHandler.class, Watcher.DriverStopHandler.class)
043      .bindSetEntry(Clock.RuntimeStopHandler.class, Watcher.DriverRuntimeStopHandler.class)
044      .bindSetEntry(ServiceContextActiveHandlers.class, Watcher.ContextActiveHandler.class)
045      .bindSetEntry(ServiceContextClosedHandlers.class, Watcher.ContextClosedHandler.class)
046      .bindSetEntry(ServiceContextFailedHandlers.class, Watcher.ContextFailedHandler.class)
047      .bindSetEntry(ServiceTaskFailedHandlers.class, Watcher.TaskFailedHandler.class)
048      .bindSetEntry(ServiceTaskCompletedHandlers.class, Watcher.TaskCompletedHandler.class)
049      .bindSetEntry(ServiceTaskMessageHandlers.class, Watcher.TaskMessageHandler.class)
050      .bindSetEntry(ServiceTaskRunningHandlers.class, Watcher.TaskRunningHandler.class)
051      .bindSetEntry(ServiceTaskSuspendedHandlers.class, Watcher.TaskSuspendedHandler.class)
052      .bindSetEntry(ServiceEvaluatorAllocatedHandlers.class, Watcher.EvaluatorAllocatedHandler.class)
053      .bindSetEntry(ServiceEvaluatorFailedHandlers.class, Watcher.EvaluatorFailedHandler.class)
054      .bindSetEntry(ServiceEvaluatorCompletedHandlers.class, Watcher.EvaluatorCompletedHandler.class)
055      .build();
056
057  /**
058   * Construct a WatcherConfiguration.
059   */
060  private WatcherConfiguration() {
061  }
062}