SpringBoot+AOP——切换数据源

妖狐艹你老母 2022-11-28 13:59 153阅读 0赞

1.  新建一个Maven项目，所用依赖如下

<parent>
          <groupId>org.springframework.boot</groupId>
          <artifactId>spring-boot-starter-parent</artifactId>
          <version>2.0.3.RELEASE</version>
          <relativePath/>
       </parent>
    
      <properties>
        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
        <java.version>1.8</java.version>
      </properties>
    
    	<dependencies>
    		<dependency>
    			<groupId>org.springframework.boot</groupId>
        		<artifactId>spring-boot-starter-web</artifactId>
    		</dependency>
    		<dependency>
    			<groupId>org.springframework.boot</groupId>
        		<artifactId>spring-boot-starter-test</artifactId>
    		</dependency>
    
    		<dependency>
    		    <groupId>mysql</groupId>
    		    <artifactId>mysql-connector-java</artifactId>
    		    <version>8.0.20</version>
    		</dependency>
    
    		<dependency>
    		    <groupId>com.alibaba</groupId>
    		    <artifactId>druid</artifactId>
    		    <version>1.1.22</version>
    		</dependency>
    		<dependency>
    	    	<groupId>org.springframework.boot</groupId>
    	    	<artifactId>spring-boot-devtools</artifactId>
    	    </dependency>
    	    
    		<dependency>
    		    <groupId>com.github.abel533</groupId>
    		    <artifactId>mapper</artifactId>
    		    <version>2.3.0</version>
    		</dependency>
    
    		<dependency>
    		    <groupId>com.alibaba</groupId>
    		    <artifactId>fastjson</artifactId>
    		    <version>1.2.72</version>
    		</dependency>
    	    <dependency>
    	    	<groupId>org.mybatis.spring.boot</groupId>
    	    	<artifactId>mybatis-spring-boot-starter</artifactId>
    	    	<version>1.3.2</version>
    	    </dependency>
    	    
    		<dependency>
    			<groupId>com.github.pagehelper</groupId>
    			<artifactId>pagehelper</artifactId>
    			<version>5.1.10</version>
    		</dependency>
    		
    	    <dependency>
    		    <groupId>org.springframework.boot</groupId>
    		  	<artifactId>spring-boot-starter-aop</artifactId>
    	    </dependency>
      </dependencies>
    </project>

1.  配置数据源(此处不展示配置信息，配置信息根据value内的key写在SpringBoot配置文件内)

@Configuration
    public class Test1DruidDataSourcesConfig { 
    
    	@Value(value = "${spring.datasource.driver-class-name}")
    	private String driverClass;
    	
    	@Value(value = "${spring.datasource.url1}")
    	private String jdbcUrl1;
    	
    	@Value(value = "${spring.datasource.url2}")
    	private String jdbcUrl2;
    	
    	@Value(value = "${spring.datasource.username}")
    	private String username;
    	
    	@Value(value = "${spring.datasource.password}")
    	private String password;
    	
    	@Value(value = "${spring.datasource.druid.initial-size}")
    	private Integer initialSize;
    	
    	@Value(value = "${spring.datasource.druid.min-idle}")
    	private Integer minIdle;
    
    	@Value(value = "${spring.datasource.druid.max-active}")
    	private Integer maxActive;
    	
    	@Value(value = "${spring.datasource.druid.max-wait}")
    	private Long maxWaitMillis;
    	
    	@Bean(value = "df")
    	public DataSourcesFilter dataSourcesFilter() { 
    		return new DataSourcesFilter();
    	}
    	
    	@Bean(name = "test1DataSource")
    	@Primary
    	public DataSource primaryDataSource() throws SQLException { 
    		DruidDataSource dataSource = new DruidDataSource();
    		dataSource.setDriverClassName(driverClass);
    		dataSource.setUrl(jdbcUrl1);
    		dataSource.setUsername(username);
    		dataSource.setPassword(password);
    		dataSource.setInitialSize(initialSize);
    		dataSource.setMinIdle(minIdle);
    		dataSource.setMaxActive(maxActive);
    		dataSource.setMaxWait(maxWaitMillis);
    		dataSource.setLogDifferentThread(true);
    		dataSource.setFilters("stat,wall,slf4j");
    		return dataSource;
    	}
    	@Bean(name = "test2DataSource")
    	public DataSource dataSource() throws SQLException { 
    		DruidDataSource dataSource = new DruidDataSource();
    		dataSource.setDriverClassName(driverClass);
    		dataSource.setUrl(jdbcUrl2);
    		dataSource.setUsername(username);
    		dataSource.setPassword(password);
    		dataSource.setInitialSize(initialSize);
    		dataSource.setMinIdle(minIdle);
    		dataSource.setMaxActive(maxActive);
    		dataSource.setMaxWait(maxWaitMillis);
    		dataSource.setLogDifferentThread(true);
    		dataSource.setFilters("stat,wall,slf4j");
    		return dataSource;
    	}
    	//连接池管理
    	@Bean
    	public DynamicDataSource dynamicDataSource(@Qualifier("test1DataSource")DataSource test1DataSource,
    			@Qualifier("test2DataSource")DataSource test2DataSource) { 
    		Map<Object, Object> targetDataSource = new HashMap<Object, Object>(4);
    		targetDataSource.put(DataSourceType.DataBaseType.TEST01, test1DataSource);
    		targetDataSource.put(DataSourceType.DataBaseType.TEST02, test2DataSource);
    		DynamicDataSource dynamicDataSource = new DynamicDataSource();
    		dynamicDataSource.setTargetDataSources(targetDataSource);
    		dynamicDataSource.setDefaultTargetDataSource(test1DataSource);
    		return dynamicDataSource;
    	}
    	
    	@Bean
    	@Primary
    	public SqlSessionFactory sqlSessionFactory(@Autowired DynamicDataSource dataSource) throws Exception { 
    		SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
    		bean.setDataSource(dataSource);
            PageInterceptor interceptor = new PageInterceptor();
            Properties properties = new Properties();
            properties.setProperty("offsetAsPageNum","true");
            properties.setProperty("rowBoundsWithCount","true");
            properties.setProperty("reasonable","true");
            interceptor.setProperties(properties);
    		bean.setPlugins(new Interceptor[] { interceptor});
    		return bean.getObject();
    	}
    	
    	@Bean
    	@Primary
    	public SqlSessionTemplate sqlSessionTemplate(@Autowired SqlSessionFactory sessionFactory) { 
    		return new SqlSessionTemplate(sessionFactory);
    	}
    	
    }

1.  定义一个数据源切换的方法

public class DataSourceType { 
    	private static final Logger LOGGER = LoggerFactory.getLogger(DataSourceType.class);
    	public enum DataBaseType{ 
    		TEST01, TEST02
    	}
    	private static final ThreadLocal<DataBaseType> TYPE = new ThreadLocal<DataBaseType>();
    	public static void setDataBaseType(DataBaseType dataBaseType) { 
    		if(dataBaseType == null) { 
    			throw new NullPointerException("数据类型不存在");
    		}
    		LOGGER.info("[当前数据源切换为]: {}", dataBaseType);
    		TYPE.set(dataBaseType);
    	}
    	public static DataBaseType getDataBaseType() { 
    		DataBaseType dataBaseType = TYPE.get() == null ? DataBaseType.TEST01 : TYPE.get();
    		LOGGER.info("[获取当前数据源为]: {}", dataBaseType);
    		return dataBaseType;
    	}
    	public static void clearDataBaseType() { 
    		TYPE.remove();
    	}
    }

1.  定义一个DataSource连接池路由，重写determineCurrentLookupKey方法

public class DynamicDataSource extends AbstractRoutingDataSource{ 
    	@Override
    	protected Object determineCurrentLookupKey() { 
    		DataSourceType.DataBaseType dataBaseType = DataSourceType.getDataBaseType();
    		return dataBaseType;
    	}
    }

1.  定义一个连接池切面进行切换连接池

@Aspect
    @Component
    public class DataSourceAop { 
    
    	private static final Logger LOGGER = LoggerFactory.getLogger(DataSourceAop.class);
    	
    	@Before("execution(* com.demo.mybatis.*.service.*.test01*(..))")
    	public void setDataSource1() { 
    		LOGGER.info("执行切换数据源切面操作。。。TEST01");
    		DataSourceType.setDataBaseType(DataBaseType.TEST01);
    	}
    
    	@Before("execution(* com.demo.mybatis.*.service.*.test02*(..))")
    	public void setDataSource2() { 
    		LOGGER.info("执行切换数据源切面操作。。。TEST02");
    		DataSourceType.setDataBaseType(DataBaseType.TEST02);
    	}
    	
    }

1.  编写对应的方法

@Service
    public class MyQueryService { 
    
    	@Autowired
    	private SqlSessionFactory sessionFactory;
    	
    	public List<Map> test01queryAsMap(String sql, Map<String, Object> params){ 
    		SqlSession session = sessionFactory.openSession(TransactionIsolationLevel.NONE);
    		SqlMapper sqlMapper = new SqlMapper(session);
    		String sqlExcute = "<script>"
    				+ sql
    				+ "</script>";
    		return sqlMapper.selectList(sqlExcute, params, Map.class);
    	}
    
    	public List<Map> test02queryAsMap(String sql, Map<String, Object> params){ 
    		SqlSession session = sessionFactory.openSession(TransactionIsolationLevel.NONE);
    		SqlMapper sqlMapper = new SqlMapper(session);
    		String sqlExcute = "<script>"
    				+ sql
    				+ "</script>";
    		return sqlMapper.selectList(sqlExcute, params, Map.class);
    	}
    	
    }

1.  定义启动类并调用方法，观察控制台输出内容

@SpringBootApplication
    public class MybatisMapperApplication implements CommandLineRunner
    { 
    	@Autowired
    	private MyQueryController controller;
    	
        public static void main( String[] args )
        { 
            new SpringApplicationBuilder(MybatisMapperApplication.class).run(args);
        }
    
    	@Override
    	public void run(String... args) throws Exception { 
    		System.out.println(JSON.toJSONString(controller.query("select * from fbp_employee", new HashMap<String, Object>())));
    		System.out.println(JSON.toJSONString(controller.query1("select * from fbp_employee", new HashMap<String, Object>())));
    		
    	}
    }