FileServiceProxy.h

Go to the documentation of this file.

#ifndef art_Framework_IO_Sources_detail_FileServiceProxy_h
#define art_Framework_IO_Sources_detail_FileServiceProxy_h

#include "art/Framework/Services/FileServiceInterfaces/CatalogInterface.h"
#include "art/Framework/Services/FileServiceInterfaces/FileDeliveryStatus.h"
#include "art/Framework/Services/FileServiceInterfaces/FileTransfer.h"
#include "art/Framework/Services/FileServiceInterfaces/FileTransferStatus.h"
#include "art/Framework/Services/Registry/ServiceHandle.h"

#include <string>
#include <vector>

namespace art::detail {
  class FileServiceProxy;
}

class art::detail::FileServiceProxy {
public:
  explicit FileServiceProxy(std::vector<std::string>&& fileNames,
                            size_t attempts = 5,
                            double waitBetweenAttempts = 5.0);
  ~FileServiceProxy();

  std::string next();
  void finish();

private:
  std::string obtainURI_();
  std::string obtainFileFromURI_();

  ServiceHandle<CatalogInterface> ci_{};
  ServiceHandle<FileTransfer> ft_{};
  struct FileEntity {
    explicit FileEntity(size_t const attempts) noexcept
      : attemptsRemaining{attempts}
    {}
    std::string uri{};
    std::string pfn{};
    FileDeliveryStatus uriStatus{FileDeliveryStatus::PENDING};
    FileTransferStatus ftStatus{FileTransferStatus::PENDING};
    size_t attemptsRemaining;
  } currentItem_;
  size_t const attemptsPerPhase_;
  double const waitBetweenAttempts_;
};

inline std::string
art::detail::FileServiceProxy::next()
{
  finish();
  return obtainURI_();
}

#endif /* art_Framework_IO_Sources_detail_FileServiceProxy_h */

// Local Variables:
// mode: c++
// End: