Function main

analytical_engine/test/graphx_loader_test.cc:72–160

}

int main(int argc, char** argv) {
  if (argc != 2) {
    printf("usage: ./graphx_loader_test <ipc_socket>\n");
    return 1;
  }
  int index = 1;
  std::string ipc_socket = std::string(argv[index++]);

  vineyard::Client client;
  VINEYARD_CHECK_OK(client.Connect(ipc_socket));

  LOG(INFO) << "Connected to IPCServer: " << ipc_socket;

  grape::InitMPIComm();

  int vertices_num = 5;
  int edges_num = 4;
  vineyard::ObjectID arrow_frag_id;
  grape::CommSpec comm_spec;
  comm_spec.Init(MPI_COMM_WORLD);
  gs::GraphXPartitioner<int64_t> partitioner;
  std::vector<int> pid2Fid;
  for (grape::fid_t i = 0; i < comm_spec.fnum(); ++i) {
    pid2Fid.push_back(i);
  }
  partitioner.Init(pid2Fid);
  {
    std::vector<int64_t> oids, vdatas, src_oids, dst_oids, edatas;
    initVerticesEdges(comm_spec.fid(), comm_spec.fnum(), vertices_num, oids,
                      src_oids, dst_oids);
    initLongData(vdatas, vertices_num);
    initLongData(edatas, edges_num);
    LOG(INFO) << "Finish init data";
    gs::GraphXRawDataBuilder<int64_t, uint64_t, int64_t, int64_t> builder(
        client, oids, vdatas, src_oids, dst_oids, edatas);
    auto res = builder.MySeal(client);
    LOG(INFO) << "Built raw data: " << res->id()
              << ", edge num: " << res->GetEdgeNum()
              << ", vertex num: " << res->GetVertexNum();

    gs::GraphXLoader<int64_t, uint64_t, int64_t, int64_t> loader(
        res->id(), client, comm_spec, partitioner);
    arrow_frag_id = boost::leaf::try_handle_all(
        [&loader]() { return loader.LoadFragment(); },
        [](const vineyard::GSError& e) {
          LOG(FATAL) << e.error_msg;
          return 0;
        },
        [](const boost::leaf::error_info& unmatched) {
          LOG(FATAL) << "Unmatched error " << unmatched;
          return 0;
        });
  }
  LOG(INFO) << "Got pritive arrow fragment id: " << arrow_frag_id;
  {
    std::vector<int64_t> oids, src_oids, dst_oids;
    std::vector<char> vdata_buffer, edata_buffer;
    std::vector<int32_t> vdata_offset, edata_offset;