index refactoring

index.h

@@ -2,6 +2,7 @@
 
 #include "exception.h"
 #include "parser.h"
+#include "row.h"
 
 #include <iostream>
 #include <utility>
@@ -12,7 +13,7 @@
 
 namespace usql {
 
-using IndexValue=std::variant<long, std::string>;
+using IndexValue = std::variant<long, std::string>;
 using rowid_t = size_t;   // int is now enough but size_t is correct
 
 static const int k_default_rowids_size = 16;
@@ -27,6 +28,55 @@ public:
 			throw Exception("creating index on unsupported type");
 	}
 
+
+	std::vector<rowid_t> search(ValueNode *key) {
+		return search(to_index_value(key));
+	}
+
+	void insert(ColValue *key, rowid_t rowid) {
+		return insert(to_index_value(key), rowid);
+	}
+
+	void remove(ColValue *key, rowid_t rowid) {
+		return remove(to_index_value(key), rowid);
+	}
+
+	void truncate() {
+		m_index.clear();
+	}
+
+	[[nodiscard]] const std::string &get_column_name() const {
+		return m_column_name;
+	}
+
+	[[nodiscard]] const std::string &get_index_name() const {
+		return m_index_name;
+	}
+
+	[[nodiscard]] ColumnType get_data_type() const {
+		return m_data_type;
+	}
+
+private:
+
+	IndexValue to_index_value(ValueNode *key) {
+		if (m_data_type == ColumnType::integer_type)
+			return key->getIntegerValue();
+		else if (m_data_type == ColumnType::varchar_type)
+			return key->getStringValue();
+		else
+			throw Exception("using index on unsupported type");
+	}
+
+	IndexValue to_index_value(ColValue *key) {
+		if (m_data_type == ColumnType::integer_type)
+			return key->getIntValue();
+		else if (m_data_type == ColumnType::varchar_type)
+			return key->getStringValue();
+		else
+			throw Exception("using index on unsupported type");
+	}
+
 	void insert(IndexValue key, rowid_t rowid) {
 		auto search = m_index.find(key);
 		if (search != m_index.end()) {
@@ -60,20 +110,8 @@ public:
 		}
 	}
 
-	std::vector<rowid_t> search(ValueNode * key) {
-		if (m_data_type == ColumnType::integer_type)
-			return search(key->getIntegerValue());
-		else if (m_data_type == ColumnType::varchar_type)
-			return search(key->getStringValue());
-		else
-			throw Exception("using index on unsupported type");
-	}
 
 
-	void truncate() {
-		m_index.clear();
-	}
-
 //	void dump() {
 //		std::for_each(m_index.begin(), m_index.end(),
 //			      [](std::pair<IndexValue, std::vector<rowid_t>> element){
@@ -83,25 +121,13 @@ public:
 //		});
 //	}
 
-	[[nodiscard]] const std::string &get_column_name() const {
-		return m_column_name;
-	}
-
-	[[nodiscard]] const std::string &get_index_name() const {
-		return m_index_name;
-	}
-
-	[[nodiscard]] ColumnType get_data_type() const {
-		return m_data_type;
-	}
-
 private:
-	bool			m_uniq;
-	std::string		m_index_name;
-	std::string		m_column_name;
-	ColumnType		m_data_type;
+	bool m_uniq;
+	std::string m_index_name;
+	std::string m_column_name;
+	ColumnType m_data_type;
 
 	std::map<IndexValue, std::vector<rowid_t> > m_index;
 };
 
-}	// namespace
+}    // namespace