This is an automated email from the ASF dual-hosted git repository.
lijibing pushed a commit to branch branch-3.0
in repository https://gitbox.apache.org/repos/asf/doris.git
The following commit(s) were added to refs/heads/branch-3.0 by this push:
new 7780dda2fdc branch-3.0: [improvement](mysql)Support mysql
COM_RESET_CONNECTION command. #44747 (#44787)
7780dda2fdc is described below
commit 7780dda2fdc268ee586e5a3d539072db64ffd649
Author: github-actions[bot]
<41898282+github-actions[bot]@users.noreply.github.com>
AuthorDate: Mon Dec 2 10:21:43 2024 +0800
branch-3.0: [improvement](mysql)Support mysql COM_RESET_CONNECTION command.
#44747 (#44787)
Cherry-picked from #44747
Co-authored-by: James <[email protected]>
---
fe/fe-core/src/main/java/org/apache/doris/qe/ConnectContext.java | 5 +++++
fe/fe-core/src/main/java/org/apache/doris/qe/ConnectProcessor.java | 7 +++++++
.../src/main/java/org/apache/doris/qe/MysqlConnectProcessor.java | 3 +++
3 files changed, 15 insertions(+)
diff --git a/fe/fe-core/src/main/java/org/apache/doris/qe/ConnectContext.java
b/fe/fe-core/src/main/java/org/apache/doris/qe/ConnectContext.java
index 776457d5531..0f140ed5bdd 100644
--- a/fe/fe-core/src/main/java/org/apache/doris/qe/ConnectContext.java
+++ b/fe/fe-core/src/main/java/org/apache/doris/qe/ConnectContext.java
@@ -346,6 +346,11 @@ public class ConnectContext {
return lastDBOfCatalog.get(catalog);
}
+ // Used by COM_RESET_CONNECTION
+ public void clearLastDBOfCatalog() {
+ lastDBOfCatalog.clear();
+ }
+
public void setNotEvalNondeterministicFunction(boolean
notEvalNondeterministicFunction) {
this.notEvalNondeterministicFunction = notEvalNondeterministicFunction;
}
diff --git a/fe/fe-core/src/main/java/org/apache/doris/qe/ConnectProcessor.java
b/fe/fe-core/src/main/java/org/apache/doris/qe/ConnectProcessor.java
index 2340aa37aeb..fcc6c2362cf 100644
--- a/fe/fe-core/src/main/java/org/apache/doris/qe/ConnectProcessor.java
+++ b/fe/fe-core/src/main/java/org/apache/doris/qe/ConnectProcessor.java
@@ -44,6 +44,7 @@ import org.apache.doris.common.util.DebugUtil;
import org.apache.doris.common.util.SqlUtils;
import org.apache.doris.common.util.Util;
import org.apache.doris.datasource.CatalogIf;
+import org.apache.doris.datasource.InternalCatalog;
import org.apache.doris.metric.MetricRepo;
import org.apache.doris.mysql.MysqlChannel;
import org.apache.doris.mysql.MysqlPacket;
@@ -198,6 +199,12 @@ public abstract class ConnectProcessor {
ctx.getState().setOk();
}
+ protected void handleResetConnection() {
+ ctx.changeDefaultCatalog(InternalCatalog.INTERNAL_CATALOG_NAME);
+ ctx.clearLastDBOfCatalog();
+ ctx.getState().setOk();
+ }
+
protected void handleStmtReset() {
ctx.getState().setOk();
}
diff --git
a/fe/fe-core/src/main/java/org/apache/doris/qe/MysqlConnectProcessor.java
b/fe/fe-core/src/main/java/org/apache/doris/qe/MysqlConnectProcessor.java
index 376d4740e63..da3baced5fa 100644
--- a/fe/fe-core/src/main/java/org/apache/doris/qe/MysqlConnectProcessor.java
+++ b/fe/fe-core/src/main/java/org/apache/doris/qe/MysqlConnectProcessor.java
@@ -264,6 +264,9 @@ public class MysqlConnectProcessor extends ConnectProcessor
{
case COM_SET_OPTION:
handleSetOption();
break;
+ case COM_RESET_CONNECTION:
+ handleResetConnection();
+ break;
default:
ctx.getState().setError(ErrorCode.ERR_UNKNOWN_COM_ERROR,
"Unsupported command(" + command + ")");
LOG.warn("Unsupported command(" + command + ")");
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
