lbug 0.16.0

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
56
57
58
59
60
61

#pragma once

#include "function/gds/compute.h"
#include "function/gds/gds.h"

namespace lbug {
namespace function {

class GDSVertexCompute : public VertexCompute {
public:
    explicit GDSVertexCompute(common::NodeOffsetMaskMap* nodeMask) : nodeMask{nodeMask} {}

    bool beginOnTable(common::table_id_t tableID) override {
        if (nodeMask != nullptr) {
            nodeMask->pin(tableID);
        }
        beginOnTableInternal(tableID);
        return true;
    }

protected:
    bool skip(common::offset_t offset) {
        if (nodeMask != nullptr && nodeMask->hasPinnedMask()) {
            return !nodeMask->valid(offset);
        }
        return false;
    }

    virtual void beginOnTableInternal(common::table_id_t tableID) = 0;

protected:
    common::NodeOffsetMaskMap* nodeMask;
};

class GDSResultVertexCompute : public GDSVertexCompute {
public:
    GDSResultVertexCompute(storage::MemoryManager* mm, GDSFuncSharedState* sharedState)
        : GDSVertexCompute{sharedState->getGraphNodeMaskMap()}, sharedState{sharedState}, mm{mm} {
        localFT = sharedState->factorizedTablePool.claimLocalTable(mm);
    }
    ~GDSResultVertexCompute() override {
        sharedState->factorizedTablePool.returnLocalTable(localFT);
    }

protected:
    std::unique_ptr<common::ValueVector> createVector(const common::LogicalType& type) {
        auto vector = std::make_unique<common::ValueVector>(type.copy(), mm);
        vector->state = common::DataChunkState::getSingleValueDataChunkState();
        vectors.push_back(vector.get());
        return vector;
    }

protected:
    GDSFuncSharedState* sharedState;
    storage::MemoryManager* mm;
    processor::FactorizedTable* localFT = nullptr;
    std::vector<common::ValueVector*> vectors;
};

} // namespace function
} // namespace lbug