How to use

partCols

method

in

org.apache.hive.hcatalog.api.HCatCreateTableDesc$Builder

@Test
public void testPartitionSchema() throws Exception {
 try {
  HCatClient client = HCatClient.create(new Configuration(hcatConf));
  final String dbName = "myDb";
  final String tableName = "myTable";
  client.dropDatabase(dbName, true, HCatClient.DropDBMode.CASCADE);
  client.createDatabase(HCatCreateDBDesc.create(dbName).build());
  List<HCatFieldSchema> columnSchema = Arrays.asList(new HCatFieldSchema("foo", Type.INT, ""),
    new HCatFieldSchema("bar", Type.STRING, ""));
  List<HCatFieldSchema> partitionSchema = Arrays.asList(new HCatFieldSchema("dt", Type.STRING, ""),
    new HCatFieldSchema("grid", Type.STRING, ""));
  client.createTable(HCatCreateTableDesc.create(dbName, tableName, columnSchema).partCols(partitionSchema).build());
  HCatTable table = client.getTable(dbName, tableName);
  List<HCatFieldSchema> partitionColumns = table.getPartCols();
  assertArrayEquals("Didn't get expected partition-schema back from the HCatTable.",
    partitionSchema.toArray(), partitionColumns.toArray());
  client.dropDatabase(dbName, false, HCatClient.DropDBMode.CASCADE);
 }
 catch (Exception unexpected) {
  LOG.error("Unexpected exception!", unexpected);
  assertTrue("Unexpected exception! " + unexpected.getMessage(), false);
 }
}

public void createTable(String db, String table, String partitionCols) throws Exception {
  List<HCatFieldSchema> cols = new ArrayList<HCatFieldSchema>();
  cols.add(new HCatFieldSchema("userid", Type.INT, "userid"));
  cols.add(new HCatFieldSchema("viewtime", Type.BIGINT, "view time"));
  cols.add(new HCatFieldSchema("pageurl", Type.STRING, "page url visited"));
  cols.add(new HCatFieldSchema("ip", Type.STRING, "IP Address of the User"));
  ArrayList<HCatFieldSchema> ptnCols = new ArrayList<HCatFieldSchema>();
  for (String partitionCol : partitionCols.split(",")) {
    ptnCols.add(new HCatFieldSchema(partitionCol, Type.STRING, null));
  }
  // Remove this once NotificationListener is fixed and available in HCat snapshot
  Map<String, String> tblProps = new HashMap<String, String>();
  tblProps.put(HCatConstants.HCAT_MSGBUS_TOPIC_NAME, "hcat." + db + "." + table);
  HCatCreateTableDesc tableDesc = HCatCreateTableDesc.create(db, table, cols).fileFormat("textfile")
      .partCols(ptnCols).tblProps(tblProps ).build();
  hcatClient.createTable(tableDesc);
  List<String> tables = hcatClient.listTableNamesByPattern(db, "*");
  assertTrue(tables.contains(table));
}