lbug 0.20.0

An in-process property graph database management system built for query speed and scalability
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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87

#include "common/constants.h"
#include "main/attached_database.h"
#include "main/database_manager.h"
#include "planner/operator/scan/logical_count_rel_table.h"
#include "processor/operator/scan/count_rel_table.h"
#include "processor/plan_mapper.h"
#include "storage/partition_storage_registry.h"
#include "storage/storage_manager.h"

using namespace lbug::common;
using namespace lbug::planner;
using namespace lbug::storage;

namespace lbug {
namespace processor {

std::unique_ptr<PhysicalOperator> PlanMapper::mapCountRelTable(
    const LogicalOperator* logicalOperator) {
    auto& logicalCountRelTable = logicalOperator->constCast<LogicalCountRelTable>();
    auto outSchema = logicalCountRelTable.getSchema();

    // Resolve the storage manager that owns the rel tables: main by default, or the
    // attached LBUG database recorded on the count operator for attached rels. Foreign
    // attached databases (sqlite/duckdb/postgres) are never routed here: the binder
    // resolves their rel names in the main catalog, so dbName stays empty for them.
    auto dbName = logicalCountRelTable.getDbName();
    auto storageManager = StorageManager::Get(*clientContext);
    if (!dbName.empty()) {
        auto* attachedDB = main::DatabaseManager::Get(*clientContext)->getAttachedDatabase(dbName);
        if (attachedDB->getDBType() == common::ATTACHED_LBUG_DB_TYPE) {
            auto* attachedLbug = static_cast<main::AttachedLbugDatabase*>(attachedDB);
            storageManager = attachedLbug->getStorageManager();
        }
    }

    // Get the node tables for scanning bound nodes. For attached LBUG databases the
    // node table IDs belong to the attached catalog/storage manager, not the main one.
    std::vector<NodeTable*> nodeTables;
    auto* attachedLbugDB = static_cast<main::AttachedLbugDatabase*>(nullptr);
    if (!dbName.empty()) {
        auto* attachedDB = main::DatabaseManager::Get(*clientContext)->getAttachedDatabase(dbName);
        if (attachedDB->getDBType() == common::ATTACHED_LBUG_DB_TYPE) {
            attachedLbugDB = static_cast<main::AttachedLbugDatabase*>(attachedDB);
        }
    }
    for (auto tableID : logicalCountRelTable.getBoundNodeTableIDs()) {
        if (attachedLbugDB != nullptr) {
            auto* attachedSM = attachedLbugDB->getStorageManager();
            nodeTables.push_back(attachedSM->getTable(tableID)->ptrCast<NodeTable>());
        } else {
            nodeTables.push_back(
                storage::PartitionStorageRegistry::resolveNodeTableByID(clientContext, tableID)
                    ->ptrCast<NodeTable>());
        }
    }

    // Get the rel tables
    std::vector<RelTable*> relTables;
    for (auto tableID : logicalCountRelTable.getRelTableIDs()) {
        relTables.push_back(storageManager->getTable(tableID)->ptrCast<RelTable>());
    }

    // Determine rel data direction from extend direction
    auto extendDirection = logicalCountRelTable.getDirection();
    RelDataDirection relDirection;
    if (extendDirection == ExtendDirection::FWD) {
        relDirection = RelDataDirection::FWD;
    } else if (extendDirection == ExtendDirection::BWD) {
        relDirection = RelDataDirection::BWD;
    } else {
        // For BOTH, we'll scan FWD (shouldn't reach here as optimizer filters BOTH)
        relDirection = RelDataDirection::FWD;
    }

    // Get the output position for the count expression
    auto countOutputPos = getDataPos(*logicalCountRelTable.getCountExpr(), *outSchema);

    auto printInfo = std::make_unique<CountRelTablePrintInfo>(
        logicalCountRelTable.getRelGroupEntry()->getName());

    return std::make_unique<CountRelTable>(std::move(nodeTables), std::move(relTables),
        relDirection, countOutputPos, logicalCountRelTable.getReturnNullOnZero(), getOperatorID(),
        std::move(printInfo));
}

} // namespace processor
} // namespace lbug