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

#include "binder/expression/expression_util.h"
#include "binder/expression/rel_expression.h"
#include "binder/expression_binder.h"
#include "function/rewrite_function.h"
#include "function/schema/vector_node_rel_functions.h"
#include "function/struct/vector_struct_functions.h"

using namespace lbug::common;
using namespace lbug::binder;

namespace lbug {
namespace function {

static std::shared_ptr<Expression> startRewriteFunc(const RewriteFunctionBindInput& input) {
    DASSERT(input.arguments.size() == 1);
    auto param = input.arguments[0].get();
    if (ExpressionUtil::isRelPattern(*param)) {
        return param->constCast<RelExpression>().getSrcNode();
    }
    auto extractKey = input.expressionBinder->createLiteralExpression(InternalKeyword::SRC);
    return input.expressionBinder->bindScalarFunctionExpression({input.arguments[0], extractKey},
        StructExtractFunctions::name);
}

function_set StartNodeFunction::getFunctionSet() {
    function_set set;
    auto function = std::make_unique<RewriteFunction>(name,
        std::vector<LogicalTypeID>{LogicalTypeID::REL}, startRewriteFunc);
    set.push_back(std::move(function));
    return set;
}

static std::shared_ptr<Expression> endRewriteFunc(const RewriteFunctionBindInput& input) {
    DASSERT(input.arguments.size() == 1);
    auto param = input.arguments[0].get();
    if (ExpressionUtil::isRelPattern(*param)) {
        return param->constCast<RelExpression>().getDstNode();
    }
    auto extractKey = input.expressionBinder->createLiteralExpression(InternalKeyword::DST);
    return input.expressionBinder->bindScalarFunctionExpression({input.arguments[0], extractKey},
        StructExtractFunctions::name);
}

function_set EndNodeFunction::getFunctionSet() {
    function_set set;
    auto function = std::make_unique<RewriteFunction>(name,
        std::vector<LogicalTypeID>{LogicalTypeID::REL}, endRewriteFunc);
    set.push_back(std::move(function));
    return set;
}

} // namespace function
} // namespace lbug