FlushableOutput.h

Go to the documentation of this file.

/*
 * FlushableOutput.h
 *
 *  Created on: Feb 8, 2022
 *      Author: mschefer
 */
 
#ifndef _SEIMPLEMENTATION_OUTPUT_FLUSHABLEOUTPUT_H_
#define _SEIMPLEMENTATION_OUTPUT_FLUSHABLEOUTPUT_H_
 
#include "Table/Row.h"
 
#include "SEFramework/Output/Output.h"
 
namespace SourceXtractor {
 
class FlushableOutput : public Output {
 
public:
  using SourceToRowConverter = std::function<Euclid::Table::Row(const SourceInterface&)>;
 
  FlushableOutput(SourceToRowConverter source_to_row, size_t flush_size)
      : m_source_to_row(source_to_row), m_flush_size(flush_size), m_total_rows_written(0) {
  }
 
  virtual ~FlushableOutput() = default;
 
  size_t flush() override {
    if (!m_rows.empty()) {
      writeRows(m_rows);
    }
    m_total_rows_written += m_rows.size();
    m_rows.clear();
    return m_total_rows_written;
  }
 
  void outputSource(const SourceInterface& source) override {
    m_rows.emplace_back(m_source_to_row(source));
    if (m_flush_size > 0 && m_rows.size() % m_flush_size == 0) {
      flush();
    }
  }
 
protected:
  virtual void writeRows(const std::vector<Euclid::Table::Row>& rows) = 0;
 
private:
  SourceToRowConverter m_source_to_row;
  size_t m_flush_size;
 
  std::vector<Euclid::Table::Row> m_rows {};
  size_t m_total_rows_written;
};
 
}
 
#endif /* SEIMPLEMENTATION_SEIMPLEMENTATION_OUTPUT_FLUSHABLEOUTPUT_H_ */