Method scanFlat

src/storage/table/arrow_rel_table.cpp:224–355

}

bool ArrowRelTable::scanFlat(transaction::Transaction* transaction, TableScanState& scanState) {
    auto& relScanState = scanState.cast<RelTableScanState>();
    if (relScanState.arrowScanCompleted || !relScanState.arrowSrcKeyVector ||
        !relScanState.arrowDstKeyVector) {
        return false;
    }

    scanState.resetOutVectors();
    auto outputCount = 0u;
    constexpr uint64_t maxRowsPerCall = DEFAULT_VECTOR_CAPACITY;
    auto activeBoundSelPos = INVALID_SEL;
    auto activeBoundOffset = INVALID_OFFSET;
    auto hasActiveBound = false;
    const auto outputToArrowColumnIdx = getOutputToArrowColumnIdx(scanState.columnIDs);

    while (outputCount < maxRowsPerCall && relScanState.arrowCurrentBatchIdx < arrays.size()) {
        const auto& batch = arrays[relScanState.arrowCurrentBatchIdx];
        auto batchLength = getArrowBatchLength(batch);
        if (relScanState.arrowCurrentBatchOffset >= batchLength) {
            relScanState.arrowCurrentBatchIdx++;
            relScanState.arrowCurrentBatchOffset = 0;
            continue;
        }

        auto srcOffsetInBatch = relScanState.arrowCurrentBatchOffset;
        auto numChildren = batch.n_children < 0 ? 0u : static_cast<uint64_t>(batch.n_children);
        if (numChildren == 0 || !batch.children || !schema.children ||
            static_cast<uint64_t>(fromColumnIdx) >= numChildren ||
            static_cast<uint64_t>(toColumnIdx) >= numChildren || !batch.children[fromColumnIdx] ||
            !batch.children[toColumnIdx] || !schema.children[fromColumnIdx] ||
            !schema.children[toColumnIdx]) {
            relScanState.arrowCurrentBatchOffset++;
            continue;
        }

        auto* srcChildArray = batch.children[fromColumnIdx];
        auto* srcChildSchema = schema.children[fromColumnIdx];
        auto* dstChildArray = batch.children[toColumnIdx];
        auto* dstChildSchema = schema.children[toColumnIdx];
        auto srcOffsetToRead = srcChildArray->offset + srcOffsetInBatch;
        auto dstOffsetToRead = dstChildArray->offset + srcOffsetInBatch;
        readSingleArrowValue(srcChildSchema, srcChildArray, *relScanState.arrowSrcKeyVector,
            srcOffsetToRead, 0);
        if (relScanState.arrowSrcKeyVector->isNull(0)) {
            relScanState.arrowCurrentBatchOffset++;
            continue;
        }
        readSingleArrowValue(dstChildSchema, dstChildArray, *relScanState.arrowDstKeyVector,
            dstOffsetToRead, 0);
        if (relScanState.arrowDstKeyVector->isNull(0)) {
            relScanState.arrowCurrentBatchOffset++;
            continue;
        }

        offset_t srcNodeOffset = INVALID_OFFSET;
        offset_t dstNodeOffset = INVALID_OFFSET;
        if (!fromNodeTable->lookupPK(transaction, relScanState.arrowSrcKeyVector.get(), 0,
                srcNodeOffset)) {