spring从数据库读入数据源并初始化

本是古典何须时尚 2022-05-23 01:15 135阅读 0赞

@Component
    public class DcDataSourceInitEvent implements ApplicationContextAware, ApplicationListener<ContextRefreshedEvent> {
    
        private ConfigurableApplicationContext applicationContext;
    
        @Resource
        private DataSourceMapper dataSourceMapper;
    
    
        @Override
        public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
            this.applicationContext = (ConfigurableApplicationContext) applicationContext;
        }
    
    
        @Override
        public void onApplicationEvent(ContextRefreshedEvent contextRefreshedEvent)  {
            DataSourceCfg searchParam = new DataSourceCfg();
            searchParam.setDisabled(0);
            List<DataSourceCfg> dsList = dataSourceMapper.getDataSourceCfgList(searchParam);
    
            if (CollectionUtils.isNotEmpty(dsList)) {
                DefaultListableBeanFactory beanFactory = (DefaultListableBeanFactory) applicationContext.getBeanFactory();//定义spring相关信息
                BeanDefinitionBuilder beanDefinitionBuilder = BeanDefinitionBuilder.genericBeanDefinition(JdbcTemplate.class);
    
                for (DataSourceCfg dc : dsList) {
                    String configJson = dc.getConfig();
                    DruidDataSourceConfig config = JSON.parseObject(configJson, DruidDataSourceConfig.class);
    
                    DruidDataSource ds = new DruidDataSource();//设置diuid相关信息
                    ds.setDriverClassName(config.getDriverClassName());
                    ds.setUrl(config.getUrl());
                    ds.setUsername(config.getUsername());
                    ds.setPassword(config.getPassword());
                    ds.setInitialSize(config.getInitialSize());
                    ds.setMinIdle(config.getMinIdle());
                    ds.setMaxActive(config.getMaxActive());
                    ds.setMaxWait(config.getMaxWait());
                    ds.setTimeBetweenEvictionRunsMillis(config.getTimeBetweenEvictionRunsMillis());
                    ds.setMinEvictableIdleTimeMillis(config.getMinEvictableIdleTimeMillis());
                    ds.setMaxEvictableIdleTimeMillis(config.getMaxEvictableIdleTimeMillis());
                    ds.setValidationQuery(config.getValidationQuery());
                    ds.setTestWhileIdle(config.isTestWhileIdle());
                    ds.setTestOnBorrow(config.isTestOnBorrow());
                    ds.setTestOnReturn(config.isTestOnReturn());
                    try {
                        ds.setFilters(config.getFilters());
                    } catch (SQLException e) {
                        e.printStackTrace();
                    }
                    ds.setConnectionProperties("config.decrypt=true;config.decrypt.key=" + config.getPublicKey());
                    ds.setConnectionInitSqls(config.getConnectionInitSqls());
    
                    JdbcTemplate jdbcTemplate = new JdbcTemplate(ds);
                    beanFactory.registerSingleton(dc.getDataSource(), jdbcTemplate);//注册bean
                }
            }
        }
    
    }

数据库druid连接池相关信息:

{
    	"driverClassName": "org.postgresql.Driver",
    	"filters": "config",
    	"initialSize": 1,
    	"maxActive": 20,
    	"maxEvictableIdleTimeMillis": 3000000,
    	"maxWait": 60000,
    	"minEvictableIdleTimeMillis": 300000,
    	"minIdle": 3,
    	"password": "LcVkmR8FwPDUbxGGZredWFOUeMq57621xxxx4vg5A5Lganl9nOx7IXw8S3t3zNMAG4J7poZ15HqxAfhDPknYUDIkfA==",
    	"publicKey": "MFwwDQYJKoZIhvcNAQEBBQxxxxADSwAwSAJBAIVMhWXIMXNsdKSGQ5d3AnbVSy3+Uv5HsMIp0lojhxBD9oAyDVOrNi+mqn6O1Rw1TjP7ASjdSHj9v3RBld/hud8CAwEAAQ==",
    	"testOnBorrow": false,
    	"testOnReturn": false,
    	"testWhileIdle": true,
    	"timeBetweenEvictionRunsMillis": 60000,
    	"url": "jdbc:postgresql://127.0.0.1:1921/postgres",
    	"username": "postgres",
    	"validationQuery": "SELECT 1"
    }

beanFactory.registerSingleton(dc.getDataSource(), jdbcTemplate);//注册bean

这里注入了一个bean到spring，例如: myJdbcTemplate, 在其他地方就可以通过:

@Autowired

private JdbcTemplate myJdbcTemplate; 若用的其他如mybaits，只要按照相关<bean>信息构造类注入到spring即可。