lbug 0.16.1

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

#include "function/sequence/sequence_functions.h"

#include "catalog/catalog.h"
#include "catalog/catalog_entry/sequence_catalog_entry.h"
#include "function/scalar_function.h"
#include "main/client_context.h"
#include "transaction/transaction.h"

using namespace lbug::common;

namespace lbug {
namespace function {

struct CurrVal {
    static void operation(string_t& input, ValueVector& result, void* dataPtr) {
        auto ctx = reinterpret_cast<FunctionBindData*>(dataPtr)->clientContext;
        auto catalog = catalog::Catalog::Get(*ctx);
        auto transaction = transaction::Transaction::Get(*ctx);
        auto sequenceName = input.getAsString();
        auto sequenceEntry =
            catalog->getSequenceEntry(transaction, sequenceName, ctx->useInternalCatalogEntry());
        result.setValue(0, sequenceEntry->currVal());
    }
};

struct NextVal {
    static void operation(string_t& input, ValueVector& result, void* dataPtr) {
        auto ctx = reinterpret_cast<FunctionBindData*>(dataPtr)->clientContext;
        auto cnt = reinterpret_cast<FunctionBindData*>(dataPtr)->count;
        auto catalog = catalog::Catalog::Get(*ctx);
        auto transaction = transaction::Transaction::Get(*ctx);
        auto sequenceName = input.getAsString();
        auto sequenceEntry =
            catalog->getSequenceEntry(transaction, sequenceName, ctx->useInternalCatalogEntry());
        sequenceEntry->nextKVal(transaction, cnt, result);
        result.state->getSelVectorUnsafe().setSelSize(cnt);
    }
};

function_set CurrValFunction::getFunctionSet() {
    function_set functionSet;
    functionSet.push_back(make_unique<ScalarFunction>(name,
        std::vector<LogicalTypeID>{LogicalTypeID::STRING}, LogicalTypeID::INT64,
        ScalarFunction::UnarySequenceExecFunction<string_t, ValueVector, CurrVal>));
    return functionSet;
}

function_set NextValFunction::getFunctionSet() {
    function_set functionSet;
    auto func = make_unique<ScalarFunction>(name, std::vector<LogicalTypeID>{LogicalTypeID::STRING},
        LogicalTypeID::INT64,
        ScalarFunction::UnarySequenceExecFunction<string_t, ValueVector, NextVal>);
    func->isReadOnly = false;
    functionSet.push_back(std::move(func));
    return functionSet;
}

} // namespace function
} // namespace lbug