/**
 * 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.preemptor;

import java.util.Objects;

import com.google.common.base.MoreObjects;

import org.apache.aurora.scheduler.resources.ResourceBag;
import org.apache.aurora.scheduler.resources.ResourceManager;
import org.apache.aurora.scheduler.storage.entities.IAssignedTask;
import org.apache.aurora.scheduler.storage.entities.ITaskConfig;

import static java.util.Objects.requireNonNull;

/**
 * A victim to be considered as a candidate for preemption.
 */
public final class PreemptionVictim {
  private final IAssignedTask task;

  private PreemptionVictim(IAssignedTask task) {
    this.task = requireNonNull(task);
  }

  public static PreemptionVictim fromTask(IAssignedTask task) {
    return new PreemptionVictim(task);
  }

  public String getSlaveHost() {
    return task.getSlaveHost();
  }

  public boolean isProduction() {
    return task.getTask().isProduction();
  }

  public String getRole() {
    return task.getTask().getJob().getRole();
  }

  public int getPriority() {
    return task.getTask().getPriority();
  }

  public ResourceBag getResourceBag() {
    return ResourceManager.bagFromResources(task.getTask().getResources());
  }

  public String getTaskId() {
    return task.getTaskId();
  }

  public ITaskConfig getConfig() {
    return task.getTask();
  }

  @Override
  public boolean equals(Object o) {
    if (!(o instanceof PreemptionVictim)) {
      return false;
    }

    PreemptionVictim other = (PreemptionVictim) o;
    return Objects.equals(task, other.task);
  }

  @Override
  public int hashCode() {
    return Objects.hashCode(task);
  }

  @Override
  public String toString() {
    return MoreObjects.toStringHelper(this)
        .add("task", task)
        .toString();
  }
}