fixed

Author: sunxiaojian

api/src/main/java/org/apache/gravitino/rel/expressions/distributions/Distributions.java

@@ -74,6 +74,17 @@ public static Distribution of(Strategy strategy, int number, Expression... expre
     return new DistributionImpl(strategy, number, expressions);
   }
 
+  /**
+   * Create a distribution by the given strategy, the number of buckets is not used.
+   *
+   * @param strategy The strategy to use
+   * @param expressions The expressions to distribute by
+   * @return The created distribution
+   */
+  public static Distribution auto(Strategy strategy, Expression... expressions) {
+    return new DistributionImpl(strategy, -1, expressions);
+  }
+
   /**
    * Create a distribution on columns. Like distribute by (a) or (a, b), for complex like
    * distributing by (func(a), b) or (func(a), func(b)), please use {@link DistributionImpl.Builder}

catalogs/catalog-jdbc-doris/src/main/java/org/apache/gravitino/catalog/doris/utils/DorisUtils.java

@@ -203,7 +203,7 @@ public static Distribution extractDistributionInfoFromSql(String createTableSql)
                   .toArray(String[]::new);
 
       // Default bucket number is 1. auto is -1.
-      int bucketNum = getBucketNum(matcher);
+      int bucketNum = extractBucketNum(matcher);
 
       return new DistributionImpl.Builder()
           .withStrategy(Strategy.getByName(distributionType))
@@ -218,7 +218,7 @@ public static Distribution extractDistributionInfoFromSql(String createTableSql)
     throw new RuntimeException("Failed to extract distribution info in sql:" + createTableSql);
   }
 
-  private static int getBucketNum(Matcher matcher) {
+  private static int extractBucketNum(Matcher matcher) {
     int bucketNum = 1;
     if (matcher.find(5)) {
       String bucketValue = matcher.group(5);

catalogs/catalog-jdbc-doris/src/test/java/org/apache/gravitino/catalog/doris/integration/test/CatalogDorisIT.java

@@ -61,6 +61,7 @@
 import org.apache.gravitino.rel.expressions.NamedReference;
 import org.apache.gravitino.rel.expressions.distributions.Distribution;
 import org.apache.gravitino.rel.expressions.distributions.Distributions;
+import org.apache.gravitino.rel.expressions.distributions.Strategy;
 import org.apache.gravitino.rel.expressions.literals.Literal;
 import org.apache.gravitino.rel.expressions.literals.Literals;
 import org.apache.gravitino.rel.expressions.sorts.SortOrder;
@@ -1004,4 +1005,35 @@ void testAllDistribution() {
       tableCatalog.dropTable(tableIdentifier);
     }
   }
+
+  @Test
+  void testAllDistributionWithAuto() {
+    Distribution distribution =
+        Distributions.auto(Strategy.HASH, NamedReference.field(DORIS_COL_NAME1));
+
+    String tableName = GravitinoITUtils.genRandomName("test_distribution_table");
+    NameIdentifier tableIdentifier = NameIdentifier.of(schemaName, tableName);
+    Column[] columns = createColumns();
+    Index[] indexes = Indexes.EMPTY_INDEXES;
+    Map<String, String> properties = createTableProperties();
+    Transform[] partitioning = Transforms.EMPTY_TRANSFORM;
+    TableCatalog tableCatalog = catalog.asTableCatalog();
+    tableCatalog.createTable(
+        tableIdentifier,
+        columns,
+        table_comment,
+        properties,
+        partitioning,
+        distribution,
+        null,
+        indexes);
+    // load table
+    Table loadTable = tableCatalog.loadTable(tableIdentifier);
+
+    Assertions.assertEquals(distribution.strategy(), loadTable.distribution().strategy());
+    Assertions.assertEquals(distribution.number(), loadTable.distribution().number());
+    Assertions.assertArrayEquals(
+        distribution.expressions(), loadTable.distribution().expressions());
+    tableCatalog.dropTable(tableIdentifier);
+  }
 }

common/src/main/java/org/apache/gravitino/dto/rel/DistributionDTO.java

@@ -163,7 +163,8 @@ public DistributionDTO build() {
       strategy = strategy == null ? Strategy.HASH : strategy;
 
       Preconditions.checkState(args != null, "expressions cannot be null");
-      Preconditions.checkState(number >= 0, "bucketNum must be greater than 0");
+      // Check if the number of buckets is greater than -1, -1 is auto.
+      Preconditions.checkState(number >= -1, "bucketNum must be greater than -1");
       return new DistributionDTO(strategy, number, args);
     }
   }

docs/table-partitioning-distribution-sort-order-indexes.md

@@ -84,9 +84,21 @@ Distributions.of(Strategy.HASH, 4, NamedReference.field("score"));
 ```
 
 </TabItem>
+
+<TabItem value="java" label="Java">
+if you want to use auto distribution, you can use the following code, it will set the number is -1.
+
+```java
+// Auto distribution with strategy and fields
+Distributions.auto(Strategy.HASH, NamedReference.field("score"));
+```
+</TabItem>
+
 </Tabs>
 
 
+
+
 ## Sort ordering
 
 To define a sorted order table, you should use the following three components to construct a valid sorted order table.