table.h

Go to the documentation of this file.

 
#ifndef GRAPHSCOPE_PROPERTY_TABLE_H_
#define GRAPHSCOPE_PROPERTY_TABLE_H_
 
#include <map>
#include <memory>
#include <string_view>
 
#include "flex/utils/id_indexer.h"
#include "flex/utils/property/column.h"
#include "grape/io/local_io_adaptor.h"
#include "grape/serialization/out_archive.h"
 
namespace gs {
 
class Table {
 public:
 Table();
 ~Table();
 
 void init(const std::string& name, const std::string& work_dir,
 const std::vector<std::string>& col_name,
 const std::vector<PropertyType>& types,
 const std::vector<StorageStrategy>& strategies_);
 
 void open(const std::string& name, const std::string& snapshot_dir,
 const std::string& work_dir,
 const std::vector<std::string>& col_name,
 const std::vector<PropertyType>& property_types,
 const std::vector<StorageStrategy>& strategies_);
 
 void open_in_memory(const std::string& name, const std::string& snapshot_dir,
 const std::vector<std::string>& col_name,
 const std::vector<PropertyType>& property_types,
 const std::vector<StorageStrategy>& strategies_);
 
 void open_with_hugepages(const std::string& name,
 const std::string& snapshot_dir,
 const std::vector<std::string>& col_name,
 const std::vector<PropertyType>& property_types,
 const std::vector<StorageStrategy>& strategies_,
 bool force);
 
 void touch(const std::string& name, const std::string& work_dir);
 
 void copy_to_tmp(const std::string& name, const std::string& snapshot_dir,
 const std::string& work_dir);
 
 void dump(const std::string& name, const std::string& snapshot_dir);
 
 void reset_header(const std::vector<std::string>& col_name);
 
 std::vector<std::string> column_names() const;
 
 std::string column_name(size_t index) const;
 
 int get_column_id_by_name(const std::string& name) const;
 
 std::vector<PropertyType> column_types() const;
 
 std::shared_ptr<ColumnBase> get_column(const std::string& name);
 
 const std::shared_ptr<ColumnBase> get_column(const std::string& name) const;
 
 std::vector<Any> get_row(size_t row_id) const;
 
 std::shared_ptr<ColumnBase> get_column_by_id(size_t index);
 
 const std::shared_ptr<ColumnBase> get_column_by_id(size_t index) const;
 
 size_t col_num() const;
 size_t row_num() const;
 std::vector<std::shared_ptr<ColumnBase>>& columns();
 std::vector<ColumnBase*>& column_ptrs();
 
 void insert(size_t index, const std::vector<Any>& values);
 
 // insert properties except for the primary key
 // col_ind_mapping: the mapping from the column index in
 // the raw file row to the column index in the schema
 void insert(size_t index, const std::vector<Any>& values,
 const std::vector<int32_t>& col_ind_mapping);
 
 void resize(size_t row_num);
 
 Any at(size_t row_id, size_t col_id);
 
 Any at(size_t row_id, size_t col_id) const;
 
 void ingest(uint32_t index, grape::OutArchive& arc);
 
 void close();
 
 private:
 void buildColumnPtrs();
 void initColumns(const std::vector<std::string>& col_name,
 const std::vector<PropertyType>& types,
 const std::vector<StorageStrategy>& strategies_);
 
 IdIndexer<std::string, int> col_id_indexer_;
 
 std::vector<std::shared_ptr<ColumnBase>> columns_;
 std::vector<ColumnBase*> column_ptrs_;
 
 bool touched_;
};
 
} // namespace gs
 
#endif // GRAPHSCOPE_PROPERTY_TABLE_H_