Java Code Examples for org.apache.mesos.Protos#ResourceProviderID

/**
 * ResourceLabels are used to map {@link com.mesosphere.sdk.specification.ServiceSpec} entities to resources which
 * have actually been consumed.  Generally speaking there should be no difference between the resources from a Mesos
 * perspective and the resources from a {@link com.mesosphere.sdk.specification.ServiceSpec} perspective.
 * <p>
 * There is one special case in which they differ, MOUNT volume resources.  In this case the {@link ResourceSpec}
 * and actually consumed resources differ in value.  Because MOUNT volumes are atomic if the {@link ResourceSpec}
 * indicates a need for 25 GB of disk and 50 GB of disk are actually offered, then 50 GB must be consumed.
 * <p>
 * To capture this change the concept of an "updated" ResourceSpec is needed.  We modify the ResourceSpec in this
 * scenario to match the actually consumed resources to facilitate consistent protobuf construction.
 */
ResourceLabels(
    ResourceSpec original,
    ResourceSpec updated,
    String resourceId,
    Optional<String> resourceNamespace,
    Optional<String> persistenceId,
    Optional<Protos.ResourceProviderID> providerId,
    Optional<Protos.Resource.DiskInfo.Source> diskSource,
    Optional<String> frameworkId)
{
  this.original = original;
  this.updated = updated;
  this.resourceId = resourceId;
  this.resourceNamespace = resourceNamespace;
  this.persistenceId = persistenceId;
  this.providerId = providerId;
  this.diskSource = diskSource;
  this.frameworkId = frameworkId;
}
 

public static Protos.Resource getExistingExecutorVolume(
    VolumeSpec volumeSpec,
    Optional<String> resourceId,
    Optional<String> resourceNamespace,
    Optional<String> persistenceId,
    Optional<Protos.ResourceProviderID> providerId,
    Optional<Protos.Resource.DiskInfo.Source> diskSource,
    Optional<String> frameworkId)
{

  Protos.Resource.Builder builder = ResourceBuilder
      .fromSpec(volumeSpec, resourceId, resourceNamespace, persistenceId, providerId, diskSource, frameworkId)
      .build()
      .toBuilder();

  return builder.build();
}
 

public static VolumeEvaluationStage getExisting(
    VolumeSpec volumeSpec,
    Collection<String> taskNames,
    Optional<String> resourceId,
    Optional<String> resourceNamespace,
    Optional<String> persistenceId,
    Optional<Protos.ResourceProviderID> providerId,
    Optional<Protos.Resource.DiskInfo.Source> diskSource,
    Optional<String> frameworkId)
{
  return new VolumeEvaluationStage(
      volumeSpec,
      taskNames,
      resourceId,
      resourceNamespace,
      persistenceId,
      providerId,
      diskSource,
      frameworkId);
}
 

private VolumeEvaluationStage(
    VolumeSpec volumeSpec,
    Collection<String> taskNames,
    Optional<String> resourceId,
    Optional<String> resourceNamespace,
    Optional<String> persistenceId,
    Optional<Protos.ResourceProviderID> providerId,
    Optional<Protos.Resource.DiskInfo.Source> diskSource,
    Optional<String> frameworkId)
{
  this.logger = LoggingUtils.getLogger(getClass(), resourceNamespace);
  this.volumeSpec = volumeSpec;
  this.taskNames = taskNames;
  this.resourceId = resourceId;
  this.resourceNamespace = resourceNamespace;
  this.persistenceId = persistenceId;
  this.providerId = providerId;
  this.diskSource = diskSource;
  this.frameworkId = frameworkId;
}
 

public static ResourceBuilder fromSpec(
    VolumeSpec spec,
    Optional<String> resourceId,
    Optional<String> resourceNamespace,
    Optional<String> persistenceId,
    Optional<Protos.ResourceProviderID> providerId,
    Optional<Protos.Resource.DiskInfo.Source> diskSource,
    Optional<String> frameworkId)
{
  ResourceBuilder resourceBuilder = fromSpec(spec, resourceId, resourceNamespace, frameworkId);

  providerId.ifPresent(resourceBuilder::setProviderId);

  switch (spec.getType()) {
    case ROOT:
      if (diskSource.isPresent()) {
        throw new IllegalStateException("Source must not be set on a ROOT volume");
      }
      return resourceBuilder.setRootVolume(spec.getContainerPath(), persistenceId);
    case MOUNT:
      if (!diskSource.isPresent()) {
        throw new IllegalStateException("Source must be set on a MOUNT volume");
      }
      return resourceBuilder
          .setMountVolume(spec.getContainerPath(), persistenceId, diskSource.get());
    default:
      throw new IllegalStateException(String.format("Unexpected disk type: %s", spec.getType()));
  }
}
 

public static Optional<Protos.ResourceProviderID> getProviderId(Protos.Resource resource) {
  if (resource.hasProviderId()) {
    return Optional.of(resource.getProviderId());
  }

  return Optional.empty();
}
 

private static OfferEvaluationStage toEvaluationStage(
    Collection<String> taskSpecNames,
    ResourceSpec resourceSpec,
    Optional<String> resourceId,
    Optional<String> resourceNamespace,
    Optional<String> persistenceId,
    Optional<Protos.ResourceProviderID> providerId,
    Optional<Protos.Resource.DiskInfo.Source> diskSource,
    Optional<String> frameworkId)
{
  if (resourceSpec instanceof NamedVIPSpec) {
    return new NamedVIPEvaluationStage(
        (NamedVIPSpec) resourceSpec, taskSpecNames, resourceId, resourceNamespace, frameworkId);
  } else if (resourceSpec instanceof PortSpec) {
    return new PortEvaluationStage(
        (PortSpec) resourceSpec,
        taskSpecNames,
        resourceId,
        resourceNamespace,
        frameworkId);
  } else if (resourceSpec instanceof VolumeSpec) {
    return VolumeEvaluationStage.getExisting(
        (VolumeSpec) resourceSpec,
        taskSpecNames,
        resourceId,
        resourceNamespace,
        persistenceId,
        providerId,
        diskSource,
        frameworkId);
  } else {
    return new ResourceEvaluationStage(
        resourceSpec,
        taskSpecNames,
        resourceId,
        resourceNamespace,
        frameworkId);
  }
}
 

/**
 * Assigns the resource provider id to the provided value.
 */
private ResourceBuilder setProviderId(Protos.ResourceProviderID providerId) {
  this.providerId = Optional.of(providerId);
  return this;
}
 

public Optional<Protos.ResourceProviderID> getProviderId() {
  return providerId;
}