com.microsoft.reef.poison.task.PoisonedTaskStartHandler Maven / Gradle / Ivy
/**
* Copyright (C) 2014 Microsoft Corporation
*
* 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 com.microsoft.reef.poison.task;
import java.util.Random;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.inject.Inject;
import com.microsoft.reef.poison.PoisonException;
import com.microsoft.reef.poison.PoisonedAlarmHandler;
import com.microsoft.reef.poison.params.CrashProbability;
import com.microsoft.reef.poison.params.CrashTimeout;
import com.microsoft.reef.task.events.TaskStart;
import com.microsoft.tang.annotations.Parameter;
import com.microsoft.wake.EventHandler;
import com.microsoft.wake.time.Clock;
public final class PoisonedTaskStartHandler implements EventHandler {
private static final Logger LOG = Logger.getLogger(PoisonedTaskStartHandler.class.getName());
private final Random random = new Random();
private final double crashProbability;
private final int timeOut;
private final Clock clock;
@Inject
public PoisonedTaskStartHandler(
final @Parameter(CrashProbability.class) double crashProbability,
final @Parameter(CrashTimeout.class) int timeOut,
final Clock clock) {
this.crashProbability = crashProbability;
this.timeOut = timeOut;
this.clock = clock;
}
@Override
public void onNext(final TaskStart taskStart) {
LOG.log(Level.INFO, "Starting Task poison injector with prescribed dose: {0} units",
this.crashProbability);
if (this.random.nextDouble() <= this.crashProbability) {
final int timeToCrash = this.random.nextInt(this.timeOut) * 1000;
LOG.log(Level.INFO, "Dosage lethal! Crashing in {0} msec.", timeToCrash);
if (timeToCrash == 0) {
throw new PoisonException("Crashed at: " + System.currentTimeMillis());
} else {
this.clock.scheduleAlarm(timeToCrash, new PoisonedAlarmHandler());
}
} else {
LOG.info("Dosage not lethal");
}
}
}