Add SQLFederationUnsupportedSQLException to handle some case that sql…

… federation not support now

kernel/sql-federation/core/src/main/java/org/apache/shardingsphere/sqlfederation/engine/SQLFederationEngine.java

@@ -18,6 +18,7 @@
 package org.apache.shardingsphere.sqlfederation.engine;
 
 import lombok.Getter;
+import lombok.extern.slf4j.Slf4j;
 import org.apache.calcite.adapter.enumerable.EnumerableInterpretable;
 import org.apache.calcite.adapter.enumerable.EnumerableRel;
 import org.apache.calcite.linq4j.Enumerator;
@@ -52,6 +53,7 @@
 import org.apache.shardingsphere.sqlfederation.optimizer.SQLFederationExecutionPlan;
 import org.apache.shardingsphere.sqlfederation.optimizer.context.OptimizerContext;
 import org.apache.shardingsphere.sqlfederation.optimizer.context.planner.OptimizerPlannerContext;
+import org.apache.shardingsphere.sqlfederation.optimizer.exception.syntax.SQLFederationUnsupportedSQLException;
 import org.apache.shardingsphere.sqlfederation.optimizer.metadata.schema.SQLFederationTable;
 import org.apache.shardingsphere.sqlfederation.optimizer.planner.cache.ExecutionPlanCacheKey;
 import org.apache.shardingsphere.sqlfederation.optimizer.statement.SQLStatementCompiler;
@@ -73,6 +75,7 @@
 /**
  * SQL federation engine.
  */
+@Slf4j
 @Getter
 public final class SQLFederationEngine implements AutoCloseable {
 
@@ -155,14 +158,21 @@ private boolean isQuerySystemSchema(final SQLStatementContext sqlStatementContex
      * @param callback callback
      * @param federationContext federation context
      * @return result set
+     * @throws SQLFederationUnsupportedSQLException SQL federation unsupported SQL exception
      */
     public ResultSet executeQuery(final DriverExecutionPrepareEngine<JDBCExecutionUnit, Connection> prepareEngine,
                                   final JDBCExecutorCallback<? extends ExecuteResult> callback, final SQLFederationExecutorContext federationContext) {
-        String databaseName = federationContext.getQueryContext().getDatabaseNameFromSQLStatement().orElse(this.databaseName);
-        String schemaName = federationContext.getQueryContext().getSchemaNameFromSQLStatement().orElse(this.schemaName);
-        SQLFederationExecutionPlan executionPlan = compileQuery(prepareEngine, callback, federationContext, databaseName, schemaName);
-        resultSet = executePlan(federationContext, executionPlan, databaseName, schemaName);
-        return resultSet;
+        try {
+            String databaseName = federationContext.getQueryContext().getDatabaseNameFromSQLStatement().orElse(this.databaseName);
+            String schemaName = federationContext.getQueryContext().getSchemaNameFromSQLStatement().orElse(this.schemaName);
+            SQLFederationExecutionPlan executionPlan = compileQuery(prepareEngine, callback, federationContext, databaseName, schemaName);
+            resultSet = executePlan(federationContext, executionPlan, databaseName, schemaName);
+            return resultSet;
+            // CHECKSTYLE:OFF
+        } catch (final Exception ex) {
+            // CHECKSTYLE:ON
+            throw new SQLFederationUnsupportedSQLException(federationContext.getQueryContext().getSql());
+        }
     }
 
     private SQLFederationExecutionPlan compileQuery(final DriverExecutionPrepareEngine<JDBCExecutionUnit, Connection> prepareEngine,

kernel/sql-federation/core/src/main/java/org/apache/shardingsphere/sqlfederation/resultset/SQLFederationResultSet.java

@@ -97,7 +97,7 @@ private void handleColumnLabelAndIndex(final Map<String, Integer> columnLabelAnd
     @Override
     public boolean next() {
         boolean result = enumerator.moveNext();
-        if (result) {
+        if (result && null != enumerator.current()) {
             currentRows = enumerator.current().getClass().isArray() ? (Object[]) enumerator.current() : new Object[]{enumerator.current()};
         } else {
             currentRows = new Object[]{};

kernel/sql-federation/optimizer/src/main/java/org/apache/shardingsphere/sqlfederation/optimizer/converter/SQLNodeConverterEngine.java

@@ -35,7 +35,7 @@
 import org.apache.shardingsphere.sqlfederation.optimizer.converter.statement.merge.MergeStatementConverter;
 import org.apache.shardingsphere.sqlfederation.optimizer.converter.statement.select.SelectStatementConverter;
 import org.apache.shardingsphere.sqlfederation.optimizer.converter.statement.update.UpdateStatementConverter;
-import org.apache.shardingsphere.sqlfederation.optimizer.exception.OptimizationSQLNodeConvertException;
+import org.apache.shardingsphere.sqlfederation.optimizer.exception.convert.OptimizationSQLNodeConvertException;
 
 import java.util.Optional;
 

kernel/sql-federation/optimizer/src/main/java/org/apache/shardingsphere/sqlfederation/optimizer/exception/SQLFederationSQLException.java

@@ -0,0 +1,35 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.shardingsphere.sqlfederation.optimizer.exception;
+
+import org.apache.shardingsphere.infra.exception.core.external.sql.sqlstate.SQLState;
+import org.apache.shardingsphere.infra.exception.core.external.sql.type.feature.FeatureSQLException;
+
+/**
+ * SQL federation SQL exception.
+ */
+public abstract class SQLFederationSQLException extends FeatureSQLException {
+
+    private static final long serialVersionUID = 4689889693356895996L;
+
+    private static final int FEATURE_CODE = 20;
+
+    protected SQLFederationSQLException(final SQLState sqlState, final int errorCode, final String reason, final Object... messageArgs) {
+        super(sqlState, FEATURE_CODE, errorCode, reason, messageArgs);
+    }
+}

kernel/sql-federation/optimizer/src/main/java/org/apache/shardingsphere/sqlfederation/optimizer/exception/OptimizationSQLNodeConvertException.java → kernel/sql-federation/optimizer/src/main/java/org/apache/shardingsphere/sqlfederation/optimizer/exception/convert/OptimizationSQLNodeConvertException.java

@@ -15,18 +15,18 @@
  * limitations under the License.
  */
 
-package org.apache.shardingsphere.sqlfederation.optimizer.exception;
+package org.apache.shardingsphere.sqlfederation.optimizer.exception.convert;
 
 import org.apache.shardingsphere.infra.exception.core.external.sql.sqlstate.XOpenSQLState;
-import org.apache.shardingsphere.infra.exception.core.external.sql.type.kernel.category.MetaDataSQLException;
 import org.apache.shardingsphere.sql.parser.sql.common.statement.SQLStatement;
+import org.apache.shardingsphere.sqlfederation.optimizer.exception.SQLFederationSQLException;
 
 /**
  * Optimization SQL node convert exception.
  */
-public final class OptimizationSQLNodeConvertException extends MetaDataSQLException {
+public final class OptimizationSQLNodeConvertException extends SQLFederationSQLException {
 
-    private static final long serialVersionUID = -5486229929620713984L;
+    private static final long serialVersionUID = 7115939407266382363L;
 
     public OptimizationSQLNodeConvertException(final SQLStatement statement) {
         super(XOpenSQLState.SYNTAX_ERROR, 4, "Unsupported SQL node conversion for SQL statement `%s`.", statement.toString());

kernel/sql-federation/optimizer/src/main/java/org/apache/shardingsphere/sqlfederation/optimizer/exception/syntax/SQLFederationUnsupportedSQLException.java

@@ -0,0 +1,33 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.shardingsphere.sqlfederation.optimizer.exception.syntax;
+
+import org.apache.shardingsphere.infra.exception.core.external.sql.sqlstate.XOpenSQLState;
+import org.apache.shardingsphere.sqlfederation.optimizer.exception.SQLFederationSQLException;
+
+/**
+ * SQL federation unsupported SQL exception.
+ */
+public final class SQLFederationUnsupportedSQLException extends SQLFederationSQLException {
+
+    private static final long serialVersionUID = -8571244162760408846L;
+
+    public SQLFederationUnsupportedSQLException(final String sql) {
+        super(XOpenSQLState.SYNTAX_ERROR, 41, "SQL federation doesn't support SQL %s execution.", sql);
+    }
+}

parser/sql/dialect/mysql/src/main/antlr4/org/apache/shardingsphere/sql/parser/autogen/MySQLStatement.g4

@@ -126,6 +126,5 @@ execute
     | alterTablespace
     | dropTablespace
     | delimiter
-    ) (SEMI_ EOF? | EOF)
-    | EOF
+    ) SEMI_? EOF
     ;

test/it/parser/src/main/resources/sql/unsupported/unsupported.xml

@@ -319,9 +319,10 @@
     <sql-case id="unsupported_select_case_for_opengauss_578" value="select string_agg(name, #) from table_test;" db-types="openGauss" />
     <sql-case id="unsupported_select_case_for_opengauss_579" value="select bool_or(array[1.1,2.1,3.1]::int[] =[1,2,3]);" db-types="openGauss" />
     <sql-case id="unsupported_select_case_for_opengauss_580" value="select bool_or(@);" db-types="openGauss" />
-    <sql-case id="assertDistSQLShowRuleParseConflict" value="SHOW READWRITE_SPLITTING RULE FROM schema_name" db-types="PostgreSQL,openGauss" />
-    <sql-case id="assertDistSQLRollbackMigration" value="ROLLBACK MIGRATION 10102p0000697d5ccb2e3d960d0gif75c7c7f486fal" db-types="MySQL,PostgreSQL,openGauss,Oracle" />
-    <sql-case id="assertShowDatabasesParseConflict" value="SHOW DATABASES" db-types="Oracle" />
-    <sql-case id="assertShowSchemasParseConflict" value="SHOW SCHEMAS" db-types="Oracle" />
-    <sql-case id="assertShowTablesParseConflict" value="SHOW TABLES" db-types="Oracle" />
+    <sql-case id="assert_distsql_show_rule_parse_conflict" value="SHOW READWRITE_SPLITTING RULE FROM schema_name" db-types="PostgreSQL,openGauss" />
+    <sql-case id="assert_distsql_rollback_migration" value="ROLLBACK MIGRATION 10102p0000697d5ccb2e3d960d0gif75c7c7f486fal" db-types="MySQL,PostgreSQL,openGauss,Oracle" />
+    <sql-case id="assert_show_databases_parse_conflict" value="SHOW DATABASES" db-types="Oracle" />
+    <sql-case id="assert_show_schemas_parse_conflict" value="SHOW SCHEMAS" db-types="Oracle" />
+    <sql-case id="assert_show_tables_parse_conflict" value="SHOW TABLES" db-types="Oracle" />
+    <sql-case id="assert_parse_multi_statements" value="SHOW VARIABLES LIKE 'sql_mode'; SELECT COUNT(*) AS support_ndb FROM information_schema;" db-types="MySQL" />
 </sql-cases>