csv_reader_config.h

Go to the documentation of this file.

#pragma once
 
#include "constants.h"
#include "copy_constructors.h"
#include "value.h"
 
namespace kuzu {
namespace common {
 
struct CSVOption {
 // TODO(Xiyang): Add newline character option and delimiter can be a string.
 char escapeChar;
 char delimiter;
 char quoteChar;
 bool hasHeader;
 uint64_t skipNum;
 
 CSVOption()
 : escapeChar{CopyConstants::DEFAULT_CSV_ESCAPE_CHAR},
 delimiter{CopyConstants::DEFAULT_CSV_DELIMITER},
 quoteChar{CopyConstants::DEFAULT_CSV_QUOTE_CHAR},
 hasHeader{CopyConstants::DEFAULT_CSV_HAS_HEADER},
 skipNum{CopyConstants::DEFAULT_CSV_SKIP_NUM} {}
 EXPLICIT_COPY_DEFAULT_MOVE(CSVOption);
 
 // TODO: COPY FROM and COPY TO should support transform special options, like '\'.
 std::string toCypher() const {
 std::string header = hasHeader ? "true" : "false";
 return stringFormat("(escape ='\\{}', delim ='{}', quote='\\{}', header={})", escapeChar,
 delimiter, quoteChar, header);
 }
 
private:
 CSVOption(const CSVOption& other)
 : escapeChar{other.escapeChar}, delimiter{other.delimiter}, quoteChar{other.quoteChar},
 hasHeader{other.hasHeader}, skipNum{other.skipNum} {}
};
 
struct CSVReaderConfig {
 CSVOption option;
 bool parallel;
 
 CSVReaderConfig() : option{}, parallel{CopyConstants::DEFAULT_CSV_PARALLEL} {}
 EXPLICIT_COPY_DEFAULT_MOVE(CSVReaderConfig);
 
 static CSVReaderConfig construct(const std::unordered_map<std::string, common::Value>& options);
 
private:
 CSVReaderConfig(const CSVReaderConfig& other)
 : option{other.option.copy()}, parallel{other.parallel} {}
};
 
} // namespace common
} // namespace kuzu