Method lookupPK

src/storage/table/arrow_node_table.cpp:213–263

}

bool ArrowNodeTable::lookupPK([[maybe_unused]] const transaction::Transaction* transaction,
    common::ValueVector* keyVector, uint64_t vectorPos, common::offset_t& result) const {
    if (keyVector->isNull(vectorPos)) {
        return false;
    }

    int64_t pkArrowColumnIdx = -1;
    for (common::idx_t propIdx = 0; propIdx < nodeTableCatalogEntry->getNumProperties();
         ++propIdx) {
        if (nodeTableCatalogEntry->getColumnID(propIdx) == getPKColumnID()) {
            pkArrowColumnIdx = static_cast<int64_t>(propIdx);
            break;
        }
    }
    if (pkArrowColumnIdx < 0 || !schema.children || pkArrowColumnIdx >= schema.n_children ||
        !schema.children[pkArrowColumnIdx]) {
        return false;
    }

    auto keyToLookup = keyVector->getAsValue(vectorPos);
    auto pkType = getColumn(getPKColumnID()).getDataType().copy();
    auto singleValueState = common::DataChunkState::getSingleValueDataChunkState();
    auto arrowPKVector =
        std::make_unique<common::ValueVector>(std::move(pkType), memoryManager, singleValueState);
    arrowPKVector->state->setToFlat();

    for (size_t batchIdx = 0; batchIdx < arrays.size(); ++batchIdx) {
        const auto& batch = arrays[batchIdx];
        const auto batchLength = getArrowBatchLength(batch);
        if (batchLength == 0 || !batch.children || pkArrowColumnIdx >= batch.n_children ||
            !batch.children[pkArrowColumnIdx]) {
            continue;
        }
        auto* pkChildArray = batch.children[pkArrowColumnIdx];
        auto* pkChildSchema = schema.children[pkArrowColumnIdx];
        common::ArrowNullMaskTree nullMask(pkChildSchema, pkChildArray, pkChildArray->offset,
            pkChildArray->length);
        for (uint64_t rowIdx = 0; rowIdx < batchLength; ++rowIdx) {
            common::ArrowConverter::fromArrowArray(pkChildSchema, pkChildArray, *arrowPKVector,
                &nullMask, pkChildArray->offset + rowIdx, 0, 1);
            if (arrowPKVector->isNull(0)) {
                continue;
            }
            if (*arrowPKVector->getAsValue(0) == *keyToLookup) {
                result = batchStartOffsets[batchIdx] + rowIdx;
                return true;
            }
        }
    }
    return false;
}

bool ArrowNodeTable::isVisible([[maybe_unused]] const transaction::Transaction* transaction,
    common::offset_t offset) const {