SpringBoo中Mybatis多数据源动态切换

package com.xiaoxie.dds.service.impl; import com.alibaba.druid.pool.DruidDataSource; import com.xiaoxie.dds.domain.Ds; import com.xiaoxie.dds.domain.dto.DsDto; import com.xiaoxie.dds.manager.DynamicDataSourceManager; import com.xiaoxie.dds.mapper.DsMapper; import com.xiaoxie.dds.service.DsService; import com.xiaoxie.dds.utils.AESUtils; import lombok.RequiredArgsConstructor; import org.springframework.beans.BeanUtils; import org.springframework.beans.factory.annotation.Value; import org.springframework.stereotype.Service; import org.springframework.transaction.annotation.Transactional; import java.sql.Connection; import java.sql.DriverManager; import java.util.Date; import java.util.List; import java.util.Map; import java.util.concurrent.ConcurrentHashMap; @Service @RequiredArgsConstructor public class DsServiceImpl implements DsService { private final DynamicDataSourceManager dynamicDataSourceManager; private final Map<Object, Object> dataSourceMap = new ConcurrentHashMap<>(); @Value("${crypto.secret}") private String secretKey; private final DsMapper dsMapper; @Override public List<Ds> selectByName(String name) { return dsMapper.selectByName(name); } @Override public boolean testConnection(DsDto dsDto) { try { String password = AESUtils.decrypt(dsDto.getPassword(), secretKey); return testConnectDB(dsDto.getUrl(), dsDto.getUsername(), password); } catch (Exception e) { e.printStackTrace(); return false; } } /** * 添加数据源 * @param dsDto * @return */ @Override @Transactional public boolean addDs(DsDto dsDto){ //1、把数据库连接信息保存到数据库 Ds ds = new Ds(); BeanUtils.copyProperties(dsDto, ds); ds.setCreateTime(new Date()); dsMapper.insertSelective(ds); //2、动态添加新的数据源 try { String password = AESUtils.decrypt(dsDto.getPassword(), secretKey); DruidDataSource dataSource = new DruidDataSource(); dataSource.setUrl(dsDto.getUrl()); dataSource.setUsername(dsDto.getUsername()); dataSource.setPassword(password); dataSource.setDriverClassName("com.mysql.cj.jdbc.Driver"); dataSourceMap.put(dsDto.getName(), dataSource); dynamicDataSourceManager.setTargetDataSources(dataSourceMap); dynamicDataSourceManager.afterPropertiesSet(); } catch (Exception e) { throw new RuntimeException(e); } return true; } /** * 切换数据源 * @param dataSourceName */ @Override public void switchDataSource(String dataSourceName) { DynamicDataSourceManager.setDataSource(dataSourceName); } @Override public void clearDataSource() { DynamicDataSourceManager.clearDataSource(); } @Override public List<Ds> selectAll() { return dsMapper.selectAll(); } /** * 测试数据库是否可连接 * @param url * @param username * @param password * @return */ private boolean testConnectDB(String url, String username, String password) { try{ Connection connection = DriverManager.getConnection(url, username, password); return connection != null && !connection.isClosed(); } catch (Exception e) { return false; } } }