From b3f79f7ae4c90ff18d4ed66339bddf72bc9045a9 Mon Sep 17 00:00:00 2001
From: xionglei1 <xionglei1@xiaomi.com>
Date: Tue, 24 Sep 2019 16:46:28 +0800
Subject: [PATCH] add buckets limit: buckets > 0 while add partition

---
 fe/src/main/java/org/apache/doris/catalog/Catalog.java | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/fe/src/main/java/org/apache/doris/catalog/Catalog.java b/fe/src/main/java/org/apache/doris/catalog/Catalog.java
index 1e835ed1f397c9..aaa74901e9cbfb 100644
--- a/fe/src/main/java/org/apache/doris/catalog/Catalog.java
+++ b/fe/src/main/java/org/apache/doris/catalog/Catalog.java
@@ -2869,13 +2869,17 @@ public Pair<Long, Partition> addPartition(Database db, String tableName, OlapTab
                 }
 
                 if (distributionInfo.getType() == DistributionInfoType.HASH) {
-                    List<Column> newDistriCols = ((HashDistributionInfo) distributionInfo).getDistributionColumns();
+                    HashDistributionInfo hashDistributionInfo = (HashDistributionInfo) distributionInfo;
+                    List<Column> newDistriCols = hashDistributionInfo.getDistributionColumns();
                     List<Column> defaultDistriCols = ((HashDistributionInfo) defaultDistributionInfo)
                             .getDistributionColumns();
                     if (!newDistriCols.equals(defaultDistriCols)) {
                         throw new DdlException("Cannot assign hash distribution with different distribution cols. "
                                 + "default is: " + defaultDistriCols);
                     }
+                    if (hashDistributionInfo.getBucketNum() <= 0) {
+                        throw new DdlException("Cannot assign hash distribution buckets less than 1");
+                    }
                 }
             } else {
                 distributionInfo = defaultDistributionInfo;