shardingsphere 自定义分片策略

1、编写自定义分片类 采用精确分片算法实现PreciseShardingAlgorithm

package com.spring.config;

import org.apache.shardingsphere.api.sharding.standard.PreciseShardingAlgorithm;
import org.apache.shardingsphere.api.sharding.standard.PreciseShardingValue;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;

import java.util.Collection;

@Component
public class UserAlgorithm implements PreciseShardingAlgorithm<String> {


    @Override
    public String doSharding(Collection<String> tableNames, PreciseShardingValue<String> preciseShardingValue) {
        if(!StringUtils.isEmpty(preciseShardingValue)){
            String shardingValue = preciseShardingValue.getValue();
            int index = getIndex(shardingValue);
            for(String name : tableNames){
                if(name.endsWith(index+"")){
                    return name;
                }
            }

        }

        return null;
    }


    private int getIndex(String uid){
        uid = uid.replace("-","");
        int res =  Math.abs(uid.hashCode())%10;
        return res;
    }
}


数据库中建了10张表 根据用户ID(为UUID)的hashcode 然后取模于 表的数量
doSharding(Collection tableNames, PreciseShardingValue preciseShardingValue) 参数说明:
tableNames 为当前所有的表名,preciseShardingValue 为当前分片字段的值

2、定义完后再配置文件中 指定下自定义配置的类文件(需要指定两个属性值)

#分片字段名称
spring.shardingsphere.sharding.tables.t_user.table-strategy.standard.sharding-column=user_id
分片策略自定义对应的类名
spring.shardingsphere.sharding.tables.t_user.table-strategy.standard.precise-algorithm-class-name=com.spring.config.UserAlgorithm

对应shardingsphere版本

        <dependency>
            <groupId>org.apache.shardingsphere</groupId>
            <artifactId>sharding-jdbc-spring-boot-starter</artifactId>
            <version>4.0.0-RC1</version>
        </dependency>
  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
ShardingSphere-JDBC中,自定义分片策略需要实现`ShardingAlgorithm`接口,并在配置文件中引用该自定义算法。 例如,假设要按照部门名称前缀进行分片,可以自定义一个`DeptNamePrefixShardingAlgorithm`算法,实现如下: ```java public class DeptNamePrefixShardingAlgorithm implements ShardingAlgorithm<String> { @Override public Collection<String> doSharding(Collection<String> availableTargetNames, ShardingValue<String> shardingValue) { Set<String> result = new HashSet<>(); for (String targetName : availableTargetNames) { for (String value : shardingValue.getValues()) { if (targetName.endsWith(value.substring(0, 1))) { result.add(targetName); } } } return result; } } ``` 在ShardingSphere配置文件中引用该算法,例如: ```yaml shardingRule: tables: dept: actualDataNodes: ds${0..1}.dept_${0..3} tableStrategy: standard: shardingColumn: dept_name shardingAlgorithmName: deptNamePrefixShardingAlgorithm shardingAlgorithms: deptNamePrefixShardingAlgorithm: type: INLINE props: algorithm.expression: dept_${dept_name.substring(0, 1)} algorithm.shardingColumns: dept_name ``` 在代码中进行数据库操作时,需要将部门名称前缀作为分片键,例如: ```java try (Connection conn = dataSource.getConnection()) { String sql = "SELECT * FROM dept WHERE dept_name = ?"; try (PreparedStatement ps = conn.prepareStatement(sql)) { ps.setString(1, "IT"); try (ResultSet rs = ps.executeQuery()) { while (rs.next()) { // 处理查询结果 } } } } ``` 需要注意的是,自定义分片策略需要实现`ShardingAlgorithm`接口,并根据具体业务需求进行调整。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值