mutable_property_fragment.h

Go to the documentation of this file.

 
#ifndef GRAPHSCOPE_FRAGMENT_MUTABLE_PROPERTY_FRAGMENT_H_
#define GRAPHSCOPE_FRAGMENT_MUTABLE_PROPERTY_FRAGMENT_H_
 
#include <thread>
#include <tuple>
#include <vector>
 
#include "flex/storages/rt_mutable_graph/schema.h"
 
#include "flex/storages/rt_mutable_graph/csr/mutable_csr.h"
#include "flex/storages/rt_mutable_graph/dual_csr.h"
#include "flex/storages/rt_mutable_graph/types.h"
#include "flex/utils/arrow_utils.h"
#include "flex/utils/indexers.h"
#include "flex/utils/property/table.h"
#include "flex/utils/yaml_utils.h"
#include "grape/io/local_io_adaptor.h"
#include "grape/serialization/out_archive.h"
 
namespace gs {
 
class MutablePropertyFragment {
 public:
 MutablePropertyFragment();
 
 ~MutablePropertyFragment();
 
 void IngestEdge(label_t src_label, vid_t src_lid, label_t dst_label,
 vid_t dst_lid, label_t edge_label, timestamp_t ts,
 grape::OutArchive& arc, Allocator& alloc);
 
 void UpdateEdge(label_t src_label, vid_t src_lid, label_t dst_label,
 vid_t dst_lid, label_t edge_label, timestamp_t ts,
 const Any& arc, Allocator& alloc);
 
 void Open(const std::string& work_dir, int memory_level);
 
 void Compact(uint32_t version);
 
 void Warmup(int thread_num);
 
 void Dump(const std::string& work_dir, uint32_t version);
 
 void DumpSchema(const std::string& filename);
 
 const Schema& schema() const;
 
 Schema& mutable_schema();
 
 void Clear();
 
 Table& get_vertex_table(label_t vertex_label);
 
 const Table& get_vertex_table(label_t vertex_label) const;
 
 vid_t vertex_num(label_t vertex_label) const;
 
 size_t edge_num(label_t src_label, label_t edge_label,
 label_t dst_label) const;
 
 bool get_lid(label_t label, const Any& oid, vid_t& lid) const;
 
 Any get_oid(label_t label, vid_t lid) const;
 
 vid_t add_vertex(label_t label, const Any& id);
 std::shared_ptr<CsrConstEdgeIterBase> get_outgoing_edges(
 label_t label, vid_t u, label_t neighbor_label, label_t edge_label) const;
 
 std::shared_ptr<CsrConstEdgeIterBase> get_incoming_edges(
 label_t label, vid_t u, label_t neighbor_label, label_t edge_label) const;
 
 std::shared_ptr<CsrEdgeIterBase> get_outgoing_edges_mut(
 label_t label, vid_t u, label_t neighbor_label, label_t edge_label);
 
 std::shared_ptr<CsrEdgeIterBase> get_incoming_edges_mut(
 label_t label, vid_t u, label_t neighbor_label, label_t edge_label);
 
 CsrConstEdgeIterBase* get_outgoing_edges_raw(label_t label, vid_t u,
 label_t neighbor_label,
 label_t edge_label) const;
 
 CsrConstEdgeIterBase* get_incoming_edges_raw(label_t label, vid_t u,
 label_t neighbor_label,
 label_t edge_label) const;
 
 CsrBase* get_oe_csr(label_t label, label_t neighbor_label,
 label_t edge_label);
 
 const CsrBase* get_oe_csr(label_t label, label_t neighbor_label,
 label_t edge_label) const;
 
 CsrBase* get_ie_csr(label_t label, label_t neighbor_label,
 label_t edge_label);
 
 const CsrBase* get_ie_csr(label_t label, label_t neighbor_label,
 label_t edge_label) const;
 
 void loadSchema(const std::string& filename);
 
 Schema schema_;
 std::vector<IndexerType> lf_indexers_;
 std::vector<CsrBase*> ie_, oe_;
 std::vector<DualCsrBase*> dual_csr_list_;
 std::vector<Table> vertex_data_;
 
 size_t vertex_label_num_, edge_label_num_;
};
 
} // namespace gs
 
#endif // GRAPHSCOPE_FRAGMENT_MUTABLE_PROPERTY_FRAGMENT_H_