TestingPayloadStorage.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 <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<size_t, 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, int 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::function<bool(const std::string&)>& writer);
    };
  } // Conditions namespace
} // Belle2 namespace