PayloadProvider.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2019 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Martin Ritter                                            *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
 
#include <framework/database/PayloadMetadata.h>
#include <framework/database/Downloader.h>
#include <framework/utilities/FileSystem.h>
 
#include <vector>
#include <string>
#include <unordered_map>
 
namespace Belle2::Conditions {
  class PayloadProvider {
  public:
 
    struct PayloadLocation {
      std::string base;
      bool isRemote;
    };
 
    enum class EDirectoryLayout {
      c_flat,
      c_hashed,
    };
 
    explicit PayloadProvider(const std::vector<std::string>& locations, const std::string& cachedir = "", int timeout = 60);
 
    bool find(PayloadMetadata& meta);
  private:
    bool getLocalFile(const PayloadLocation& loc, PayloadMetadata& meta) const;
    bool getRemoteFile(const PayloadLocation& loc, PayloadMetadata& meta);
    bool getTemporaryFile(const std::string& url, PayloadMetadata& meta, bool silentOnMissing);
    std::string getFilename(EDirectoryLayout structure, const PayloadMetadata& payload) const;
    std::vector<PayloadLocation> m_locations;
    PayloadLocation m_cacheDir;
    Downloader& m_downloader{Downloader::getDefaultInstance()};
    std::unordered_map<std::string, std::unique_ptr<FileSystem::TemporaryFile>> m_temporaryFiles;
    int m_timeout;
  };
} // Belle2::Conditions namespace