update_transaction.h

Go to the documentation of this file.

 
#ifndef GRAPHSCOPE_DATABASE_UPDATE_TRANSACTION_H_
#define GRAPHSCOPE_DATABASE_UPDATE_TRANSACTION_H_
 
#include <limits>
#include <utility>
 
#include "flat_hash_map/flat_hash_map.hpp"
#include "flex/engines/graph_db/database/transaction_utils.h"
#include "flex/storages/rt_mutable_graph/csr/mutable_csr.h"
#include "flex/storages/rt_mutable_graph/types.h"
#include "flex/utils/allocators.h"
#include "flex/utils/id_indexer.h"
#include "flex/utils/property/table.h"
#include "flex/utils/property/types.h"
#include "grape/serialization/in_archive.h"
 
namespace gs {
 
class MutablePropertyFragment;
class WalWriter;
class VersionManager;
 
class UpdateTransaction {
 public:
  UpdateTransaction(MutablePropertyFragment& graph, Allocator& alloc,
                    const std::string& work_dir, WalWriter& logger,
                    VersionManager& vm, timestamp_t timestamp);
 
  ~UpdateTransaction();
 
  timestamp_t timestamp() const;
 
  void Commit();
 
  void Abort();
 
  bool AddVertex(label_t label, const Any& oid, const std::vector<Any>& props);
 
  bool AddEdge(label_t src_label, const Any& src, label_t dst_label,
               const Any& dst, label_t edge_label, const Any& value);
 
  class vertex_iterator {
   public:
    vertex_iterator(label_t label, vid_t cur, vid_t& num,
                    UpdateTransaction* txn);
    ~vertex_iterator();
    bool IsValid() const;
    void Next();
    void Goto(vid_t target);
 
    Any GetId() const;
 
    vid_t GetIndex() const;
 
    Any GetField(int col_id) const;
 
    bool SetField(int col_id, const Any& value);
 
   private:
    label_t label_;
    vid_t cur_;
 
    vid_t& num_;
    UpdateTransaction* txn_;
  };
 
  class edge_iterator {
   public:
    edge_iterator(bool dir, label_t label, vid_t v, label_t neighbor_label,
                  label_t edge_label, const vid_t* aeb, const vid_t* aee,
                  std::shared_ptr<CsrConstEdgeIterBase> init_iter,
                  UpdateTransaction* txn);
    ~edge_iterator();
 
    Any GetData() const;
 
    void SetData(const Any& value);
 
    bool IsValid() const;
 
    void Next();
 
    void Forward(size_t offset);
 
    vid_t GetNeighbor() const;
 
    label_t GetNeighborLabel() const;
 
    label_t GetEdgeLabel() const;
 
   private:
    bool dir_;
 
    label_t label_;
    vid_t v_;
 
    label_t neighbor_label_;
    label_t edge_label_;
 
    const vid_t* added_edges_cur_;
    const vid_t* added_edges_end_;
 
    std::shared_ptr<CsrConstEdgeIterBase> init_iter_;
    UpdateTransaction* txn_;
    size_t offset_;
  };
 
  vertex_iterator GetVertexIterator(label_t label);
 
  edge_iterator GetOutEdgeIterator(label_t label, vid_t u,
                                   label_t neighbor_label, label_t edge_label);
 
  edge_iterator GetInEdgeIterator(label_t label, vid_t u,
                                  label_t neighbor_label, label_t edge_label);
 
  Any GetVertexField(label_t label, vid_t lid, int col_id) const;
 
  bool SetVertexField(label_t label, vid_t lid, int col_id, const Any& value);
 
  void SetEdgeData(bool dir, label_t label, vid_t v, label_t neighbor_label,
                   vid_t nbr, label_t edge_label, const Any& value);
 
  bool GetUpdatedEdgeData(bool dir, label_t label, vid_t v,
                          label_t neighbor_label, vid_t nbr, label_t edge_label,
                          Any& ret) const;
 
  static void IngestWal(MutablePropertyFragment& graph,
                        const std::string& work_dir, uint32_t timestamp,
                        char* data, size_t length, Allocator& alloc);
 
 private:
  friend class GraphDBSession;
  void batch_commit(UpdateBatch& batch);
 
  void set_edge_data_with_offset(bool dir, label_t label, vid_t v,
                                 label_t neighbor_label, vid_t nbr,
                                 label_t edge_label, const Any& value,
                                 size_t offset);
 
  size_t get_in_csr_index(label_t src_label, label_t dst_label,
                          label_t edge_label) const;
 
  size_t get_out_csr_index(label_t src_label, label_t dst_label,
                           label_t edge_label) const;
 
  bool oid_to_lid(label_t label, const Any& oid, vid_t& lid) const;
 
  Any lid_to_oid(label_t label, vid_t lid) const;
 
  void release();
 
  void applyVerticesUpdates();
 
  void applyEdgesUpdates();
 
  MutablePropertyFragment& graph_;
  Allocator& alloc_;
  WalWriter& logger_;
  VersionManager& vm_;
  timestamp_t timestamp_;
 
  grape::InArchive arc_;
  int op_num_;
 
  size_t vertex_label_num_;
  size_t edge_label_num_;
 
  std::vector<std::shared_ptr<IdIndexerBase<vid_t>>> added_vertices_;
  std::vector<vid_t> added_vertices_base_;
  std::vector<vid_t> vertex_nums_;
  std::vector<ska::flat_hash_map<vid_t, vid_t>> vertex_offsets_;
  std::vector<Table> extra_vertex_properties_;
 
  std::vector<ska::flat_hash_map<vid_t, std::vector<vid_t>>> added_edges_;
  std::vector<ska::flat_hash_map<
      vid_t, ska::flat_hash_map<vid_t, std::pair<Any, size_t>>>>
      updated_edge_data_;
 
  std::vector<std::string> sv_vec_;
};
 
}  // namespace gs
 
#endif  // GRAPHSCOPE_DATABASE_UPDATE_TRANSACTION_H_