refactor: QueryGraphviz.cpp to use Connectors class (#800)

Summary:
Pull Request resolved: #800

Refactor QueryGraphviz.cpp to use the existing `Connectors` class instead of inline connector registration code.

Reviewed By: xiaoxmeng

Differential Revision: D91510943

fbshipit-source-id: 08130ea06dfe77f67e618f19c9dab357de06fb2a

Author: mbasmanova

axiom/cli/CMakeLists.txt

@@ -41,14 +41,15 @@ target_link_libraries(
   velox_parse_utils
 )
 
-add_executable(axiom_graphviz QueryGraphviz.cpp SqlQueryRunner.cpp)
+add_executable(axiom_graphviz Connectors.cpp QueryGraphviz.cpp SqlQueryRunner.cpp)
 
 target_link_libraries(
   axiom_graphviz
   axiom_runner_local_runner
   axiom_runner_multifragment_plan
   axiom_optimizer
   axiom_tpch_connector_metadata
+  axiom_hive_connector_metadata
   axiom_sql_presto_parser
   velox_exec_test_lib
   velox_parse_parser

axiom/cli/QueryGraphviz.cpp

@@ -36,9 +36,8 @@
 #include <fstream>
 #include <iostream>
 #include <sstream>
+#include "axiom/cli/Connectors.h"
 #include "axiom/cli/SqlQueryRunner.h"
-#include "axiom/connectors/tpch/TpchConnectorMetadata.h"
-#include "velox/connectors/Connector.h"
 
 DEFINE_string(
     query,
@@ -52,23 +51,6 @@ DEFINE_string(
 namespace facebook::axiom {
 namespace {
 
-std::string initializeTpchConnector() {
-  auto emptyConfig = std::make_shared<velox::config::ConfigBase>(
-      std::unordered_map<std::string, std::string>{});
-
-  velox::connector::tpch::TpchConnectorFactory factory;
-  auto connector = factory.newConnector("tpch", emptyConfig);
-  velox::connector::registerConnector(connector);
-
-  connector::ConnectorMetadata::registerMetadata(
-      connector->connectorId(),
-      std::make_shared<connector::tpch::TpchConnectorMetadata>(
-          dynamic_cast<velox::connector::tpch::TpchConnector*>(
-              connector.get())));
-
-  return connector->connectorId();
-}
-
 /// Returns the path to the 'dot' executable.
 std::string findDotExecutable() {
   static const std::vector<std::string> kDotPaths = {
@@ -153,10 +135,12 @@ int main(int argc, char** argv) {
   facebook::velox::memory::MemoryManager::initialize(
       facebook::velox::memory::MemoryManager::Options{});
 
+  facebook::axiom::Connectors connectors;
+  auto connector = connectors.registerTpchConnector();
+
   axiom::sql::SqlQueryRunner runner;
-  runner.initialize([](auto& /*history*/) {
-    return std::make_pair(
-        facebook::axiom::initializeTpchConnector(), std::nullopt);
+  runner.initialize([&](auto& /*history*/) {
+    return std::make_pair(connector->connectorId(), std::nullopt);
   });
 
   const auto dot = runner.toDot(query);