Files
usql/table.h
2021-11-22 15:05:25 +01:00

71 lines
2.1 KiB
C++

#pragma once
#include "index.h"
#include "parser.h"
#include "row.h"
#include <vector>
#include <iterator> // For std::forward_iterator_tag
#include <cstddef> // For std::ptrdiff_t
namespace usql {
//using IndexValue=std::variant<ColNullValue, ColIntegerValue, ColStringValue>;
using IndexValue=std::variant<long, std::string>;
struct Table {
Table(const Table &other);
Table(const std::string& name, const std::vector<ColDefNode>& columns);
ColDefNode get_column_def(const std::string &col_name);
ColDefNode get_column_def(int col_index);
[[nodiscard]] int columns_count() const { return (int) m_col_defs.size(); };
[[nodiscard]] size_t rows_count() const { return m_rows.size(); };
Row& create_empty_row();
void commit_row(const Row &row);
void commit_copy_of_row(const Row &row);
Row& get_row(int rowid) { return m_rows[rowid]; };
static void validate_column(const ColDefNode *col_def, ValueNode *col_val);
static void validate_column(const ColDefNode *col_def, ColValue &col_val);
void validate_row(const Row &row);
std::string csv_string();
int load_csv_string(const std::string &content);
int load_csv_file(const std::string &filename);
void print();
std::string m_name;
std::vector<ColDefNode> m_col_defs;
std::vector<Row> m_rows;
std::vector<Index<IndexValue>> m_indexes;
static long string_to_long(const std::string &s);
static double string_to_double(const std::string &s);
void create_row_from_vector(const std::vector<ColDefNode> &colDefs, const std::vector<std::string> &csv_line);
void create_index(const Index<IndexValue>& index);
void drop_index(const std::string &column);
void index_row(Index<IndexValue> &index, const Row &row, const size_t rowid);
void index_row(const Row &row, const size_t rowid);
void index_rows(const std::string &index_name);
Index<IndexValue> * get_index(const std::string &index_name);
Index<IndexValue> * get_index_for_column(const std::string &col_name);
std::vector<int> index_search(const std::string &col_name, IndexValue key);
typedef std::vector<Row>::iterator iterator;
iterator fs_begin() { return m_rows.begin(); }
iterator fs_end() { return m_rows.end(); }
};
} // namespace