Пример #1
0
 private static Optional<String> getClientHost(SrmPrepareToPutRequest request) {
   if (request.getTransferParameters() != null
       && request.getTransferParameters().getArrayOfClientNetworks() != null) {
     String[] clientNetworks =
         request.getTransferParameters().getArrayOfClientNetworks().getStringArray();
     if (clientNetworks != null && clientNetworks.length > 0 && clientNetworks[0] != null) {
       return Optional.of(clientNetworks[0]);
     }
   }
   return Optional.absent();
 }
Пример #2
0
 private static TPutFileRequest[] getFileRequests(SrmPrepareToPutRequest request)
     throws SRMInvalidRequestException {
   TPutFileRequest[] fileRequests = null;
   if (request.getArrayOfFileRequests() != null) {
     fileRequests = request.getArrayOfFileRequests().getRequestArray();
   }
   if (fileRequests == null || fileRequests.length < 1) {
     throw new SRMInvalidRequestException("request contains no file requests.");
   }
   return fileRequests;
 }
Пример #3
0
 private static void checkFileStorageType(
     SrmPrepareToPutRequest request, TFileStorageType expectedStorageType)
     throws SRMNotSupportedException {
   TFileStorageType storageType = request.getDesiredFileStorageType();
   if (storageType != null && !storageType.equals(expectedStorageType)) {
     throw new SRMNotSupportedException(
         "DesiredFileStorageType " + storageType + " is not supported.");
   }
 }
Пример #4
0
 private static TOverwriteMode getOverwriteMode(SrmPrepareToPutRequest request)
     throws SRMNotSupportedException {
   TOverwriteMode overwriteMode = request.getOverwriteOption();
   if (overwriteMode != null && overwriteMode.equals(TOverwriteMode.WHEN_FILES_ARE_DIFFERENT)) {
     throw new SRMNotSupportedException(
         "Overwrite Mode WHEN_FILES_ARE_DIFFERENT is not supported.");
   }
   return overwriteMode;
 }
Пример #5
0
 private String[] getProtocols() throws SRMInvalidRequestException {
   String[] protocols = null;
   TTransferParameters transferParameters = request.getTransferParameters();
   if (transferParameters != null && transferParameters.getArrayOfTransferProtocols() != null) {
     protocols = transferParameters.getArrayOfTransferProtocols().getStringArray();
   }
   protocols = Tools.trimStringArray(protocols);
   if (protocols == null || protocols.length < 1) {
     throw new SRMInvalidRequestException("request contains no transfer protocols.");
   }
   return protocols;
 }
Пример #6
0
 private static long getLifetime(SrmPrepareToPutRequest request, long max)
     throws SRMInvalidRequestException {
   long lifetimeInSeconds = 0;
   if (request.getDesiredTotalRequestTime() != null) {
     long reqLifetime = (long) request.getDesiredTotalRequestTime().intValue();
     if (reqLifetime < 0) {
       /* [ SRM 2.2, 5.5.2 ]
        * q)    If input parameter desiredTotalRequestTime is 0 (zero), each file
        *       request must be tried at least once. Negative value must be invalid.
        */
       throw new SRMInvalidRequestException("Negative desiredTotalRequestTime is invalid.");
     }
     lifetimeInSeconds = reqLifetime;
   }
   if (lifetimeInSeconds <= 0) {
     // Revisit: Behaviour doesn't match the SRM spec
     return max;
   }
   long lifetime = TimeUnit.SECONDS.toMillis(lifetimeInSeconds);
   return lifetime > max ? max : lifetime;
 }
Пример #7
0
 private static String getExtraInfo(SrmPrepareToPutRequest request, String key) {
   ArrayOfTExtraInfo storageSystemInfo = request.getStorageSystemInfo();
   if (storageSystemInfo == null) {
     return null;
   }
   TExtraInfo[] extraInfoArray = storageSystemInfo.getExtraInfoArray();
   if (extraInfoArray == null || extraInfoArray.length <= 0) {
     return null;
   }
   for (TExtraInfo extraInfo : extraInfoArray) {
     if (extraInfo.getKey().equals(key)) {
       return extraInfo.getValue();
     }
   }
   return null;
 }
Пример #8
0
  private SrmPrepareToPutResponse srmPrepareToPut()
      throws IllegalStateTransition, InterruptedException, SRMNotSupportedException,
          SRMInvalidRequestException, SRMInternalErrorException {
    checkFileStorageType(request, TFileStorageType.PERMANENT);
    String[] protocols = getProtocols();
    String clientHost = getClientHost(request).or(this.clientHost);
    String spaceToken = request.getTargetSpaceToken();
    TRetentionPolicy retentionPolicy = null;
    TAccessLatency accessLatency = null;
    if (request.getTargetFileRetentionPolicyInfo() != null) {
      retentionPolicy = request.getTargetFileRetentionPolicyInfo().getRetentionPolicy();
      accessLatency = request.getTargetFileRetentionPolicyInfo().getAccessLatency();
    }
    TPutFileRequest[] fileRequests = getFileRequests(request);

    long lifetime = getLifetime(request, configuration.getPutLifetime());
    TOverwriteMode overwriteMode = getOverwriteMode(request);

    String[] supportedProtocols = storage.supportedPutProtocols();
    boolean isAnyProtocolSupported = any(asList(protocols), in(asList(supportedProtocols)));
    if (!isAnyProtocolSupported) {
      throw new SRMNotSupportedException(
          "Protocol(s) not supported: " + Arrays.toString(protocols));
    }

    URI[] surls = new URI[fileRequests.length];
    Long[] sizes = new Long[fileRequests.length];
    boolean[] wantPermanent = new boolean[fileRequests.length];
    for (int i = 0; i < fileRequests.length; ++i) {
      TPutFileRequest fileRequest = fileRequests[i];
      if (fileRequest == null) {
        throw new SRMInvalidRequestException("file request #" + (i + 1) + " is null.");
      }
      if (fileRequest.getTargetSURL() == null) {
        throw new SRMInvalidRequestException("surl of file request #" + (i + 1) + " is null.");
      }
      URI surl = URI.create(fileRequest.getTargetSURL().toString());
      UnsignedLong knownSize = fileRequest.getExpectedFileSize();
      if (knownSize != null) {
        sizes[i] = knownSize.longValue();
        if (sizes[i] < 0) {
          throw new SRMInvalidRequestException("Negative file size is not allowed.");
        }
      }
      wantPermanent[i] = true; // for now, extract type info from space token in the future
      /*                nextRequest.getFileStorageType()==
      TFileStorageType.PERMANENT;*/
      surls[i] = surl;
    }

    PutRequest r =
        new PutRequest(
            user,
            credential.getId(),
            surls,
            sizes,
            wantPermanent,
            protocols,
            lifetime,
            configuration.getGetRetryTimeout(),
            configuration.getGetMaxNumOfRetries(),
            clientHost,
            spaceToken,
            retentionPolicy,
            accessLatency,
            request.getUserRequestDescription());
    try (JDC ignored = r.applyJdc()) {
      String priority = getExtraInfo(request, "priority");
      if (priority != null) {
        try {
          r.setPriority(Integer.parseInt(priority));
        } catch (NumberFormatException e) {
          LOGGER.warn("Ignoring non-integer user priority: {}", priority);
        }
      }

      if (overwriteMode != null) {
        r.setOverwriteMode(overwriteMode);
      }

      srm.schedule(r);
      // RequestScheduler will take care of the rest
      // getRequestScheduler.add(r);
      // Return the request status
      return r.getSrmPrepareToPutResponse(configuration.getPutSwitchToAsynchronousModeDelay());
    }
  }