Function createLargeBatchComplexGraph

test/api/arrow_rel_table_test.cpp:357–404

}

void createLargeBatchComplexGraph(main::Connection& connection) {
    constexpr int64_t NUM_NODES = 2050;
    constexpr int64_t NUM_EDGES = 2049;
    constexpr int64_t NODE_SPLIT = 1025;
    constexpr int64_t EDGE_SPLIT = 1025;

    {
        auto schema = makeComplexPersonSchema();
        std::vector<PersonRow> batch0;
        std::vector<PersonRow> batch1;
        for (int64_t i = 0; i < NUM_NODES; ++i) {
            auto row = PersonRow{i, "Person", 20 + (i % 40),
                DATE_2020_01_01 + static_cast<int32_t>(i % 5), {i, i + 1}};
            if (i < NODE_SPLIT) {
                batch0.push_back(row);
            } else {
                batch1.push_back(row);
            }
        }
        std::vector<ArrowArrayWrapper> arrays;
        arrays.push_back(makePersonBatch(batch0));
        arrays.push_back(makePersonBatch(batch1));
        auto result = ArrowTableSupport::createViewFromArrowTable(connection, "lb_person",
            std::move(schema), std::move(arrays));
        ASSERT_TRUE(result.queryResult->isSuccess()) << result.queryResult->getErrorMessage();
    }

    {
        auto schema = makeKnowsSchema();
        std::vector<KnowsRow> batch0;
        std::vector<KnowsRow> batch1;
        for (int64_t i = 0; i < NUM_EDGES; ++i) {
            KnowsRow row{i, i + 1, i, i < 3 ? "first3" : "rest",
                i < 5 ? DATE_2020_01_01 : DATE_2021_01_01, {i % 3}};
            if (i < EDGE_SPLIT) {
                batch0.push_back(row);
            } else {
                batch1.push_back(row);
            }
        }
        std::vector<ArrowArrayWrapper> arrays;
        arrays.push_back(makeKnowsBatch(batch0));
        arrays.push_back(makeKnowsBatch(batch1));
        auto result = ArrowTableSupport::createRelTableFromArrowTable(connection, "lb_chain",
            "lb_person", "lb_person", std::move(schema), std::move(arrays));
        ASSERT_TRUE(result.queryResult->isSuccess()) << result.queryResult->getErrorMessage();
    }
}

} // namespace