Java Hive.createDatabase方法代码示例

import org.apache.hadoop.hive.ql.metadata.Hive; //导入方法依赖的package包/类
/**
 * Before test.
 *
 * @throws Exception the exception
 */
@BeforeTest
public void beforeTest() throws Exception {
  // Check if hadoop property set
  System.out.println("###HADOOP_PATH " + System.getProperty("hadoop.bin.path"));
  assertNotNull(System.getProperty("hadoop.bin.path"));
  createDriver();
  ss = new SessionState(hiveConf, "testuser");
  SessionState.start(ss);
  Hive client = Hive.get(hiveConf);
  Database database = new Database();
  database.setName(dataBase);
  client.createDatabase(database, true);
  SessionState.get().setCurrentDatabase(dataBase);
  sessionid = SessionState.get().getSessionId();
  driverConf.setBoolean(LensConfConstants.QUERY_ADD_INSERT_OVEWRITE, false);
  QueryContext context = createContext("USE " + dataBase, this.queryConf);
  driver.execute(context);
  driverConf.setBoolean(LensConfConstants.QUERY_ADD_INSERT_OVEWRITE, true);
  driverConf.setBoolean(LensConfConstants.QUERY_PERSISTENT_RESULT_INDRIVER, true);
}
 

import org.apache.hadoop.hive.ql.metadata.Hive; //导入方法依赖的package包/类
@BeforeTest
public void setUp() throws Exception {
  super.setUp();
  Hive hive = Hive.get(new HiveConf());
  Database db = new Database();
  db.setName(TEST_DB);
  hive.createDatabase(db, true);
  LensClientConfig lensClientConfig = new LensClientConfig();
  lensClientConfig.setLensDatabase(TEST_DB);
  lensClientConfig.set(LensConfConstants.SERVER_BASE_URL,
      "http://localhost:" + getTestPort() + "/lensapi");
  LensClient client = new LensClient(lensClientConfig);
  mlClient = new LensMLClient(client);
}
 

import org.apache.hadoop.hive.ql.metadata.Hive; //导入方法依赖的package包/类
/**
 * Setup.
 *
 * @throws AlreadyExistsException the already exists exception
 * @throws HiveException          the hive exception
 * @throws IOException            Signals that an I/O exception has occurred.
 */
@BeforeTest
public void setup() throws AlreadyExistsException, HiveException, IOException {
  SessionState.start(conf);
  Hive client = Hive.get(conf);
  Database database = new Database();
  database.setName(TestDBStorage.class.getSimpleName());
  client.createDatabase(database);
  SessionState.get().setCurrentDatabase(TestDBStorage.class.getSimpleName());
}
 

import org.apache.hadoop.hive.ql.metadata.Hive; //导入方法依赖的package包/类
public static void createTestDatabaseResources(String[] testDatabases, HiveConf conf) throws Exception {
  File srcJarDir = new File("target/testjars/");
  if (!srcJarDir.exists()) {
    // nothing to setup
    return;
  }
  File resDir = new File("target/resources");
  if (!resDir.exists()) {
    resDir.mkdir();
  }

  // Create databases and resource dirs
  Hive hive = Hive.get(conf);
  File testJarFile = new File("target/testjars/test.jar");
  File serdeJarFile = new File("target/testjars/serde.jar");
  for (String db : testDatabases) {
    Database database = new Database();
    database.setName(db);
    hive.createDatabase(database, true);
    File dbDir = new File(resDir, db);
    if (!dbDir.exists()) {
      dbDir.mkdir();
    }
    // Add a jar in the directory
    try {

      String[] jarOrder = {
        "x_" + db + ".jar",
        "y_" + db + ".jar",
        "z_" + db + ".jar",
        "serde.jar",
      };

      // Jar order is -> z, y, x, File listing order is x, y, z
      // We are explicitly specifying jar order
      FileUtils.writeLines(new File(dbDir, "jar_order"), Arrays.asList(jarOrder[2], jarOrder[1],
        jarOrder[0], jarOrder[3]));

      FileUtils.copyFile(testJarFile, new File(dbDir, jarOrder[0]));
      FileUtils.copyFile(testJarFile, new File(dbDir, jarOrder[1]));
      FileUtils.copyFile(testJarFile, new File(dbDir, jarOrder[2]));
      FileUtils.copyFile(serdeJarFile, new File(dbDir, jarOrder[3]));
    } catch (FileNotFoundException fnf) {
      log.error("File not found.", fnf);
    }
  }
}
 

import org.apache.hadoop.hive.ql.metadata.Hive; //导入方法依赖的package包/类
@Test(dataProvider = "mediaTypeData")
public void testOpenSessionWithDatabase(MediaType mt) throws Exception {
  // TEST1 - Check if call with database parameter sets current database
  // Create the test DB
  Hive hive = Hive.get(new HiveConf());
  final String testDbName = TestSessionResource.class.getSimpleName();
  Database testOpenDb = new Database();
  testOpenDb.setName(testDbName);
  hive.createDatabase(testOpenDb, true);

  final WebTarget target = target().path("session");
  final FormDataMultiPart mp = new FormDataMultiPart();

  mp.bodyPart(new FormDataBodyPart(FormDataContentDisposition.name("username").build(), "foo"));
  mp.bodyPart(new FormDataBodyPart(FormDataContentDisposition.name("password").build(), "bar"));
  mp.bodyPart(new FormDataBodyPart(FormDataContentDisposition.name("database").build(), testDbName));
  mp.bodyPart(new FormDataBodyPart(FormDataContentDisposition.name("sessionconf").fileName("sessionconf").build(),
    new LensConf(), mt));

  final LensSessionHandle handle = target.request(mt).post(Entity.entity(mp, MediaType.MULTIPART_FORM_DATA_TYPE),
    LensSessionHandle.class);
  Assert.assertNotNull(handle);

  // Check if DB set in session service.
  HiveSessionService service = LensServices.get().getService(SessionService.NAME);
  LensSessionImpl session = service.getSession(handle);
  Assert.assertEquals(session.getCurrentDatabase(), testDbName, "Expected current DB to be set to " + testDbName);
  APIResult result = target.queryParam("sessionid", handle).request().delete(APIResult.class);
  Assert.assertEquals(result.getStatus(), APIResult.Status.SUCCEEDED);

  // TEST 2 - Try set database with invalid db name
  final String invalidDB = testDbName + "_invalid_db";
  final FormDataMultiPart form2 = new FormDataMultiPart();

  form2.bodyPart(new FormDataBodyPart(FormDataContentDisposition.name("username").build(), "foo"));
  form2.bodyPart(new FormDataBodyPart(FormDataContentDisposition.name("password").build(), "bar"));
  form2.bodyPart(new FormDataBodyPart(FormDataContentDisposition.name("database").build(), invalidDB));
  form2.bodyPart(new FormDataBodyPart(FormDataContentDisposition.name("sessionconf").fileName("sessionconf").build(),
    new LensConf(), mt));
  try {
    final LensSessionHandle handle2 = target.request(mt).post(Entity.entity(form2,
        MediaType.MULTIPART_FORM_DATA_TYPE), LensSessionHandle.class);
    Assert.fail("Expected above call to fail with not found exception");
  } catch (NotFoundException nfe) {
    // PASS
  }
}