项目原本使用springboot+mybatis动态数据源,最近项目需对接hive数据库,需添加新数据源,原本以为直接在spring.datasource.dynamic.datasource配置中添加一个数据源配置即可,结果发现不支持hive数据库,经查找各种springboot动态数据源之hive的博客后发现都是需要自己定义数据源然后去进行配置需要的数据源,但是这种方式都是需要把原有的数据源也进行重新编写,否则的话自定义的数据源对象会覆盖已有的动态数据源对象,本文介绍不修改已有数据源配置文件的情况下,添加其他数据源的方案。
步骤一:
新增一个数据源配置类,用于获取数据源参数信息,并实现BeanPostProcessor接口:
仅需实现其postProcessAfterInitialization方法,在这个方法中,把自定义的数据源(可是任意类型)添加到动态数据源DynamicRoutingDataSource中,并指定一个名字
import com.alibaba.druid.pool.DruidDataSource;
import com.baomidou.dynamic.datasource.DynamicRoutingDataSource;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.beans.factory.config.BeanPostProcessor;
import org.springframework.context.annotation.Configuration;
@Configuration
public class HiveConfig implements BeanPostProcessor {
@Value("${hive.url}")
private String url;
@Value("${hive.username}")
private String username;
@Value("${hive.password}")
private String password;
@Value("${hive.driverClassName}")
private String driverClassName;
@Override
public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
if(bean instanceof DynamicRoutingDataSource){
DruidDataSource ds= new DruidDataSource();
ds.setUrl(url);
ds.setUsername(username);
ds.setPassword(password);
ds.setDriverClassName(driverClassName);
((DynamicRoutingDataSource)bean).addDataSource("hive",ds);
}
return bean;
}
}
步骤二:
在需要这个数据源的地方使用@DS注解,指定所需用的数据源名称
import com.baomidou.dynamic.datasource.annotation.DS;
import org.apache.ibatis.annotations.Select;
import org.springframework.stereotype.Repository;
import java.util.List;
import java.util.Map;
@DS("hive")
@Repository
public interface TestDao {
@Select("select id,name from test")
List<Map<String,Object>> queryList();
}