[Enhancement] Allow CN to run script (StarRocks#44982)

Signed-off-by: Binglin Chang <[email protected]>
vivo · May 9, 2024 · 3b2003e · 3b2003e
1 parent ed92273
commit 3b2003e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/fe/fe-core/src/main/java/com/starrocks/qe/ExecuteScriptExecutor.java b/fe/fe-core/src/main/java/com/starrocks/qe/ExecuteScriptExecutor.java
@@ -24,7 +24,7 @@
 import com.starrocks.rpc.BackendServiceClient;
 import com.starrocks.server.GlobalStateMgr;
 import com.starrocks.sql.ast.ExecuteScriptStmt;
-import com.starrocks.system.Backend;
+import com.starrocks.system.ComputeNode;
 import com.starrocks.thrift.TNetworkAddress;
 import groovy.lang.Binding;
 import groovy.lang.GroovyShell;
@@ -79,7 +79,7 @@ private static ShowResultSet executeFrontendScript(ExecuteScriptStmt stmt, Conne
     }
 
     private static ShowResultSet executeBackendScript(ExecuteScriptStmt stmt, ConnectContext ctx) throws UserException {
-        Backend be = GlobalStateMgr.getCurrentState().getNodeMgr().getClusterInfo().getBackend(stmt.getBeId());
+        ComputeNode be = GlobalStateMgr.getCurrentState().getNodeMgr().getClusterInfo().getBackendOrComputeNode(stmt.getBeId());
         if (be == null) {
             throw new UserException("node not found: " + stmt.getBeId());
         }