hbase-table-writer.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_EXEC_HBASE_TABLE_WRITER_H
#define IMPALA_EXEC_HBASE_TABLE_WRITER_H

#include <jni.h>
#include <boost/thread.hpp>
#include <boost/scoped_ptr.hpp>
#include <vector>
#include <string>
#include <utility>

#include "common/status.h"
#include "runtime/runtime-state.h"
#include "runtime/descriptors.h"
#include "runtime/row-batch.h"
#include "runtime/hbase-table.h"

namespace impala {

class HBaseTableWriter {
 public:
  HBaseTableWriter(HBaseTableDescriptor* table_desc,
                   const std::vector<ExprContext*>& output_expr_ctxs,
                   RuntimeProfile* profile);
  Status AppendRowBatch(RowBatch* batch);

  void Close(RuntimeState* state);

  Status Init(RuntimeState* state);

  static Status InitJNI();

 private:
  Status CreatePut(JNIEnv* env, const void* rk, int rk_len, jobject* put);

  Status CreateByteArray(JNIEnv* env, const std::string& s,
                         jbyteArray* j_array);

  Status CreateByteArray(JNIEnv* env, const void* data, int data_len,
                         jbyteArray* j_array);

  Status CreatePutList(JNIEnv* env, int num_puts);

  Status CleanUpJni();

  HBaseTableDescriptor* table_desc_;

  boost::scoped_ptr<HBaseTable> table_;

  const std::vector<ExprContext*> output_expr_ctxs_;

  std::vector<int> output_exprs_byte_sizes_;

  jobject put_list_;

  static jclass put_cl_;

  static jmethodID put_ctor_;

  static jmethodID put_add_id_;

  static jclass list_cl_;

  static jmethodID list_ctor_;

  static jmethodID list_add_id_;

  std::vector<jbyteArray> cf_arrays_;

  std::vector<jbyteArray> qual_arrays_;

  RuntimeProfile* runtime_profile_;
  RuntimeProfile::Counter* encoding_timer_;
  RuntimeProfile::Counter* htable_put_timer_;
};

}  // namespace impala
#endif