关于complex分片策略的疑惑

Question

shardingjdbc如果分片策略为complex，设置uid、bid为分片键，如果用户执行的语句条件只包含uid而不包含bid，那它会进入complex算法类执行分片逻辑吗

Answer 1

引自免费微信小程序：皆我百晓生

在ShardingSphere（原Sharding-JDBC）框架中，当使用complex分片策略时，具体分片逻辑由用户自定义的ComplexShardingAlgorithm实现类决定。如果SQL查询仅包含uid作为条件，而未包含bid，那么是否执行complex分片算法取决于该算法类的实现。

1. complex分片策略概述

complex分片策略允许基于复杂的业务逻辑进行数据分片。用户需提供一个实现ComplexShardingAlgorithm接口的类，该接口有两个方法需要实现：

• doEqualSharding：用于处理等值查询。
• doInSharding 或 doBetweenSharding：用于处理IN或范围查询。

2. 分片键与SQL查询

当SQL查询中包含多个分片键时，ShardingSphere会尝试使用所有分片键进行分片。但是，如果查询中仅包含部分分片键，如本例中的uid，则ShardingSphere会根据定义的复杂分片算法来判断如何处理。

3. 处理单个分片键的查询

对于仅包含uid的查询，如果ComplexShardingAlgorithm的实现能够处理单个分片键的情况，那么它将被调用来确定数据应从哪些分片中读取。这意味着在doEqualSharding、doInSharding或doBetweenSharding方法中，你需要考虑到即使只有一个分片键存在的情况。

4. 示例代码

下面是一个示例实现，展示如何在ComplexShardingAlgorithm中处理单个分片键的情况：

import org.apache.shardingsphere.api.sharding.complex.ComplexKeysShardingValue;
import org.apache.shardingsphere.api.sharding.complex.ComplexShardingAlgorithm;
import org.apache.shardingsphere.api.sharding.complex.ComplexShardingCondition;

import java.util.Collection;
import java.util.LinkedHashSet;
import java.util.Map;

public class CustomComplexShardingAlgorithm implements ComplexShardingAlgorithm {

    @Override
    public Collection<String> doSharding(Collection<String> availableTargetNames, ComplexKeysShardingValue shardingValue) {
        Collection<String> result = new LinkedHashSet<>();
        Map<String, Collection<Comparable<?>>> columnRouteValues = shardingValue.getColumnNameAndShardingValues();
        
        // 假设我们只关注uid
        Collection<Comparable<?>> uidValues = columnRouteValues.get("uid");
        if (uidValues != null && !uidValues.isEmpty()) {
            // 这里可以是任何复杂的逻辑，例如基于uid的哈希算法选择分片
            for (Comparable<?> uid : uidValues) {
                String logicTableName = "t_user"; // 逻辑表名
                int shardId = uid.hashCode() % 2; // 简化示例，实际可能更复杂
                result.add(logicTableName + "_db" + shardId);
            }
        }
        return result;
    }
}

5. 结论

简而言之，即使查询中只包含部分分片键，如uid，只要你的ComplexShardingAlgorithm实现能够处理这种情况，那么ShardingSphere就会调用你的算法来进行分片。因此，确保你的算法足够灵活，能够适应各种查询场景是关键。