Method scanPKColumn

src/storage/table/node_table.cpp:904–946

}

bool NodeTable::scanPKColumn(const Transaction* transaction, const Value& keyToLookup,
    std::vector<ColumnPredicateSet> columnPredicateSets, offset_t& result) const {
    auto dataChunk = constructDataChunkForColumns({pkColumnID});
    std::vector<ValueVector*> outVectors = {&dataChunk.getValueVectorMutable(0)};
    auto scanState =
        std::make_unique<NodeTableScanState>(nullptr, std::move(outVectors), dataChunk.state);
    scanState->source = TableScanSource::COMMITTED;
    scanState->setToTable(transaction, const_cast<NodeTable*>(this), {pkColumnID},
        std::move(columnPredicateSets));
    const auto numNodeGroups = nodeGroups->getNumNodeGroupsNoLock();
    for (node_group_idx_t nodeGroupIdx = 0; nodeGroupIdx < numNodeGroups; ++nodeGroupIdx) {
        auto* nodeGroup = nodeGroups->getNodeGroupNoLock(nodeGroupIdx);
        if (nodeGroup->getNumChunkedGroups() == 0) {
            continue;
        }
        scanState->nodeGroup = nodeGroup;
        scanState->nodeGroupIdx = nodeGroupIdx;
        nodeGroup->initializeScanState(transaction, *scanState);
        while (true) {
            const auto scanResult = nodeGroup->scan(transaction, *scanState);
            if (scanResult == NODE_GROUP_SCAN_EMPTY_RESULT) {
                break;
            }
            auto* scannedVector = scanState->outputVectors[0];
            for (idx_t i = 0; i < scannedVector->state->getSelSize(); ++i) {
                const auto pos = scannedVector->state->getSelVector()[i];
                if (scannedVector->isNull(pos)) {
                    continue;
                }
                if (!(*scannedVector->getAsValue(pos) == keyToLookup)) {
                    continue;
                }
                const auto offset = StorageUtils::getStartOffsetOfNodeGroup(nodeGroupIdx) +
                                    scanResult.startRow + pos;
                if (isVisibleNoLock(transaction, offset)) {
                    result = offset;
                    return true;
                }
            }
        }
    }
    return false;
}

void NodeTable::scanIndexColumns(main::ClientContext* context, IndexScanHelper& scanHelper,
    const NodeGroupCollection& nodeGroups_, std::optional<uint64_t> queryID) const {