include/hbase/test-util/mini-cluster.h

/*
 * 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.
 *
 */
#pragma once

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

namespace hbase {

/**
 * JNI wrapper for the Java based mini cluster implementation. StartCluster() must be called before calling any other
 * methods that operate on the JVM, otherwise it triggers an assert that results in a crash.
 */
class MiniCluster {
 public:
  virtual ~MiniCluster();
  jobject StartCluster(int32_t num_region_servers);
  void StopCluster();
  jobject CreateTable(const std::string &table, const std::string &family);
  jobject CreateTable(const std::string &table, const std::vector<std::string> &families);
  jobject CreateTable(const std::string &table, const std::string &family,
                      const std::vector<std::string> &keys);
  jobject CreateTable(const std::string &table, const std::vector<std::string> &families,
                      const std::vector<std::string> &keys);
  // moves region to server
  void MoveRegion(const std::string &region, const std::string &server);
  // returns the Configuration instance for the cluster
  jobject GetConf();
  // returns the value for config key retrieved from cluster
  std::string GetConfValue(const std::string &key);

 private:
  JNIEnv *env_;
  jclass testing_util_class_;
  jclass table_name_class_;
  jclass put_class_;
  jclass conf_class_;
  jmethodID stop_rs_mid_;
  jmethodID get_conf_mid_;
  jmethodID set_conf_mid_;
  jmethodID tbl_name_value_of_mid_;
  jmethodID create_table_mid_;
  jmethodID create_table_families_mid_;
  jmethodID create_table_with_split_mid_;
  jmethodID put_mid_;
  jmethodID put_ctor_;
  jmethodID add_col_mid_;
  jmethodID create_conn_mid_;
  jmethodID get_conn_mid_;
  jmethodID get_table_mid_;
  jmethodID conf_get_mid_;
  jmethodID get_admin_mid_;
  jmethodID move_mid_;
  jmethodID str_ctor_mid_;
  jobject htu_;
  jobject cluster_;
  // Mutex to be acquired to make changes to the cluster state (start/stop).
  std::mutex lock_;
  // To make Setup() calls idempotent.
  bool inited_ = false;
  JavaVM *jvm_;
  JNIEnv *CreateVM(JavaVM **jvm);
  void Setup();
  jbyteArray StrToByteChar(const std::string &str);
  jobject admin();
  // Util to get a global ref so that the objected is not GC'ed
  jclass FindClassAndGetGlobalRef(const char* cls_name);
};
} /*namespace hbase*/