SystemValidator.h

Go to the documentation of this file.

/*
 * Copyright 2022 HEAVY.AI, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#pragma once

#include <Shared/measure.h>
#include "Catalog/Catalog.h"
#include "Logger/Logger.h"

#include "LeafAggregator.h"
#include "QueryState.h"

class DBHandler;

namespace system_validator {
std::string validate_table_epochs(
    const std::vector<Catalog_Namespace::TableEpochInfo>& table_epochs,
    const std::string& table_name,
    const bool is_cluster_validation);

bool should_validate_epoch(const TableDescriptor* table_descriptor);

class DistributedValidate {
 public:
  DistributedValidate(const std::string& type,
                      const bool is_repair_type_remove,
                      Catalog_Namespace::Catalog& cat,
                      LeafAggregator& leaf_aggregator,
                      const Catalog_Namespace::SessionInfo session_info,
                      DBHandler& db_handler)
      : cat_(cat)
      , type_(type)
      , is_repair_type_remove_(is_repair_type_remove)
      , leaf_aggregator_(leaf_aggregator)
      , session_info_(session_info)
      , db_handler_(db_handler) {
    if (to_upper(type) != "CLUSTER") {
      throw std::runtime_error{
          "Unexpected validation type specified. Only the \"VALIDATE CLUSTER;\" command "
          "is currently supported."};
    }
  }

  std::string validate(query_state::QueryStateProxy query_state_proxy) const {
    return nullptr;
  };

 private:
  std::string validateEpochs(const std::vector<TTableMeta>& table_meta_vector) const;

  Catalog_Namespace::Catalog& cat_;
  const std::string type_;
  const bool is_repair_type_remove_;
  LeafAggregator& leaf_aggregator_;
  const Catalog_Namespace::SessionInfo session_info_;
  DBHandler& db_handler_;
};

class SingleNodeValidator {
 public:
  SingleNodeValidator(const std::string& type, Catalog_Namespace::Catalog& catalog)
      : catalog_(catalog) {
    if (!type.empty()) {
      throw std::runtime_error{
          "Unexpected validation type specified. Only the \"VALIDATE;\" command is "
          "currently supported."};
    }
  }

  std::string validate() const;

 private:
  Catalog_Namespace::Catalog& catalog_;
};
}  // namespace system_validator