コード例 #1
0
  @Override
  public synchronized void submit(final EvaluatorRequest req) {
    LOG.log(
        Level.FINEST,
        "Got an EvaluatorRequest: number: {0}, memory = {1}, cores = {2}.",
        new Object[] {req.getNumber(), req.getMegaBytes(), req.getNumberOfCores()});

    if (req.getMegaBytes() <= 0) {
      throw new IllegalArgumentException("Given an unsupported memory size: " + req.getMegaBytes());
    }
    if (req.getNumberOfCores() <= 0) {
      throw new IllegalArgumentException(
          "Given an unsupported core number: " + req.getNumberOfCores());
    }
    if (req.getNumber() <= 0) {
      throw new IllegalArgumentException(
          "Given an unsupported number of evaluators: " + req.getNumber());
    }
    if (req.getNodeNames() == null) {
      throw new IllegalArgumentException("Node names cannot be null");
    }
    if (req.getRackNames() == null) {
      throw new IllegalArgumentException("Rack names cannot be null");
    }
    if (req.getRuntimeName() == null) {
      throw new IllegalArgumentException("Runtime name cannot be null");
    }
    // for backwards compatibility, we will always set the relax locality flag
    // to true unless the user configured racks, in which case we will check for
    // the ANY modifier (*), if not there, then we won't relax the locality
    boolean relaxLocality = true;
    if (!req.getRackNames().isEmpty()) {
      for (final String rackName : req.getRackNames()) {
        if (Constants.ANY_RACK.equals(rackName)) {
          relaxLocality = true;
          break;
        }
        relaxLocality = false;
      }
    }
    // if the user specified any node, then we assume they do not want to relax locality
    if (!req.getNodeNames().isEmpty()) {
      relaxLocality = false;
    }

    try (LoggingScope ls = loggingScopeFactory.evaluatorSubmit(req.getNumber())) {
      final ResourceRequestEvent request =
          ResourceRequestEventImpl.newBuilder()
              .setResourceCount(req.getNumber())
              .setVirtualCores(req.getNumberOfCores())
              .setMemorySize(req.getMegaBytes())
              .addNodeNames(req.getNodeNames())
              .addRackNames(req.getRackNames())
              .setRelaxLocality(relaxLocality)
              .setRuntimeName(req.getRuntimeName())
              .build();
      this.resourceRequestHandler.onNext(request);
    }
  }
コード例 #2
0
  @Override
  public synchronized void submit(final EvaluatorRequest req) {
    LOG.log(
        Level.FINEST,
        "Got an EvaluatorRequest: number: {0}, memory = {1}, cores = {2}.",
        new Object[] {req.getNumber(), req.getMegaBytes(), req.getNumberOfCores()});

    if (req.getMegaBytes() <= 0) {
      throw new IllegalArgumentException("Given an unsupported memory size: " + req.getMegaBytes());
    }
    if (req.getNumberOfCores() <= 0) {
      throw new IllegalArgumentException(
          "Given an unsupported core number: " + req.getNumberOfCores());
    }
    if (req.getNumber() <= 0) {
      throw new IllegalArgumentException(
          "Given an unsupported number of evaluators: " + req.getNumber());
    }
    if (req.getNodeNames() == null) {
      throw new IllegalArgumentException("Node names cannot be null");
    }
    if (req.getRackNames() == null) {
      throw new IllegalArgumentException("Rack names cannot be null");
    }

    try (LoggingScope ls = loggingScopeFactory.evaluatorSubmit(req.getNumber())) {
      final ResourceRequestEvent request =
          ResourceRequestEventImpl.newBuilder()
              .setResourceCount(req.getNumber())
              .setVirtualCores(req.getNumberOfCores())
              .setMemorySize(req.getMegaBytes())
              .addNodeNames(req.getNodeNames())
              .addRackNames(req.getRackNames())
              .build();
      this.resourceRequestHandler.onNext(request);
    }
  }