fix

apache · Dec 9, 2024 · 73990ae · 73990ae
1 parent 7b8a29c
commit 73990ae
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/...rc/main/java/org/apache/doris/nereids/trees/plans/commands/AdminRebalanceDiskCommand.java b/...rc/main/java/org/apache/doris/nereids/trees/plans/commands/AdminRebalanceDiskCommand.java
@@ -19,6 +19,7 @@
 
 import org.apache.doris.catalog.Env;
 import org.apache.doris.common.AnalysisException;
+import org.apache.doris.common.DdlException;
 import org.apache.doris.common.ErrorCode;
 import org.apache.doris.common.ErrorReport;
 import org.apache.doris.common.util.NetUtils;
@@ -41,7 +42,7 @@
 /**
  * admin rebalance disk
  */
-public class AdminRebalanceDiskCommand extends Command {
+public class AdminRebalanceDiskCommand extends Command implements NoForward {
     private static final Logger LOG = LogManager.getLogger(AdminRebalanceDiskCommand.class);
     private final long timeoutS = 24 * 3600; // default 24 hours
     private List<String> backends;
@@ -95,6 +96,12 @@ private void handleRebalanceDisk() throws AnalysisException {
         Env.getCurrentEnv().getTabletScheduler().rebalanceDisk(needRebalanceDiskBackends, timeoutS);
     }
 
+    @Override
+    protected void checkSupportedInCloudMode(ConnectContext ctx) throws DdlException {
+        LOG.info("AdminRebalanceDiskCommand not supported in cloud mode");
+        throw new DdlException("Unsupported operation");
+    }
+
     @Override
     public <R, C> R accept(PlanVisitor<R, C> visitor, C context) {
         return visitor.visitAdminRebalanceDiskCommand(this, context);