feat:branch prediction optimization

apache · Sep 29, 2023 · 8265b51 · 8265b51
1 parent a7a55ff
commit 8265b51
Show file tree

Hide file tree

Showing 2 changed files with 63 additions and 56 deletions.
diff --git a/...a/org/apache/shardingsphere/proxy/frontend/mysql/command/MySQLCommandExecutorFactory.java b/...a/org/apache/shardingsphere/proxy/frontend/mysql/command/MySQLCommandExecutorFactory.java
@@ -49,6 +49,7 @@
 import org.apache.shardingsphere.proxy.frontend.mysql.command.query.text.query.MySQLComQueryPacketExecutor;
 
 import java.sql.SQLException;
+import java.util.Objects;
 
 /**
  * Command executor factory for MySQL.
@@ -73,33 +74,35 @@ public static CommandExecutor newInstance(final MySQLCommandPacketType commandPa
         } else {
             log.debug("Execute packet type: {}", commandPacketType);
         }
-        switch (commandPacketType) {
-            case COM_QUIT:
-                return new MySQLComQuitExecutor();
-            case COM_INIT_DB:
-                return new MySQLComInitDbExecutor((MySQLComInitDbPacket) commandPacket, connectionSession);
-            case COM_FIELD_LIST:
-                return new MySQLComFieldListPacketExecutor((MySQLComFieldListPacket) commandPacket, connectionSession);
-            case COM_QUERY:
-                return new MySQLComQueryPacketExecutor((MySQLComQueryPacket) commandPacket, connectionSession);
-            case COM_PING:
-                return new MySQLComPingExecutor(connectionSession);
-            case COM_STMT_PREPARE:
-                return new MySQLComStmtPrepareExecutor((MySQLComStmtPreparePacket) commandPacket, connectionSession);
-            case COM_STMT_EXECUTE:
-                return new MySQLComStmtExecuteExecutor((MySQLComStmtExecutePacket) commandPacket, connectionSession);
-            case COM_STMT_SEND_LONG_DATA:
-                return new MySQLComStmtSendLongDataExecutor((MySQLComStmtSendLongDataPacket) commandPacket, connectionSession);
-            case COM_STMT_RESET:
-                return new MySQLComStmtResetExecutor((MySQLComStmtResetPacket) commandPacket, connectionSession);
-            case COM_STMT_CLOSE:
-                return new MySQLComStmtCloseExecutor((MySQLComStmtClosePacket) commandPacket, connectionSession);
-            case COM_SET_OPTION:
-                return new MySQLComSetOptionExecutor((MySQLComSetOptionPacket) commandPacket, connectionSession);
-            case COM_RESET_CONNECTION:
-                return new MySQLComResetConnectionExecutor(connectionSession);
-            default:
-                return new MySQLUnsupportedCommandExecutor(commandPacketType);
+        if (Objects.equals(commandPacketType,MySQLCommandPacketType.COM_QUERY)){
+            return new MySQLComQueryPacketExecutor((MySQLComQueryPacket) commandPacket, connectionSession);
+        }else {
+            switch (commandPacketType) {
+                case COM_QUIT:
+                    return new MySQLComQuitExecutor();
+                case COM_INIT_DB:
+                    return new MySQLComInitDbExecutor((MySQLComInitDbPacket) commandPacket, connectionSession);
+                case COM_FIELD_LIST:
+                    return new MySQLComFieldListPacketExecutor((MySQLComFieldListPacket) commandPacket, connectionSession);
+                case COM_PING:
+                    return new MySQLComPingExecutor(connectionSession);
+                case COM_STMT_PREPARE:
+                    return new MySQLComStmtPrepareExecutor((MySQLComStmtPreparePacket) commandPacket, connectionSession);
+                case COM_STMT_EXECUTE:
+                    return new MySQLComStmtExecuteExecutor((MySQLComStmtExecutePacket) commandPacket, connectionSession);
+                case COM_STMT_SEND_LONG_DATA:
+                    return new MySQLComStmtSendLongDataExecutor((MySQLComStmtSendLongDataPacket) commandPacket, connectionSession);
+                case COM_STMT_RESET:
+                    return new MySQLComStmtResetExecutor((MySQLComStmtResetPacket) commandPacket, connectionSession);
+                case COM_STMT_CLOSE:
+                    return new MySQLComStmtCloseExecutor((MySQLComStmtClosePacket) commandPacket, connectionSession);
+                case COM_SET_OPTION:
+                    return new MySQLComSetOptionExecutor((MySQLComSetOptionPacket) commandPacket, connectionSession);
+                case COM_RESET_CONNECTION:
+                    return new MySQLComResetConnectionExecutor(connectionSession);
+                default:
+                    return new MySQLUnsupportedCommandExecutor(commandPacketType);
+            }
         }
     }
 }
diff --git a/...ava/org/apache/shardingsphere/proxy/frontend/mysql/command/MySQLCommandPacketFactory.java b/...ava/org/apache/shardingsphere/proxy/frontend/mysql/command/MySQLCommandPacketFactory.java
@@ -38,6 +38,8 @@
 import org.apache.shardingsphere.proxy.backend.session.ConnectionSession;
 import org.apache.shardingsphere.proxy.frontend.mysql.command.query.binary.MySQLServerPreparedStatement;
 
+import java.util.Objects;
+
 /**
  * Command packet factory for MySQL.
  */
@@ -55,35 +57,37 @@ public final class MySQLCommandPacketFactory {
      */
     public static MySQLCommandPacket newInstance(final MySQLCommandPacketType commandPacketType, final MySQLPacketPayload payload,
                                                  final ConnectionSession connectionSession, final boolean sqlCommentParseEnabled) {
-        switch (commandPacketType) {
-            case COM_QUIT:
-                return new MySQLComQuitPacket();
-            case COM_INIT_DB:
-                return new MySQLComInitDbPacket(payload);
-            case COM_FIELD_LIST:
-                return new MySQLComFieldListPacket(payload);
-            case COM_QUERY:
-                return new MySQLComQueryPacket(payload, sqlCommentParseEnabled);
-            case COM_STMT_PREPARE:
-                return new MySQLComStmtPreparePacket(payload, sqlCommentParseEnabled);
-            case COM_STMT_EXECUTE:
-                MySQLServerPreparedStatement serverPreparedStatement =
-                        connectionSession.getServerPreparedStatementRegistry().getPreparedStatement(payload.getByteBuf().getIntLE(payload.getByteBuf().readerIndex()));
-                return new MySQLComStmtExecutePacket(payload, serverPreparedStatement.getSqlStatementContext().getSqlStatement().getParameterCount());
-            case COM_STMT_SEND_LONG_DATA:
-                return new MySQLComStmtSendLongDataPacket(payload);
-            case COM_STMT_RESET:
-                return new MySQLComStmtResetPacket(payload);
-            case COM_STMT_CLOSE:
-                return new MySQLComStmtClosePacket(payload);
-            case COM_SET_OPTION:
-                return new MySQLComSetOptionPacket(payload);
-            case COM_PING:
-                return new MySQLComPingPacket();
-            case COM_RESET_CONNECTION:
-                return new MySQLComResetConnectionPacket();
-            default:
-                return new MySQLUnsupportedCommandPacket(commandPacketType);
+        if (Objects.equals(commandPacketType,MySQLCommandPacketType.COM_QUERY)){
+            return new MySQLComQueryPacket(payload, sqlCommentParseEnabled);
+        }else {
+            switch (commandPacketType) {
+                case COM_QUIT:
+                    return new MySQLComQuitPacket();
+                case COM_INIT_DB:
+                    return new MySQLComInitDbPacket(payload);
+                case COM_FIELD_LIST:
+                    return new MySQLComFieldListPacket(payload);
+                case COM_STMT_PREPARE:
+                    return new MySQLComStmtPreparePacket(payload, sqlCommentParseEnabled);
+                case COM_STMT_EXECUTE:
+                    MySQLServerPreparedStatement serverPreparedStatement =
+                            connectionSession.getServerPreparedStatementRegistry().getPreparedStatement(payload.getByteBuf().getIntLE(payload.getByteBuf().readerIndex()));
+                    return new MySQLComStmtExecutePacket(payload, serverPreparedStatement.getSqlStatementContext().getSqlStatement().getParameterCount());
+                case COM_STMT_SEND_LONG_DATA:
+                    return new MySQLComStmtSendLongDataPacket(payload);
+                case COM_STMT_RESET:
+                    return new MySQLComStmtResetPacket(payload);
+                case COM_STMT_CLOSE:
+                    return new MySQLComStmtClosePacket(payload);
+                case COM_SET_OPTION:
+                    return new MySQLComSetOptionPacket(payload);
+                case COM_PING:
+                    return new MySQLComPingPacket();
+                case COM_RESET_CONNECTION:
+                    return new MySQLComResetConnectionPacket();
+                default:
+                    return new MySQLUnsupportedCommandPacket(commandPacketType);
+            }
         }
     }
 }