Merge pull request #64 from mgarolera/create_column_families

Send column families when creating a table.
googleapis · Oct 15, 2014 · e581a3e · e581a3e
2 parents 047fb5b + 9cbc118
commit e581a3e
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/src/main/java/org/apache/hadoop/hbase/client/AnviltopAdmin.java b/src/main/java/org/apache/hadoop/hbase/client/AnviltopAdmin.java
@@ -2,6 +2,7 @@
 
 
 import com.google.bigtable.anviltop.AnviltopAdminServices;
+import com.google.bigtable.anviltop.AnviltopData;
 import com.google.cloud.anviltop.hbase.AnviltopOptions;
 import com.google.cloud.anviltop.hbase.adapters.ColumnDescriptorAdapter;
 import com.google.cloud.hadoop.hbase.AnviltopAdminClient;
@@ -115,11 +116,15 @@ public HTableDescriptor getTableDescriptor(TableName tableName)
 
   @Override
   public void createTable(HTableDescriptor desc) throws IOException {
-    anviltopAdminClient.createTable(
-        AnviltopAdminServices.CreateTableRequest.newBuilder()
-            .setProjectId(options.getProjectId())
-            .setTableNameBytes(ByteString.copyFrom(desc.getName()))
-            .build());
+    AnviltopAdminServices.CreateTableRequest.Builder builder = AnviltopAdminServices.CreateTableRequest.newBuilder();
+    builder.setProjectId(options.getProjectId())
+        .setTableNameBytes(ByteString.copyFrom(desc.getName()));
+
+    for (HColumnDescriptor column : desc.getColumnFamilies()) {
+      builder.addColumnFamilies(AnviltopData.ColumnFamily.newBuilder().setName(column.getNameAsString()));
+    }
+
+    anviltopAdminClient.createTable(builder.build());
   }
 
   @Override