| /** |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you 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. |
| **/ |
| |
| #ifndef QUICKSTEP_QUERY_OPTIMIZER_TESTS_EXECUTION_GENERATOR_TEST_RUNNER_HPP_ |
| #define QUICKSTEP_QUERY_OPTIMIZER_TESTS_EXECUTION_GENERATOR_TEST_RUNNER_HPP_ |
| |
| #include <memory> |
| #include <set> |
| #include <string> |
| #include <utility> |
| #include <vector> |
| |
| #include "parser/SqlParserWrapper.hpp" |
| #include "query_execution/ForemanSingleNode.hpp" |
| #include "query_execution/QueryExecutionTypedefs.hpp" |
| #include "query_execution/Worker.hpp" |
| #include "query_execution/WorkerDirectory.hpp" |
| #include "query_execution/WorkerMessage.hpp" |
| #include "query_optimizer/Optimizer.hpp" |
| #include "query_optimizer/tests/TestDatabaseLoader.hpp" |
| #include "threading/ThreadIDBasedMap.hpp" |
| #include "utility/Macros.hpp" |
| #include "utility/textbased_test/TextBasedTestDriver.hpp" |
| |
| #include "tmb/id_typedefs.h" |
| #include "tmb/message_bus.h" |
| |
| namespace quickstep { |
| namespace optimizer { |
| |
| /** |
| * @brief TextBasedTestRunner for testing the ExecutionGenerator. |
| */ |
| class ExecutionGeneratorTestRunner : public TextBasedTestRunner { |
| public: |
| /** |
| * @brief If this option is enabled, recreate the entire database and |
| * repopulate the data before every test. |
| */ |
| static const std::string kResetOption; |
| |
| /** |
| * @brief Constructor. |
| */ |
| explicit ExecutionGeneratorTestRunner(const std::string &storage_path) |
| : test_database_loader_(storage_path), |
| thread_id_map_(ClientIDMap::Instance()) { |
| test_database_loader_.createTestRelation(false /* allow_vchar */); |
| test_database_loader_.loadTestRelation(); |
| |
| bus_.Initialize(); |
| |
| main_thread_client_id_ = bus_.Connect(); |
| bus_.RegisterClientAsSender(main_thread_client_id_, kAdmitRequestMessage); |
| bus_.RegisterClientAsSender(main_thread_client_id_, kPoisonMessage); |
| bus_.RegisterClientAsReceiver(main_thread_client_id_, kWorkloadCompletionMessage); |
| |
| worker_.reset(new Worker(0, &bus_)); |
| |
| std::vector<client_id> worker_client_ids; |
| worker_client_ids.push_back(worker_->getBusClientID()); |
| |
| // We don't use the NUMA aware version of foreman code. |
| std::vector<int> numa_nodes; |
| numa_nodes.push_back(-1); |
| |
| workers_.reset(new WorkerDirectory(1 /* number of workers */, |
| worker_client_ids, numa_nodes)); |
| foreman_.reset( |
| new ForemanSingleNode(main_thread_client_id_, |
| workers_.get(), |
| &bus_, |
| test_database_loader_.catalog_database(), |
| test_database_loader_.storage_manager())); |
| |
| foreman_->start(); |
| worker_->start(); |
| } |
| |
| ~ExecutionGeneratorTestRunner() { |
| QueryExecutionUtil::BroadcastPoisonMessage(main_thread_client_id_, &bus_); |
| worker_->join(); |
| foreman_->join(); |
| } |
| |
| void runTestCase(const std::string &input, |
| const std::set<std::string> &options, |
| std::string *output) override; |
| |
| private: |
| SqlParserWrapper sql_parser_; |
| TestDatabaseLoader test_database_loader_; |
| Optimizer optimizer_; |
| |
| MessageBusImpl bus_; |
| std::unique_ptr<ForemanSingleNode> foreman_; |
| std::unique_ptr<Worker> worker_; |
| |
| std::unique_ptr<WorkerDirectory> workers_; |
| |
| tmb::client_id main_thread_client_id_; |
| |
| // This map is needed for InsertDestination and some operators that send |
| // messages to Foreman directly. To know the reason behind the design of this |
| // map, see the note in InsertDestination.hpp. |
| ClientIDMap *thread_id_map_; |
| |
| DISALLOW_COPY_AND_ASSIGN(ExecutionGeneratorTestRunner); |
| }; |
| |
| } // namespace optimizer |
| } // namespace quickstep |
| |
| #endif /* QUICKSTEP_QUERY_OPTIMIZER_TESTS_EXECUTION_GENERATOR_TEST_RUNNER_HPP_ */ |