#ifndef VISUALDL_BACKEND_LOGIC_IM_H
#define VISUALDL_BACKEND_LOGIC_IM_H

#include <string>

#include "visualdl/backend/storage/storage.h"

namespace visualdl {

/*
 * Maintain the Storage singleton in memory, pre-compute some the statical
 * information to help visualizaton.
 */
class InformationMaintainer final {
public:
  InformationMaintainer() {}

  static InformationMaintainer &Global() {
    static InformationMaintainer *x = new InformationMaintainer();
    return *x;
  }

  /*
   * Set the disk path to store the Storage object.
   */
  void SetPersistDest(const std::string &path);

  storage::Tablet *AddTablet(const std::string &tag, int num_samples);

  /*
   * @tag: tag of the target Tablet.
   * @type: type of target Tablet.
   * @data: serialized protobuf message.
   *
   * NOTE pass in the serialized protobuf message will trigger copying, but
   * simpler to support different Tablet data formats.
   */
  void AddRecord(const std::string &tag, storage::Tablet::Type type,
                 const std::string &data,
                 storage::DataType dtype = storage::DataType::kUnknown);

  /*
   * Save the Storage Protobuf to disk.
   */
  void PersistToDisk();

  Storage &storage() { return storage_; }

private:
  Storage storage_;
};

} // namespace visualdl

#endif // VISUALDL_BACKEND_LOGIC_IM_H