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);
 
  inline Any at(size_t row_id, size_t col_id) {
    return column_ptrs_[col_id]->get(row_id);
  }
 
  inline Any at(size_t row_id, size_t col_id) const {
    return column_ptrs_[col_id]->get(row_id);
  }
 
  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_