CSVWriter.h

Go to the documentation of this file.

#pragma once
 
#include <array>
#include <filesystem>
#include <fstream>
#include <string>
#include <utility>
#include <variant>
#include <vector>
 
#include "io/logger/Logger.h"
 
template <typename T>
void write_csv_element(std::ofstream& file, const T& value);
 
template <>
void write_csv_element<std::string>(std::ofstream& file, const std::string& value);
class CSVWriter {
   public:
    using serializable_types = std::variant<size_t, int, double, std::string>;
 
    CSVWriter(std::filesystem::path file_path, const std::vector<std::string>& headers, bool append, std::string separator = ";");
 
    explicit CSVWriter(std::filesystem::path file_path, bool append, std::string separator = ";");
 
    // Delete copy constructor and assignment operator
    CSVWriter(const CSVWriter&) = delete;
    CSVWriter& operator=(CSVWriter&) = delete;
 
    // Move constructor and assignment operator
    CSVWriter(CSVWriter&&) noexcept;
    CSVWriter& operator=(CSVWriter&&) noexcept;
 
    ~CSVWriter();
 
    void initialize(const std::vector<std::string>& headers);
 
    void writeRow(const std::vector<serializable_types>& row);
 
   private:
    std::ofstream file;
 
    std::filesystem::path file_path;
 
    bool append;
 
    std::vector<std::string> headers;
 
    std::string separator;
};