Sharding-jdbc依赖
<dependency>
<groupId>com.dangdang</groupId>
<artifactId>sharding-jdbc-config-common</artifactId>
<version>1.5.4.1</version>
</dependency>
1.配置多数据源
public DataSource datasource1(){
DruidDataSource druidDataSource=new DruidDataSource();
druidDataSource.setDriverClassName("com.mysql.jdbc.Driver");
druidDataSource.setUrl("jdbc:mysql://localhost:3306/sharding");
druidDataSource.setUsername("root");
druidDataSource.setPassword("123456");
return druidDataSource;
}
public DataSource datasource2(){
DruidDataSource druidDataSource=new DruidDataSource();
druidDataSource.setDriverClassName("com.mysql.jdbc.Driver");
druidDataSource.setUrl("jdbc:mysql://localhost:3306/sharding1");
druidDataSource.setUsername("root");
druidDataSource.setPassword("123456");
return druidDataSource;
}
2.配置mybatis
private static Log logger = LogFactory.getLog(MybatisConfiguration.class);
@Autowired
XbDataSource xbDataSource;
@Bean
public SqlSessionFactory sqlSessionFactory(){
SqlSessionFactoryBean sqlSessionFactoryBean=new SqlSessionFactoryBean();
ResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
sqlSessionFactoryBean.setDataSource(xbDataSource.getShardingDataSource());
sqlSessionFactoryBean.setTypeAliasesPackage("limouchao.entity");
try {
sqlSessionFactoryBean.setMapperLocations(resolver.getResources("classpath:mapper/*.xml"));
} catch (IOException e1) {
e1.printStackTrace();
}
try {
return sqlSessionFactoryBean.getObject();
} catch (Exception e) {
logger.error("sqlSessionFactory create bean fail", e);
return null;
}
}
@Bean
public SqlSessionTemplate sqlSessionTemplate(SqlSessionFactory sqlSessionFactory){
return new SqlSessionTemplate(sqlSessionFactory);
}
@Override
public PlatformTransactionManager annotationDrivenTransactionManager() {
return new DataSourceTransactionManager(xbDataSource.getShardingDataSource());
}
3.配置Sharding-jdbc数据源
private DataSource shardingDataSource;
@PostConstruct
public void init(){
HashMap<String, DataSource> map = new HashMap<>();
DatasourceConfig datasourceConfig=new DatasourceConfig();
map.put("dataSource0", datasourceConfig.datasource1());
map.put("dataSource1", datasourceConfig.datasource2());
DataSourceRule dataSourceRule =new DataSourceRule(map);
List<String> pList = new ArrayList<>();
for (int i = 0; i < 4; i++) {
pList.add("user" + i);
}
List<TableRule> tableRuleList = new ArrayList<>();
tableRuleList.add(new TableRule.TableRuleBuilder("user")
.actualTables(pList)
.dataSourceRule(dataSourceRule)
.tableShardingStrategy(new TableShardingStrategy("id",new ProgramShardingAlgorithm())).databaseShardingStrategy(new DatabaseShardingStrategy("id", new ProgramDataBaseAlgorithm())).build());//.databaseShardingStrategy(new DatabaseShardingStrategy("id", new ProgramDataBaseAlgorithm()))
ShardingRule shardingRule = ShardingRule.builder().dataSourceRule(dataSourceRule).tableRules(tableRuleList).build();
try {
shardingDataSource = ShardingDataSourceFactory.createDataSource(shardingRule);
} catch (SQLException e) {
e.printStackTrace();
}
}
public DataSource getShardingDataSource() {
return shardingDataSource;
}
4.实现分表分库策略
1.分表策略实现SingleKeyTableShardingAlgorithm
@Component
public class ProgramShardingAlgorithm implements SingleKeyTableShardingAlgorithm<String>{
@Override
public String doEqualSharding(Collection<String> availableTargetNames,
ShardingValue<String> shardingValue) {
for (String each : availableTargetNames) {
//判断值是否在设计表中的哪一个。判断string的后缀是否为。。。
// if (each.endsWith(shardingValue.getValue()%4+"")) {
// return each;
// }
if(each.endsWith(Math.abs(shardingValue.getValue().hashCode())%4+"")){
return each;
}
}
throw new UnsupportedOperationException();
}
@Override
public Collection<String> doInSharding(Collection<String> availableTargetNames,
ShardingValue<String> shardingValue) {
Collection<String> result = new LinkedHashSet<>(availableTargetNames.size());
Collection<String> values = shardingValue.getValues();
for (String value : values) {
for (String tableNames : availableTargetNames) {
if (tableNames.endsWith(value.hashCode()%4+"")) {
result.add(tableNames);
}
}
}
return result;
}
@Override
public Collection<String> doBetweenSharding(Collection<String> availableTargetNames,
ShardingValue<String> shardingValue) {
return null;
// Collection<String> result = new LinkedHashSet<String>(availableTargetNames.size());
// Range<Integer> range = shardingValue.getValueRange();
// for (Integer i = range.lowerEndpoint(); i <= range.upperEndpoint(); i++) {
// for (String each : availableTargetNames) {
// if (each.endsWith(i % 4+ "")) {
// result.add(each);
// }
// }
// }
// return result;
}
}
2.分库实现SingleKeyDatabaseShardingAlgorithm
public class ProgramDataBaseAlgorithm implements SingleKeyDatabaseShardingAlgorithm<String>{
@Override
public String doEqualSharding(Collection<String> availableTargetNames, ShardingValue<String> shardingValue) {
for (String each : availableTargetNames) {
if (each.endsWith((Math.abs(shardingValue.getValue().hashCode())%2) + "")) {
return each;
}
}
throw new IllegalArgumentException();
}
@Override
public Collection<String> doInSharding(Collection<String> availableTargetNames, ShardingValue<String> shardingValue) {
Collection<String> result = new LinkedHashSet<String>(availableTargetNames.size());
for (String value : shardingValue.getValues()) {
for (String tableName : availableTargetNames) {
if (tableName.endsWith((Math.abs(value.hashCode())) % 2 + "")) {
System.out.println(value.hashCode()+"======"+Math.abs(value.hashCode())% 2);
result.add(tableName);
}
}
}
return result;
}
@Override
public Collection<String> doBetweenSharding(Collection<String> availableTargetNames, ShardingValue<String> shardingValue) {
// Collection<String> result = new LinkedHashSet<String>(availableTargetNames.size());
// Range<String> range = (Range<String>) shardingValue.getValueRange();
// for (String i = range.lowerEndpoint(); i <= range.upperEndpoint(); i++) {
// for (String each : availableTargetNames) {
// if (each.endsWith(i % 4 + "")) {
// result.add(each);
// }
// }
// }
// return result;
return null;
}
}
参考url
https://www.cnblogs.com/zwt1990/p/6762135.html