Method BuildLocalCount

graphlearn/src/core/graph/graph_store.cc:305–317

}

void GraphStore::BuildLocalCount() {
  local_count_.reserve(e_types_.size() + n_types_.size());
  for (auto it = e_types_.begin(); it != e_types_.end(); ++it) {
    Graph* graph = graphs_->LookupOrCreate(it->first);
    ::graphlearn::io::GraphStorage* storage = graph->GetLocalStorage();
    local_count_.push_back(storage->GetEdgeCount() * it->second);
  }
  for (auto it = n_types_.begin(); it != n_types_.end(); ++it) {
    Noder* noder = noders_->LookupOrCreate(it->first);
    ::graphlearn::io::NodeStorage* storage = noder->GetLocalStorage();
    local_count_.push_back(storage->GetIds().Size() * it->second);
  }
}

Graph* GraphStore::GetGraph(const std::string& edge_type) {
  return graphs_->LookupOrCreate(edge_type);