Add mesos primitives, Unavailability support

Author: zmalik

src/main/scala/org/apache/mesos/chronos/scheduler/mesos/AvailabilityChecker.scala

@@ -0,0 +1,31 @@
+package org.apache.mesos.chronos.scheduler.mesos
+
+import java.util.logging.Logger
+
+import org.apache.mesos.Protos
+
+/**
+  * Helper for checking availability using mesos primitives
+  */
+object AvailabilityChecker {
+
+  private[this] val log = Logger.getLogger(getClass.getName)
+
+  def checkAvailability(offer: Protos.Offer): Boolean = {
+    var unavailability = offer.hasUnavailability
+    var now = System.nanoTime()
+    if (offer.hasUnavailability && offer.getUnavailability.hasStart) {
+      val start = offer.getUnavailability.getStart.getNanoseconds
+      if (now.>=(start)) {
+        if (offer.getUnavailability.hasDuration) {
+          return start.+(offer.getUnavailability.getDuration.getNanoseconds).>(now)
+        } else {
+          return false;
+        }
+
+      }
+    }
+    return true
+  }
+
+}

src/main/scala/org/apache/mesos/chronos/scheduler/mesos/MesosJobFramework.scala

@@ -131,7 +131,7 @@ class MesosJobFramework @Inject()(
               case None =>
                 val neededResources = new Resources(job)
                 offerResources.toIterator.find { ors =>
-                  ors._2.canSatisfy(neededResources) && ConstraintChecker.checkConstraints(ors._1, job.constraints)
+                  ors._2.canSatisfy(neededResources) && ConstraintChecker.checkConstraints(ors._1, job.constraints) && AvailabilityChecker.checkAvailability(ors._1)
                 } match {
                   case Some((offer, resources)) =>
                     // Subtract this job's resource requirements from the remaining available resources in this offer.