shardingJDBC配置-java多数据源分表

直接上代码

pom文件配置

<!-- jdbc -->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-jdbc</artifactId>
            <exclusions>
                <exclusion>
                    <groupId>org.apache.tomcat</groupId>
                    <artifactId>tomcat-jdbc</artifactId>
                </exclusion>
            </exclusions>
        </dependency>
        <dependency>
            <groupId>org.apache.shardingsphere</groupId>
            <artifactId>sharding-jdbc-core</artifactId>
            <version>4.0.0</version>
        </dependency>
        <!-- mysql驱动 -->
        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
        </dependency>
        <dependency>
            <groupId>org.mybatis.spring.boot</groupId>
            <artifactId>mybatis-spring-boot-starter</artifactId>
            <version>1.3.2</version>
        </dependency>

数据分表配置代码

package com.XXX.config.mybatis;

import com.zaxxer.hikari.HikariDataSource;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.session.SqlSessionFactory;
import org.apache.shardingsphere.api.config.sharding.ShardingRuleConfiguration;
import org.apache.shardingsphere.api.config.sharding.TableRuleConfiguration;
import org.apache.shardingsphere.api.config.sharding.strategy.InlineShardingStrategyConfiguration;
import org.apache.shardingsphere.api.config.sharding.strategy.NoneShardingStrategyConfiguration;
import org.apache.shardingsphere.shardingjdbc.api.ShardingDataSourceFactory;
import org.mybatis.spring.SqlSessionFactoryBean;
import org.mybatis.spring.SqlSessionTemplate;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;

import javax.sql.DataSource;
import java.io.IOException;
import java.sql.SQLException;
import java.util.HashMap;
import java.util.Map;
import java.util.Properties;

/**
 * 分表数据源配置
 */
@Configuration
@MapperScan(value = "com.XXX.mapper.fund",sqlSessionFactoryRef = "fundSqlSessionFactory", sqlSessionTemplateRef = "fundSqlSessionTemplate")
@Slf4j
public class FundDbConfig {

    @Value("${shardingJdbc.showSql:false}")
    private Boolean showSql;

//创建连接池
    @Bean(name = "fundDbDataSource")
    @Primary
    @ConfigurationProperties(prefix = "spring.datasource.fund")
    public DataSource fundDbDataSource() {
        DataSource dataSource = new HikariDataSource();
        //DataSourceBuilder.create().type(HikariDataSource.class).build();
        return dataSource;
    }
//事务管理
    @Bean(name = "fundTransactionManager")
    public DataSourceTransactionManager transactionManager(@Qualifier("fundDbShardingDataSource") DataSource dataSource) {
        return new DataSourceTransactionManager(dataSource);
    }

    @Bean(name = "fundSqlSessionFactory")
    public SqlSessionFactory fundSqlSessionFactory(@Qualifier("fundDbShardingDataSource") DataSource fundDbDataSource) throws IOException {
        SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
        bean.setDataSource(fundDbDataSource);
        bean.setMapperLocations(new PathMatchingResourcePatternResolver()
                .getResources("classpath:mapper/fund/*.xml"));
        try {
            return bean.getObject();
        } catch (Exception e) {
            log.error("fundSqlSessionFactory 异常",e);
            throw new RuntimeException(e);
        }
    }

    @Bean(name = "fundSqlSessionTemplate")
    public SqlSessionTemplate fundSqlSessionTemplate(@Qualifier("fundSqlSessionFactory")SqlSessionFactory fundSqlSessionFactory) throws Exception {
        SqlSessionTemplate sqlSessionTemplate=new SqlSessionTemplate(fundSqlSessionFactory);
        return sqlSessionTemplate;
    }

    @Bean(name="fundDbShardingDataSource")
    DataSource getShardingDataSource(@Qualifier("fundDbDataSource") DataSource fundDbDataSource) throws SQLException {
        ShardingRuleConfiguration shardingRuleConfig = new ShardingRuleConfiguration();
        shardingRuleConfig.getTableRuleConfigs().add(getUserFundTableRuleConfiguration());//表分片规则配置
        shardingRuleConfig.getBindingTableGroups().add("t_user_prop");//绑定表关系
        shardingRuleConfig.setDefaultDatabaseShardingStrategyConfig(new NoneShardingStrategyConfiguration());//默认库规则
        shardingRuleConfig.setDefaultTableShardingStrategyConfig(new NoneShardingStrategyConfiguration());//默认表规则
        Properties properties = new Properties();
        properties.put("sql.show", showSql);//是否展示执行sql,会详细打印出执行的所有详细的替换后的查询sql
        return ShardingDataSourceFactory.createDataSource(createDataSourceMap(fundDbDataSource), shardingRuleConfig, properties);
    }
//分片规则配置
    TableRuleConfiguration getUserFundTableRuleConfiguration () {
        //逻辑表名(映射文件中的表名)+表范围(库名.实际的表范围)
        TableRuleConfiguration result = new TableRuleConfiguration("t_user_prop", "fund_db.t_user_prop_${0..63}");
        result.setTableShardingStrategyConfig(new InlineShardingStrategyConfiguration("UserID","t_user_prop_${UserID%64}"));//表路由规则
        return result;
    }

    Map<String, DataSource> createDataSourceMap(DataSource fundDbDataSource) {
        Map<String, DataSource> result = new HashMap<>();
        result.put("fund_db", fundDbDataSource);
        return result;
    }
}

无数据分表配置

package com.XXX.config.mybatis;

import com.zaxxer.hikari.HikariDataSource;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.SqlSessionFactoryBean;
import org.mybatis.spring.SqlSessionTemplate;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;

import javax.sql.DataSource;
import java.io.IOException;

/**
 * 业务库配置
 */
@Configuration
@MapperScan(value = "com.XXX.mapper.work",sqlSessionFactoryRef = "workSqlSessionFactory", sqlSessionTemplateRef = "workSqlSessionTemplate")
@Slf4j
public class WorkDbConfig {

    @Bean(name = "workDbDataSource")
    @ConfigurationProperties(prefix = "spring.datasource.work")
    public DataSource workDbDataSource() {
        DataSource dataSource = new HikariDataSource();
        return dataSource;
    }

    @Bean(name = "workTransactionManager")
    public DataSourceTransactionManager transactionManager(@Qualifier("workDbDataSource") DataSource dataSource) {
        return new DataSourceTransactionManager(dataSource);
    }

    @Bean(name = "workSqlSessionFactory")
    public SqlSessionFactory workSqlSessionFactory(@Qualifier("workDbDataSource") DataSource workDbDataSource) throws IOException {
        SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
        bean.setDataSource(workDbDataSource);
        bean.setMapperLocations(new PathMatchingResourcePatternResolver()
                .getResources("classpath:mapper/work/*.xml"));
        try {
            return bean.getObject();
        } catch (Exception e) {
            log.error("qptreasureDbSqlSessionFactory 异常",e);
            throw new RuntimeException(e);
        }
    }

    @Bean(name = "workSqlSessionTemplate")
    public SqlSessionTemplate workSqlSessionTemplate(@Qualifier("workSqlSessionFactory")SqlSessionFactory workSqlSessionFactory) throws Exception {
        SqlSessionTemplate sqlSessionTemplate=new SqlSessionTemplate(workSqlSessionFactory);
        return sqlSessionTemplate;
    }
}

  • 1
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值