impalad-query-executor.h

Go to the documentation of this file.

// Copyright 2012 Cloudera 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.


#ifndef IMPALA_TESTUTIL_IMPALAD_QUERY_EXECUTOR_H
#define IMPALA_TESTUTIL_IMPALAD_QUERY_EXECUTOR_H

#include <string>
#include <vector>
#include <boost/scoped_ptr.hpp>
#include <boost/shared_ptr.hpp>
#include <boost/thread/thread.hpp>

#include "rpc/thrift-client.h"
#include "common/status.h"
#include "runtime/runtime-state.h"
#include "runtime/types.h"
#include "gen-cpp/Types_types.h"
#include "gen-cpp/ImpalaService.h"
#include "gen-cpp/ImpalaService_types.h"

namespace impala {

class RowBatch;

class ImpaladQueryExecutor {
 public:
  ImpaladQueryExecutor();
  ~ImpaladQueryExecutor();

  Status Setup();

  Status Exec(const std::string& query_string,
      std::vector<Apache::Hadoop::Hive::FieldSchema>* col_types);

  Status Explain(const std::string& query_string, std::string* explain_plan);

  Status FetchResult(RowBatch** batch);

  Status FetchResult(std::string* row);

  Status FetchResult(std::vector<void*>* row);

  std::string ErrorString() const;

  std::string FileErrors() const;

  RuntimeProfile* query_profile();

  bool eos() { return eos_; }

  void setExecOptions(const std::vector<std::string>& exec_options) {
    exec_options_ = exec_options;
  }

 private:
  boost::scoped_ptr<ThriftClient<ImpalaServiceClient> > client_;

  std::vector<std::string> exec_options_;

  beeswax::QueryHandle query_handle_;
  beeswax::Results query_results_;
  beeswax::QueryExplanation query_explanation_;

  bool query_in_progress_;
  int current_row_;
  bool eos_;

  Status Close();
};

}

#endif