Fix rules order on DataNodes

features/broadcast/core/src/main/java/org/apache/shardingsphere/broadcast/rule/BroadcastRule.java

@@ -20,6 +20,7 @@
 import com.cedarsoftware.util.CaseInsensitiveSet;
 import lombok.Getter;
 import org.apache.shardingsphere.broadcast.config.BroadcastRuleConfiguration;
+import org.apache.shardingsphere.broadcast.constant.BroadcastOrder;
 import org.apache.shardingsphere.broadcast.rule.attribute.BroadcastDataNodeRuleAttribute;
 import org.apache.shardingsphere.broadcast.rule.attribute.BroadcastTableNamesRuleAttribute;
 import org.apache.shardingsphere.infra.rule.ShardingSphereRule;
@@ -102,4 +103,9 @@ public Collection<String> filterBroadcastTableNames(final Collection<String> log
     public boolean isAllBroadcastTables(final Collection<String> logicTableNames) {
         return !logicTableNames.isEmpty() && tables.containsAll(logicTableNames);
     }
+
+    @Override
+    public int getOrder() {
+        return BroadcastOrder.ORDER;
+    }
 }

features/encrypt/core/src/main/java/org/apache/shardingsphere/encrypt/rule/EncryptRule.java

@@ -23,6 +23,7 @@
 import org.apache.shardingsphere.encrypt.config.EncryptRuleConfiguration;
 import org.apache.shardingsphere.encrypt.config.rule.EncryptColumnRuleConfiguration;
 import org.apache.shardingsphere.encrypt.config.rule.EncryptTableRuleConfiguration;
+import org.apache.shardingsphere.encrypt.constant.EncryptOrder;
 import org.apache.shardingsphere.encrypt.exception.metadata.EncryptTableNotFoundException;
 import org.apache.shardingsphere.encrypt.exception.metadata.MismatchedEncryptAlgorithmTypeException;
 import org.apache.shardingsphere.encrypt.rule.attribute.EncryptTableMapperRuleAttribute;
@@ -189,4 +190,9 @@ private EncryptTableRuleConfiguration getTableRuleConfiguration(final String tab
         Preconditions.checkState(result.isPresent());
         return result.get();
     }
+
+    @Override
+    public int getOrder() {
+        return EncryptOrder.ORDER;
+    }
 }

features/mask/core/src/main/java/org/apache/shardingsphere/mask/rule/MaskRule.java

@@ -27,6 +27,7 @@
 import org.apache.shardingsphere.infra.spi.type.typed.TypedSPILoader;
 import org.apache.shardingsphere.mask.config.MaskRuleConfiguration;
 import org.apache.shardingsphere.mask.config.rule.MaskTableRuleConfiguration;
+import org.apache.shardingsphere.mask.constant.MaskOrder;
 import org.apache.shardingsphere.mask.rule.attribute.MaskTableMapperRuleAttribute;
 import org.apache.shardingsphere.mask.spi.MaskAlgorithm;
 
@@ -118,4 +119,9 @@ private MaskTableRuleConfiguration getTableRuleConfiguration(final String tableN
         Preconditions.checkState(result.isPresent());
         return result.get();
     }
+
+    @Override
+    public int getOrder() {
+        return MaskOrder.ORDER;
+    }
 }

features/readwrite-splitting/core/src/main/java/org/apache/shardingsphere/readwritesplitting/rule/ReadwriteSplittingRule.java

@@ -29,6 +29,7 @@
 import org.apache.shardingsphere.readwritesplitting.config.ReadwriteSplittingRuleConfiguration;
 import org.apache.shardingsphere.readwritesplitting.config.rule.ReadwriteSplittingDataSourceGroupRuleConfiguration;
 import org.apache.shardingsphere.readwritesplitting.constant.ReadwriteSplittingDataSourceType;
+import org.apache.shardingsphere.readwritesplitting.constant.ReadwriteSplittingOrder;
 import org.apache.shardingsphere.readwritesplitting.exception.ReadwriteSplittingRuleExceptionIdentifier;
 import org.apache.shardingsphere.readwritesplitting.exception.actual.InvalidReadwriteSplittingActualDataSourceInlineExpressionException;
 import org.apache.shardingsphere.readwritesplitting.rule.attribute.ReadwriteSplittingDataSourceMapperRuleAttribute;
@@ -139,4 +140,9 @@ public ReadwriteSplittingDataSourceGroupRule getSingleDataSourceGroupRule() {
     public Optional<ReadwriteSplittingDataSourceGroupRule> findDataSourceGroupRule(final String dataSourceName) {
         return Optional.ofNullable(dataSourceRuleGroups.get(dataSourceName));
     }
+
+    @Override
+    public int getOrder() {
+        return ReadwriteSplittingOrder.ORDER;
+    }
 }

features/shadow/core/src/main/java/org/apache/shardingsphere/shadow/rule/ShadowRule.java

@@ -20,17 +20,18 @@
 import lombok.Getter;
 import org.apache.shardingsphere.infra.algorithm.core.config.AlgorithmConfiguration;
 import org.apache.shardingsphere.infra.annotation.HighFrequencyInvocation;
-import org.apache.shardingsphere.infra.rule.scope.DatabaseRule;
 import org.apache.shardingsphere.infra.rule.attribute.RuleAttributes;
+import org.apache.shardingsphere.infra.rule.scope.DatabaseRule;
 import org.apache.shardingsphere.infra.spi.type.typed.TypedSPILoader;
 import org.apache.shardingsphere.shadow.config.ShadowRuleConfiguration;
 import org.apache.shardingsphere.shadow.config.datasource.ShadowDataSourceConfiguration;
 import org.apache.shardingsphere.shadow.config.table.ShadowTableConfiguration;
+import org.apache.shardingsphere.shadow.constant.ShadowOrder;
+import org.apache.shardingsphere.shadow.rule.attribute.ShadowDataSourceMapperRuleAttribute;
+import org.apache.shardingsphere.shadow.spi.ShadowAlgorithm;
 import org.apache.shardingsphere.shadow.spi.ShadowOperationType;
 import org.apache.shardingsphere.shadow.spi.column.ColumnShadowAlgorithm;
 import org.apache.shardingsphere.shadow.spi.hint.HintShadowAlgorithm;
-import org.apache.shardingsphere.shadow.rule.attribute.ShadowDataSourceMapperRuleAttribute;
-import org.apache.shardingsphere.shadow.spi.ShadowAlgorithm;
 
 import java.util.Collection;
 import java.util.Collections;
@@ -242,4 +243,9 @@ public Optional<String> findProductionDataSourceName(final String logicDataSourc
         ShadowDataSourceRule dataSourceRule = dataSourceRules.get(logicDataSourceName);
         return null == dataSourceRule ? Optional.empty() : Optional.of(dataSourceRule.getProductionDataSource());
     }
+
+    @Override
+    public int getOrder() {
+        return ShadowOrder.ORDER;
+    }
 }

features/sharding/core/src/main/java/org/apache/shardingsphere/sharding/rule/ShardingRule.java

@@ -49,6 +49,7 @@
 import org.apache.shardingsphere.sharding.api.config.strategy.sharding.StandardShardingStrategyConfiguration;
 import org.apache.shardingsphere.sharding.api.sharding.ShardingAutoTableAlgorithm;
 import org.apache.shardingsphere.sharding.cache.ShardingCache;
+import org.apache.shardingsphere.sharding.constant.ShardingOrder;
 import org.apache.shardingsphere.sharding.exception.metadata.ShardingTableRuleNotFoundException;
 import org.apache.shardingsphere.sharding.rule.attribute.ShardingDataNodeRuleAttribute;
 import org.apache.shardingsphere.sharding.rule.attribute.ShardingTableNamesRuleAttribute;
@@ -634,4 +635,9 @@ private boolean isJoinConditionExpression(final ExpressionSegment expression) {
         BinaryOperationExpression binaryExpression = (BinaryOperationExpression) expression;
         return binaryExpression.getLeft() instanceof ColumnSegment && binaryExpression.getRight() instanceof ColumnSegment && "=".equals(binaryExpression.getOperator());
     }
+
+    @Override
+    public int getOrder() {
+        return ShardingOrder.ORDER;
+    }
 }

infra/common/src/main/java/org/apache/shardingsphere/infra/datanode/DataNodes.java

@@ -23,9 +23,12 @@
 import org.apache.shardingsphere.infra.rule.attribute.datanode.DataNodeRuleAttribute;
 import org.apache.shardingsphere.infra.rule.attribute.datasource.DataSourceMapperRuleAttribute;
 
+import java.util.ArrayList;
 import java.util.Collection;
 import java.util.Collections;
+import java.util.Comparator;
 import java.util.LinkedList;
+import java.util.List;
 import java.util.Map;
 import java.util.Optional;
 
@@ -49,7 +52,7 @@ public Collection<DataNode> getDataNodes(final String tableName) {
         if (result.isEmpty()) {
             return result;
         }
-        for (ShardingSphereRule each : rules) {
+        for (ShardingSphereRule each : getOrderedRules()) {
             Optional<DataSourceMapperRuleAttribute> dataSourceMapperRuleAttribute = each.getAttributes().findAttribute(DataSourceMapperRuleAttribute.class);
             if (dataSourceMapperRuleAttribute.isPresent()) {
                 result = buildDataNodes(result, dataSourceMapperRuleAttribute.get());
@@ -72,6 +75,12 @@ private Collection<DataNode> getDataNodesByTableName(final ShardingSphereRule ru
         return rule.getAttributes().findAttribute(DataNodeRuleAttribute.class).map(optional -> optional.getDataNodesByTableName(tableName)).orElse(Collections.emptyList());
     }
 
+    private Collection<ShardingSphereRule> getOrderedRules() {
+        List<ShardingSphereRule> result = new ArrayList<>(rules);
+        result.sort(Comparator.comparingInt(ShardingSphereRule::getOrder));
+        return result;
+    }
+
     private Collection<DataNode> buildDataNodes(final Collection<DataNode> dataNodes, final DataSourceMapperRuleAttribute dataSourceMapperRuleAttribute) {
         Collection<DataNode> result = new LinkedList<>();
         Map<String, Collection<String>> dataSourceMapper = dataSourceMapperRuleAttribute.getDataSourceMapper();

infra/common/src/main/java/org/apache/shardingsphere/infra/rule/ShardingSphereRule.java

@@ -40,4 +40,11 @@ public interface ShardingSphereRule {
     default RuleAttributes getAttributes() {
         return new RuleAttributes();
     }
+
+    /**
+     * Get rule order.
+     *
+     * @return rule order
+     */
+    int getOrder();
 }

infra/common/src/test/java/org/apache/shardingsphere/infra/fixture/FixtureRule.java

@@ -35,4 +35,9 @@ public RuleConfiguration getConfiguration() {
     public RuleAttributes getAttributes() {
         return new RuleAttributes(mock(DataSourceMapperRuleAttribute.class));
     }
+
+    @Override
+    public int getOrder() {
+        return 0;
+    }
 }

infra/common/src/test/java/org/apache/shardingsphere/infra/metadata/database/rule/RuleMetaDataShardingSphereRuleFixture.java

@@ -28,4 +28,9 @@ public class RuleMetaDataShardingSphereRuleFixture implements ShardingSphereRule
     public RuleConfiguration getConfiguration() {
         return mock(RuleConfiguration.class);
     }
+
+    @Override
+    public int getOrder() {
+        return 0;
+    }
 }

infra/common/src/test/java/org/apache/shardingsphere/infra/rule/builder/fixture/FixtureGlobalRule.java

@@ -28,4 +28,9 @@ public final class FixtureGlobalRule implements GlobalRule {
     public RuleConfiguration getConfiguration() {
         return mock(RuleConfiguration.class);
     }
+
+    @Override
+    public int getOrder() {
+        return 0;
+    }
 }

infra/distsql-handler/src/test/java/org/apache/shardingsphere/distsql/handler/fixture/DistSQLHandlerFixtureRule.java

@@ -52,4 +52,9 @@ private DataNodeRuleAttribute getDataNodeRuleAttribute() {
         when(result.findActualTableByCatalog(any(), any())).thenReturn(Optional.empty());
         return result;
     }
+
+    @Override
+    public int getOrder() {
+        return 0;
+    }
 }

infra/merge/src/test/java/org/apache/shardingsphere/infra/merge/fixture/rule/DecoratorRuleFixture.java

@@ -28,4 +28,9 @@ public final class DecoratorRuleFixture implements ShardingSphereRule {
     public RuleConfiguration getConfiguration() {
         return mock(RuleConfiguration.class);
     }
+
+    @Override
+    public int getOrder() {
+        return 0;
+    }
 }

infra/merge/src/test/java/org/apache/shardingsphere/infra/merge/fixture/rule/MergerRuleFixture.java

@@ -28,4 +28,9 @@ public final class MergerRuleFixture implements ShardingSphereRule {
     public RuleConfiguration getConfiguration() {
         return mock(RuleConfiguration.class);
     }
+
+    @Override
+    public int getOrder() {
+        return 0;
+    }
 }

infra/route/src/test/java/org/apache/shardingsphere/infra/route/fixture/rule/RouteFailureRuleFixture.java

@@ -28,4 +28,9 @@ public final class RouteFailureRuleFixture implements ShardingSphereRule {
     public RuleConfiguration getConfiguration() {
         return mock(RuleConfiguration.class);
     }
+
+    @Override
+    public int getOrder() {
+        return 0;
+    }
 }

infra/route/src/test/java/org/apache/shardingsphere/infra/route/fixture/rule/RouteRuleFixture.java

@@ -28,4 +28,9 @@ public final class RouteRuleFixture implements ShardingSphereRule {
     public RuleConfiguration getConfiguration() {
         return mock(RuleConfiguration.class);
     }
+
+    @Override
+    public int getOrder() {
+        return 0;
+    }
 }

kernel/authority/core/src/main/java/org/apache/shardingsphere/authority/rule/AuthorityRule.java

@@ -19,6 +19,7 @@
 
 import lombok.Getter;
 import org.apache.shardingsphere.authority.config.AuthorityRuleConfiguration;
+import org.apache.shardingsphere.authority.constant.AuthorityOrder;
 import org.apache.shardingsphere.authority.model.ShardingSpherePrivileges;
 import org.apache.shardingsphere.authority.spi.PrivilegeProvider;
 import org.apache.shardingsphere.infra.annotation.HighFrequencyInvocation;
@@ -109,4 +110,9 @@ public Optional<ShardingSpherePrivileges> findPrivileges(final Grantee grantee)
         }
         return Optional.empty();
     }
+
+    @Override
+    public int getOrder() {
+        return AuthorityOrder.ORDER;
+    }
 }

kernel/global-clock/core/src/main/java/org/apache/shardingsphere/globalclock/rule/GlobalClockRule.java

@@ -20,6 +20,7 @@
 import lombok.Getter;
 import org.apache.shardingsphere.globalclock.config.GlobalClockRuleConfiguration;
 import org.apache.shardingsphere.globalclock.provider.GlobalClockProvider;
+import org.apache.shardingsphere.globalclock.rule.constant.GlobalClockOrder;
 import org.apache.shardingsphere.infra.rule.scope.GlobalRule;
 import org.apache.shardingsphere.infra.spi.type.typed.TypedSPILoader;
 
@@ -50,4 +51,9 @@ public GlobalClockRule(final GlobalClockRuleConfiguration ruleConfig) {
     public Optional<GlobalClockProvider> getGlobalClockProvider() {
         return Optional.ofNullable(globalClockProvider);
     }
+
+    @Override
+    public int getOrder() {
+        return GlobalClockOrder.ORDER;
+    }
 }

kernel/logging/core/src/main/java/org/apache/shardingsphere/logging/rule/LoggingRule.java

@@ -21,6 +21,7 @@
 import org.apache.shardingsphere.infra.rule.scope.GlobalRule;
 import org.apache.shardingsphere.logging.config.LoggingRuleConfiguration;
 import org.apache.shardingsphere.logging.constant.LoggingConstants;
+import org.apache.shardingsphere.logging.constant.LoggingOrder;
 import org.apache.shardingsphere.logging.logger.ShardingSphereLogger;
 
 import java.util.Optional;
@@ -45,4 +46,9 @@ public LoggingRule(final LoggingRuleConfiguration ruleConfig) {
     public Optional<ShardingSphereLogger> getSQLLogger() {
         return configuration.getLoggers().stream().filter(each -> LoggingConstants.SQL_LOG_TOPIC.equalsIgnoreCase(each.getLoggerName())).findFirst();
     }
+
+    @Override
+    public int getOrder() {
+        return LoggingOrder.ORDER;
+    }
 }

kernel/single/core/src/main/java/org/apache/shardingsphere/single/rule/SingleRule.java

@@ -33,6 +33,7 @@
 import org.apache.shardingsphere.infra.rule.attribute.RuleAttributes;
 import org.apache.shardingsphere.infra.rule.scope.DatabaseRule;
 import org.apache.shardingsphere.single.config.SingleRuleConfiguration;
+import org.apache.shardingsphere.single.constant.SingleOrder;
 import org.apache.shardingsphere.single.datanode.SingleTableDataNodeLoader;
 import org.apache.shardingsphere.sql.parser.statement.core.segment.generic.table.SimpleTableSegment;
 
@@ -186,4 +187,9 @@ private Collection<QualifiedTable> getQualifiedTables(final ShardingSphereDataba
         }
         return result;
     }
+
+    @Override
+    public int getOrder() {
+        return SingleOrder.ORDER;
+    }
 }

kernel/sql-federation/core/src/main/java/org/apache/shardingsphere/sqlfederation/rule/SQLFederationRule.java

@@ -21,6 +21,7 @@
 import org.apache.shardingsphere.infra.metadata.database.ShardingSphereDatabase;
 import org.apache.shardingsphere.infra.rule.scope.GlobalRule;
 import org.apache.shardingsphere.sqlfederation.config.SQLFederationRuleConfiguration;
+import org.apache.shardingsphere.sqlfederation.constant.SQLFederationOrder;
 import org.apache.shardingsphere.sqlfederation.optimizer.context.OptimizerContext;
 import org.apache.shardingsphere.sqlfederation.optimizer.context.OptimizerContextFactory;
 
@@ -55,4 +56,9 @@ public void refresh(final Map<String, ShardingSphereDatabase> databases, final G
     public OptimizerContext getOptimizerContext() {
         return optimizerContext.get();
     }
+
+    @Override
+    public int getOrder() {
+        return SQLFederationOrder.ORDER;
+    }
 }

kernel/sql-federation/core/src/test/java/org/apache/shardingsphere/sqlfederation/engine/fixture/rule/SQLFederationDeciderRuleMatchFixture.java

@@ -30,4 +30,9 @@ public final class SQLFederationDeciderRuleMatchFixture implements ShardingSpher
     public RuleConfiguration getConfiguration() {
         return new SQLFederationRuleConfiguration(true, false, mock(CacheOption.class));
     }
+
+    @Override
+    public int getOrder() {
+        return 0;
+    }
 }

kernel/sql-federation/core/src/test/java/org/apache/shardingsphere/sqlfederation/engine/fixture/rule/SQLFederationDeciderRuleNotMatchFixture.java

@@ -30,4 +30,9 @@ public final class SQLFederationDeciderRuleNotMatchFixture implements ShardingSp
     public RuleConfiguration getConfiguration() {
         return new SQLFederationRuleConfiguration(false, false, mock(CacheOption.class));
     }
+
+    @Override
+    public int getOrder() {
+        return 0;
+    }
 }

kernel/sql-parser/core/src/main/java/org/apache/shardingsphere/parser/rule/SQLParserRule.java

@@ -23,6 +23,7 @@
 import org.apache.shardingsphere.infra.parser.ShardingSphereSQLParserEngine;
 import org.apache.shardingsphere.infra.rule.scope.GlobalRule;
 import org.apache.shardingsphere.parser.config.SQLParserRuleConfiguration;
+import org.apache.shardingsphere.parser.constant.SQLParserOrder;
 import org.apache.shardingsphere.sql.parser.api.CacheOption;
 
 /**
@@ -52,4 +53,9 @@ public SQLParserRule(final SQLParserRuleConfiguration ruleConfig) {
     public SQLParserEngine getSQLParserEngine(final DatabaseType databaseType) {
         return new ShardingSphereSQLParserEngine(databaseType, sqlStatementCache, parseTreeCache);
     }
+
+    @Override
+    public int getOrder() {
+        return SQLParserOrder.ORDER;
+    }
 }

kernel/sql-translator/core/src/main/java/org/apache/shardingsphere/sqltranslator/rule/SQLTranslatorRule.java

@@ -25,6 +25,7 @@
 import org.apache.shardingsphere.infra.session.query.QueryContext;
 import org.apache.shardingsphere.infra.spi.type.typed.TypedSPILoader;
 import org.apache.shardingsphere.sqltranslator.config.SQLTranslatorRuleConfiguration;
+import org.apache.shardingsphere.sqltranslator.constant.SQLTranslatorOrder;
 import org.apache.shardingsphere.sqltranslator.context.SQLTranslatorContext;
 import org.apache.shardingsphere.sqltranslator.exception.SQLTranslationException;
 import org.apache.shardingsphere.sqltranslator.spi.SQLTranslator;
@@ -75,4 +76,9 @@ public SQLTranslatorContext translate(final String sql, final List<Object> param
             throw ex;
         }
     }
+
+    @Override
+    public int getOrder() {
+        return SQLTranslatorOrder.ORDER;
+    }
 }