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.runtime.common.driver.idle; 020 021import org.apache.reef.runtime.common.driver.evaluator.Evaluators; 022import org.apache.reef.tang.InjectionFuture; 023 024import javax.inject.Inject; 025 026/** 027 * Checks for idleness of the Event handlers. 028 */ 029public final class EventHandlerIdlenessSource implements DriverIdlenessSource { 030 031 private static final IdleMessage IDLE_MESSAGE = 032 new IdleMessage("EventHandlers", "All events have been processed.", true); 033 private static final IdleMessage NOT_IDLE_MESSAGE = 034 new IdleMessage("EventHandlers", "Some events are still in flight.", false); 035 036 private final InjectionFuture<Evaluators> evaluators; 037 private final InjectionFuture<DriverIdleManager> driverIdleManager; 038 039 @Inject 040 EventHandlerIdlenessSource(final InjectionFuture<Evaluators> evaluators, 041 final InjectionFuture<DriverIdleManager> driverIdleManager) { 042 this.evaluators = evaluators; 043 this.driverIdleManager = driverIdleManager; 044 } 045 046 047 @Override 048 public IdleMessage getIdleStatus() { 049 if (this.evaluators.get().allEvaluatorsAreClosed()) { 050 return IDLE_MESSAGE; 051 } else { 052 return NOT_IDLE_MESSAGE; 053 } 054 } 055 056 public void check() { 057 if (this.evaluators.get().allEvaluatorsAreClosed()) { 058 this.driverIdleManager.get().onPotentiallyIdle(IDLE_MESSAGE); 059 } 060 } 061}