lbug 0.15.4

An in-process property graph database management system built for query speed and scalability
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55

#pragma once

#include "function/gds/bfs_graph.h"
#include "gds_auxilary_state.h"

namespace lbug {
namespace function {

class PathAuxiliaryState : public GDSAuxiliaryState {
public:
    explicit PathAuxiliaryState(std::unique_ptr<BFSGraphManager> bfsGraphManager)
        : bfsGraphManager{std::move(bfsGraphManager)} {}

    BFSGraphManager* getBFSGraphManager() { return bfsGraphManager.get(); }

    void beginFrontierCompute(common::table_id_t, common::table_id_t toTableID) override {
        bfsGraphManager->getCurrentGraph()->pinTableID(toTableID);
    }

    void switchToDense(processor::ExecutionContext* context, graph::Graph* graph) override {
        bfsGraphManager->switchToDense(context, graph);
    }

private:
    std::unique_ptr<BFSGraphManager> bfsGraphManager;
};

class WSPPathsAuxiliaryState : public GDSAuxiliaryState {
public:
    explicit WSPPathsAuxiliaryState(std::unique_ptr<BFSGraphManager> bfsGraphManager)
        : bfsGraphManager{std::move(bfsGraphManager)} {}

    BFSGraphManager* getBFSGraphManager() { return bfsGraphManager.get(); }

    void initSource(common::nodeID_t sourceNodeID) override {
        sourceParent.setCost(0);
        bfsGraphManager->getCurrentGraph()->pinTableID(sourceNodeID.tableID);
        bfsGraphManager->getCurrentGraph()->setParentList(sourceNodeID.offset, &sourceParent);
    }

    void beginFrontierCompute(common::table_id_t, common::table_id_t toTableID) override {
        bfsGraphManager->getCurrentGraph()->pinTableID(toTableID);
    }

    void switchToDense(processor::ExecutionContext* context, graph::Graph* graph) override {
        bfsGraphManager->switchToDense(context, graph);
    }

private:
    std::unique_ptr<BFSGraphManager> bfsGraphManager;
    ParentList sourceParent;
};

} // namespace function
} // namespace lbug