TestingPayloadStorage.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
#pragma once
 
#include <framework/database/PayloadMetadata.h>
 
#include <functional>
#include <unordered_map>
 
namespace Belle2 {
  class EventMetaData;
 
  namespace Conditions {
 
    class TestingPayloadStorage {
    public:
      explicit TestingPayloadStorage(const std::string& filename);
      bool get(const EventMetaData& event, PayloadMetadata& info);
      bool storeData(const std::string& name, TObject* object, const IntervalOfValidity& iov);
      bool storePayload(const std::string& name, const std::string& fileName, const IntervalOfValidity& iov);
      void reset() { m_initialized = false; }
    private:
      void read();
      std::unordered_map<std::string, std::vector<std::tuple<std::string, IntervalOfValidity>>> m_payloads;
      std::string m_filename;
      std::string m_absoluteFilename;
      std::string m_payloadDir;
      bool m_initialized{false};
      static std::string payloadFilename(const std::string& path, const std::string& name, const std::string& revision);
      bool writePayload(const std::string& fileName, const std::string& name, const TObject* object);
      bool store(const std::string& name, const IntervalOfValidity& iov, const std::string& source,
                 const std::function<bool(const std::string&)>& writer);
    };
  } // Conditions namespace
} // Belle2 namespace