[apache#1339] feat(table): Add index for tables.

Clearvive · Jan 9, 2024 · 7b0f1cf · 7b0f1cf
1 parent 9614ffd
commit 7b0f1cf
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/core/src/test/java/com/datastrato/gravitino/TestCatalogOperations.java b/core/src/test/java/com/datastrato/gravitino/TestCatalogOperations.java
@@ -22,6 +22,7 @@
 import com.datastrato.gravitino.rel.Table;
 import com.datastrato.gravitino.rel.TableCatalog;
 import com.datastrato.gravitino.rel.TableChange;
+import com.datastrato.gravitino.rel.expressions.Indexes.Index;
 import com.datastrato.gravitino.rel.expressions.distributions.Distribution;
 import com.datastrato.gravitino.rel.expressions.sorts.SortOrder;
 import com.datastrato.gravitino.rel.expressions.transforms.Transform;
@@ -83,7 +84,8 @@ public Table createTable(
       Map<String, String> properties,
       Transform[] partitions,
       Distribution distribution,
-      SortOrder[] sortOrders)
+      SortOrder[] sortOrders,
+      Index[] indexes)
       throws NoSuchSchemaException, TableAlreadyExistsException {
     AuditInfo auditInfo =
         new AuditInfo.Builder().withCreator("test").withCreateTime(Instant.now()).build();
@@ -98,6 +100,7 @@ public Table createTable(
             .withDistribution(distribution)
             .withSortOrders(sortOrders)
             .withPartitioning(partitions)
+            .withIndexes(indexes)
             .build();
 
     if (tables.containsKey(ident)) {
@@ -115,6 +118,7 @@ public Table createTable(
         .withDistribution(distribution)
         .withSortOrders(sortOrders)
         .withPartitioning(partitions)
+        .withIndexes(indexes)
         .build();
   }