com.ebay.myriad.scheduler.TaskTerminator.java Source code

Introduction

Here is the source code for com.ebay.myriad.scheduler.TaskTerminator.java

Source

/**
 * Copyright 2012-2014 eBay Software Foundation, All Rights Reserved.
 * 
 * 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.ebay.myriad.scheduler;

import java.util.Iterator;
import java.util.Set;

import javax.inject.Inject;

import org.apache.commons.collections4.CollectionUtils;
import org.apache.mesos.Protos.Status;
import org.apache.mesos.Protos.TaskID;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.ebay.myriad.configuration.MyriadConfiguration;
import com.ebay.myriad.state.NodeTask;
import com.ebay.myriad.state.SchedulerState;
import com.google.common.base.Preconditions;

/**
 * {@link TaskTerminator} is responsible for killing tasks.
 *
 */
public class TaskTerminator implements Runnable {
    private final static Logger LOGGER = LoggerFactory.getLogger(MyriadDriverManager.class);

    private MyriadConfiguration cfg;
    private SchedulerState schedulerState;
    private MyriadDriverManager driverManager;

    @Inject
    public TaskTerminator(MyriadConfiguration cfg, SchedulerState schedulerState,
            MyriadDriverManager driverManager) {
        this.cfg = cfg;
        this.schedulerState = schedulerState;
        this.driverManager = driverManager;
    }

    @Override
    public void run() {
        Set<String> killableTasks = schedulerState.getKillableTasks();

        if (CollectionUtils.isEmpty(killableTasks)) {
            return;
        }

        Status driverStatus = driverManager.getDriverStatus();
        if (Status.DRIVER_RUNNING != driverStatus) {
            LOGGER.warn("Cannot kill tasks, as driver is not running. Status: {}", driverStatus);
            return;
        }

        Iterator<String> iterator = killableTasks.iterator();

        while (iterator.hasNext()) {
            String taskIdToKill = iterator.next();
            NodeTask task = this.schedulerState.getTask(taskIdToKill);
            TaskID mesosTaskId = task.getMesosTaskId();
            Status status = this.driverManager.kill(mesosTaskId);
            this.schedulerState.removeTask(taskIdToKill);
            Preconditions.checkState(status == Status.DRIVER_RUNNING);
        }
    }
}