コード例 #1
0
ファイル: HBaseTable.java プロジェクト: ZhuChaoyu/Impala
  /**
   * Returns statistics on this table as a tabular result set. Used for the SHOW TABLE STATS
   * statement. The schema of the returned TResultSet is set inside this method.
   */
  public TResultSet getTableStats() {
    TResultSet result = new TResultSet();
    TResultSetMetadata resultSchema = new TResultSetMetadata();
    result.setSchema(resultSchema);
    resultSchema.addToColumns(new TColumn("Region Location", ColumnType.STRING.toThrift()));
    resultSchema.addToColumns(new TColumn("Start RowKey", ColumnType.STRING.toThrift()));
    resultSchema.addToColumns(new TColumn("Est. #Rows", ColumnType.BIGINT.toThrift()));
    resultSchema.addToColumns(new TColumn("Size", ColumnType.STRING.toThrift()));

    // TODO: Consider fancier stats maintenance techniques for speeding up this process.
    // Currently, we list all regions and perform a mini-scan of each of them to
    // estimate the number of rows, the data size, etc., which is rather expensive.
    try {
      long totalNumRows = 0;
      long totalHdfsSize = 0;
      List<HRegionLocation> regions =
          HBaseTable.getRegionsInRange(
              hTable_, HConstants.EMPTY_END_ROW, HConstants.EMPTY_START_ROW);
      for (HRegionLocation region : regions) {
        TResultRowBuilder rowBuilder = new TResultRowBuilder();
        HRegionInfo regionInfo = region.getRegionInfo();
        Pair<Long, Long> estRowStats =
            getEstimatedRowStats(regionInfo.getStartKey(), regionInfo.getEndKey());

        long numRows = estRowStats.first.longValue();
        long hdfsSize = getHdfsSize(regionInfo);
        totalNumRows += numRows;
        totalHdfsSize += hdfsSize;

        // Add the region location, start rowkey, number of rows and raw Hdfs size.
        rowBuilder
            .add(String.valueOf(region.getHostname()))
            .add(Bytes.toString(regionInfo.getStartKey()))
            .add(numRows)
            .addBytes(hdfsSize);
        result.addToRows(rowBuilder.get());
      }

      // Total num rows and raw Hdfs size.
      if (regions.size() > 1) {
        TResultRowBuilder rowBuilder = new TResultRowBuilder();
        rowBuilder.add("Total").add("").add(totalNumRows).addBytes(totalHdfsSize);
        result.addToRows(rowBuilder.get());
      }
    } catch (IOException e) {
      throw new RuntimeException(e);
    }
    return result;
  }
コード例 #2
0
ファイル: CatalogTest.java プロジェクト: hihihippp/Impala
 private void checkHBaseTableCols(
     Db db,
     String hiveTableName,
     String hbaseTableName,
     String[] hiveColNames,
     String[] colFamilies,
     String[] colQualifiers,
     ColumnType[] colTypes)
     throws TableLoadingException {
   checkTableCols(db, hiveTableName, 1, hiveColNames, colTypes);
   HBaseTable tbl = (HBaseTable) db.getTable(hiveTableName);
   assertEquals(tbl.getHBaseTableName(), hbaseTableName);
   List<Column> cols = tbl.getColumns();
   assertEquals(colFamilies.length, colTypes.length);
   assertEquals(colQualifiers.length, colTypes.length);
   Iterator<Column> it = cols.iterator();
   int i = 0;
   while (it.hasNext()) {
     HBaseColumn col = (HBaseColumn) it.next();
     assertEquals(col.getColumnFamily(), colFamilies[i]);
     assertEquals(col.getColumnQualifier(), colQualifiers[i]);
     ++i;
   }
 }