external-data-source-executor.h

Go to the documentation of this file.

// Copyright 2014 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_EXEC_EXTERNAL_DATA_SOURCE_EXECUTOR_H
#define IMPALA_EXEC_EXTERNAL_DATA_SOURCE_EXECUTOR_H

#include <jni.h>
#include <string>

#include "common/status.h"

#include "gen-cpp/ExternalDataSource_types.h"

namespace impala {

class ExternalDataSourceExecutor {
 public:
  ExternalDataSourceExecutor()
      : is_initialized_(false), executor_class_(NULL), executor_(NULL) {
  };

  virtual ~ExternalDataSourceExecutor();

  Status Init(const std::string& jar_path, const std::string& class_name,
      const std::string& api_version);

  Status Open(const impala::extdatasource::TOpenParams& params,
      impala::extdatasource::TOpenResult* result);

  Status GetNext(const impala::extdatasource::TGetNextParams& params,
      impala::extdatasource::TGetNextResult* result);

  Status Close(const impala::extdatasource::TCloseParams& params,
      impala::extdatasource::TCloseResult* result);

 private:
  bool is_initialized_; // Set true in Init() to ensure the class is initialized.

  jclass executor_class_;

  jobject executor_;
  jmethodID ctor_;
  jmethodID open_id_;  // ExternalDataSourceExecutor.open()
  jmethodID get_next_id_;  // ExternalDataSourceExecutor.getNext()
  jmethodID close_id_;  // ExternalDataSourceExecutor.close()
};

}

#endif