[Enhancement] improve error message of killed sql (#53708)

Signed-off-by: Murphy <[email protected]>
StarRocks · Dec 9, 2024 · 2d7f483 · 2d7f483
1 parent 880022b
commit 2d7f483
Show file tree

Hide file tree

Showing 4 changed files with 22 additions and 6 deletions.
diff --git a/fe/fe-core/src/main/java/com/starrocks/common/util/SqlUtils.java b/fe/fe-core/src/main/java/com/starrocks/common/util/SqlUtils.java
@@ -17,7 +17,12 @@
 
 package com.starrocks.common.util;
 
+import org.apache.commons.lang3.StringUtils;
+
 public class SqlUtils {
+
+    private static final int SQL_PREFIX_LENGTH = 128;
+
     public static String escapeUnquote(String ident) {
         return ident.replaceAll("``", "`");
     }
@@ -35,4 +40,14 @@ public static String getIdentSql(String ident) {
         sb.append('`');
         return sb.toString();
     }
+
+    /**
+     * Return the prefix of a sql if it's too long
+     */
+    public static String sqlPrefix(String sql) {
+        if (StringUtils.isEmpty(sql) || sql.length() < SQL_PREFIX_LENGTH) {
+            return sql;
+        }
+        return sql.substring(0, SQL_PREFIX_LENGTH) + "...";
+    }
 }
diff --git a/fe/fe-core/src/main/java/com/starrocks/qe/ConnectContext.java b/fe/fe-core/src/main/java/com/starrocks/qe/ConnectContext.java
@@ -46,6 +46,7 @@
 import com.starrocks.common.DdlException;
 import com.starrocks.common.ErrorCode;
 import com.starrocks.common.ErrorReport;
+import com.starrocks.common.util.SqlUtils;
 import com.starrocks.common.util.TimeUtils;
 import com.starrocks.common.util.UUIDUtil;
 import com.starrocks.http.HttpConnectContext;
@@ -962,7 +963,7 @@ public void kill(boolean killConnection, String cancelledMessage) {
                     Thread.sleep(10);
                     times++;
                     if (times > 100) {
-                        LOG.warn("wait for close fail, break.");
+                        LOG.warn("kill queryId={} connectId={} wait for close fail, break.", queryId, connectionId);
                         break;
                     }
                 } catch (InterruptedException e) {
@@ -1006,7 +1007,7 @@ public void checkTimeout(long now) {
             if (delta > waitTimeout * 1000L) {
                 // Need kill this connection.
                 LOG.warn("kill wait timeout connection, remote: {}, wait timeout: {}, query id: {}, sql: {}",
-                        getMysqlChannel().getRemoteHostPortString(), waitTimeout, queryId, sql);
+                        getMysqlChannel().getRemoteHostPortString(), waitTimeout, queryId, SqlUtils.sqlPrefix(sql));
 
                 killFlag = true;
                 killConnection = true;
@@ -1018,7 +1019,7 @@ public void checkTimeout(long now) {
             if (delta > timeoutSecond * 1000L) {
                 LOG.warn("kill timeout {}, remote: {}, execute timeout: {}, query id: {}, sql: {}",
                         getExecType().toLowerCase(), getMysqlChannel().getRemoteHostPortString(), timeoutSecond,
-                        queryId, sql);
+                        queryId, SqlUtils.sqlPrefix(sql));
 
                 // Only kill
                 killFlag = true;

diff --git a/fe/fe-core/src/main/java/com/starrocks/qe/ConnectScheduler.java b/fe/fe-core/src/main/java/com/starrocks/qe/ConnectScheduler.java
@@ -118,7 +118,7 @@ public void run() {
                 }
             } catch (Throwable e) {
                 //Catch Exception to avoid thread exit
-                LOG.warn("Timeout checker exception, Internal error : " + e.getMessage());
+                LOG.warn("Timeout checker exception, Internal error : {}", e.getMessage(), e);
             }
         }
     }

diff --git a/fe/fe-core/src/main/java/com/starrocks/qe/DefaultCoordinator.java b/fe/fe-core/src/main/java/com/starrocks/qe/DefaultCoordinator.java
@@ -899,7 +899,7 @@ public RowBatch getNext() throws Exception {
                 throw new RpcException("unknown", copyStatus.getErrorMsg());
             } else {
                 String errMsg = copyStatus.getErrorMsg();
-                LOG.warn("query failed: {}", errMsg);
+                LOG.warn("query {} failed: {}", connectContext.queryId, errMsg);
 
                 // hide host info
                 int hostIndex = errMsg.indexOf("host");
@@ -954,7 +954,7 @@ public void cancel(PPlanFragmentCancelReason reason, String message) {
                 queryStatus.setStatus(Status.CANCELLED);
                 queryStatus.setErrorMsg(message);
             }
-            LOG.warn("cancel execState of query, this is outside invoke");
+            LOG.info("cancel query {} because {}", connectContext.queryId, message);
             cancelInternal(reason);
         } finally {
             try {