FragmentsToFilesFilter.h

Go to the documentation of this file.

#ifndef _MADARA_FILTERS_FRAGMENTS_TO_FILES_FILTER_H_
#define _MADARA_FILTERS_FRAGMENTS_TO_FILES_FILTER_H_
 
#ifndef BOOST_NO_CXX11_SCOPED_ENUMS
#define BOOST_NO_CXX11_SCOPED_ENUMS
#endif
 
#include <string>
#include <vector>
#include "madara/knowledge/KnowledgeUpdateSettings.h"
#include "madara/knowledge/KnowledgeRecord.h"
#include "madara/utility/Utility.h"
#include "madara/utility/ScopedArray.h"
#include "madara/knowledge/containers/Vector.h"
#include "madara/logger/GlobalLogger.h"
#include "madara/filters/AggregateFilter.h"
#include "madara/MadaraExport.h"
 
namespace madara
{
namespace filters
{
class FragmentsToFilesFilter : public AggregateFilter
{
public:
  FragmentsToFilesFilter() : clear_fragments_(true) {}
 
  virtual ~FragmentsToFilesFilter() = default;
 
  MADARA_EXPORT virtual void filter(knowledge::KnowledgeMap& records,
      const transport::TransportContext&, knowledge::Variables&);
 
  inline void set_dir_mapping(
      const std::string& variable_prefix, const std::string& directory)
  {
    madara_logger_ptr_log(madara::logger::global_logger.get(),
        logger::LOG_MAJOR,
        "FragmentsToFilesFilter::set_dir_mapping: "
        "attempting to map %s -> %s\n",
        variable_prefix.c_str(), directory.c_str())
 
        if (variable_prefix != "" && directory != "")
    {
      map_[variable_prefix] = directory;
    }
  }
 
  inline std::string get_dir_mapping(const std::string& variable_prefix)
  {
    auto found = map_.find(variable_prefix);
 
    if (found != map_.end())
    {
      return found->second;
    }
    else
    {
      return "";
    }
  }
 
  inline void enable_clear_fragments(void)
  {
    clear_fragments_ = true;
  }
 
  inline void disable_clear_fragments(void)
  {
    clear_fragments_ = false;
  }
 
  bool clear_fragments_;
 
  std::map<std::string, std::string> map_;
};
}
}
 
#endif  // _MADARA_FILTERS_FRAGMENTS_TO_FILES_FILTER_H_