Class ClickHouseTools

tests/integration/clickhouse_integration_test.cpp:76–187

Tool implementations

// Tool implementations
class ClickHouseTools {
 public:
  explicit ClickHouseTools(std::shared_ptr<clickhouse::Client> client)
      : client_(std::move(client)) {}

  Tool createListDatabasesTool() {
    return create_simple_tool(
        "list_databases",
        "List all available databases in the ClickHouse instance", {},
        [this](const JsonValue& args, const ToolExecutionContext& context) {
          try {
            std::vector<std::string> databases;
            client_->Select(
                "SELECT name FROM system.databases ORDER BY name",
                [&databases](const clickhouse::Block& block) {
                  for (size_t i = 0; i < block.GetRowCount(); ++i) {
                    databases.push_back(std::string(
                        block[0]->As<clickhouse::ColumnString>()->At(i)));
                  }
                });

            std::stringstream result;
            result << "Found " << databases.size() << " databases:\n";
            for (const auto& db : databases) {
              result << "- " << db << "\n";
            }

            return JsonValue{{"result", result.str()},
                             {"databases", databases}};
          } catch (const std::exception& e) {
            throw std::runtime_error(std::string("Error: ") + e.what());
          }
        });
  }

  Tool createListTablesInDatabaseTool() {
    return create_simple_tool(
        "list_tables_in_database", "List all tables in a specific database",
        {{"database", "string"}},
        [this](const JsonValue& args, const ToolExecutionContext& context) {
          try {
            std::string database = args["database"].get<std::string>();
            std::vector<std::string> tables;

            std::string query =
                "SELECT name FROM system.tables WHERE database = '" + database +
                "' ORDER BY name";
            client_->Select(query, [&tables](const clickhouse::Block& block) {
              for (size_t i = 0; i < block.GetRowCount(); ++i) {
                tables.push_back(std::string(
                    block[0]->As<clickhouse::ColumnString>()->At(i)));
              }
            });

            std::stringstream result;
            result << "Found " << tables.size() << " tables in database '"
                   << database << "':\n";
            for (const auto& table : tables) {