CentralDogma

配置中心,也可以是别的比如库等,核心用于存储数据源配置信息,也可以结合openmetadata进行资产管理

  1.         <dependency>
  2.             <groupId>com.linecorp.centraldogma</groupId>
  3.             <artifactId>centraldogma-client-spring-boot-starter</artifactId>
  4.             <version>0.51.1</version>
  5.         </dependency>
  1. centraldogma:
  2.   enable: true
  3.   hosts:
  4.     - localhost:36462
  5.   access-token: appToken-d
  6.   project: datasource
  7.   repo: dev
  8.   filename: /conf.json
  1. [
  2.   {
  3.     "id": "61e9107e283fcc36e66213a1",
  4.     "alias": "mysql",
  5.     "url": "",
  6.     "username": "root",
  7.     "password": "123456",
  8.     "supportPool": true,
  9.     "maximumPoolSize": 10,
  10.     "maxLifetime": 2,
  11.     "driver": "com.mysql.jdbc.Driver",
  12.     "connectionTestQuery": "select 1"
  13.   }
  14. ]

MyDataSource

成员变量 dataSourceManager 和 dataSource, 前者为配置中心的配置数据,后者用于DataSource接口中声明的方法.

  1. public class MyDataSource implements DataSource {
  2.     @Setter
  3.     private DataSourceManager dataSourceManager;
  5.     private DataSource dataSource;
  8.     @SneakyThrows
  9.     protected void init() {
  10.         if (dataSourceManager.getSupportPool()) {
  11.             HikariConfig config = new HikariConfig();
  12.             config.setUsername(dataSourceManager.getUsername());
  13.             config.setJdbcUrl(dataSourceManager.getUrl());
  14.             config.setPassword(dataSourceManager.getPassword());
  15.             config.setMaximumPoolSize(dataSourceManager.getMaximumPoolSize());
  16.             config.setMaxLifetime(dataSourceManager.getMaxLifetime());
  17.             config.setDriverClassName(dataSourceManager.getDriver());
  18.             config.setConnectionTestQuery(dataSourceManager.getConnectionTestQuery());
  19.             this.dataSource = new HikariDataSource(config);
  20.         } else {
  21.             SimpleDriverDataSource dataSource = new SimpleDriverDataSource();
  22.             dataSource.setUsername(dataSourceManager.getUsername());
  23.             dataSource.setPassword(dataSourceManager.getPassword());
  24.             dataSource.setUrl(dataSourceManager.getUrl());
  25.             dataSource.setDriverClass((Class<? extends Driver>) Class.forName(dataSourceManager.getDriver()));
  26.             this.dataSource = dataSource;
  27.         }
  28.     }
  30.     @Override
  31.     public Connection getConnection() throws SQLException {
  32.         return dataSource.getConnection();
  33.     }
  35.     @Override
  36.     public Connection getConnection(String s, String s1) throws SQLException {
  37.         return dataSource.getConnection();
  38.     }
  40.     @Override
  41.     public <T> T unwrap(Class<T> aClass) throws SQLException {
  42.         return dataSource.unwrap(aClass);
  43.     }
  45.     @Override
  46.     public boolean isWrapperFor(Class<?> aClass) throws SQLException {
  47.         return dataSource.isWrapperFor(aClass);
  48.     }
  50.     @Override
  51.     public PrintWriter getLogWriter() throws SQLException {
  52.         return dataSource.getLogWriter();
  53.     }
  55.     @Override
  56.     public void setLogWriter(PrintWriter printWriter) throws SQLException {
  57.         dataSource.setLogWriter(printWriter);
  58.     }
  60.     @Override
  61.     public int getLoginTimeout() throws SQLException {
  62.         return dataSource.getLoginTimeout();
  63.     }
  65.     @Override
  66.     public void setLoginTimeout(int i) throws SQLException {
  67.         dataSource.setLoginTimeout(dataSource.getLoginTimeout());
  68.     }
  70.     @Override
  71.     public Logger getParentLogger() throws SQLFeatureNotSupportedException {
  72.         return dataSource.getParentLogger();
  73.     }
  74. }

CommandLineRunner

  1. 监听配置中心传来的数据
  2. 循环注册bean
     @Bean
 @ConditionalOnProperty(prefix = "centraldogma", value = "enable", havingValue = "true")
 @ConditionalOnMissingBean
 public CommandLineRunner dataSourceCommandLineRunner(CentralDogma dogma, ObjectMapper objectMapper, ConfigurableApplicationContext applicationContext) {
     Watcher watcher = dogma.fileWatcher(configuration.getProject(), configuration.getRepo(), Query.ofText(configuration.getFilename()));
     watcher.watch((revision, value) -> {
         log.info("Updated to {} at {}", value, revision);
         try {
             BeanDefinitionRegistry beanFactory = (BeanDefinitionRegistry) applicationContext.getBeanFactory();
             List<DataSourceManager> all = objectMapper.readValue(value.toString(), new TypeReference<List<DataSourceManager>>() {
             });
             for (DataSourceManager dataSourceManager : all) {
                if (beanFactory.containsBeanDefinition(dataSourceManager.getId())) {
                     log.info("datasource exist id:{},name:{}", dataSourceManager.getId(), dataSourceManager.getAlias());
                     beanFactory.removeBeanDefinition(dataSourceManager.getId());
                     log.info("old datasource remove id:{},name:{}", dataSourceManager.getId(), dataSourceManager.getAlias());
                 }
                 BeanDefinitionBuilder beanDefinitionBuilder = BeanDefinitionBuilder.genericBeanDefinition(OneserviceDataSource.class);
                 beanDefinitionBuilder.addPropertyValue("dataSourceManager", dataSourceManager);// setter 注入
                 beanDefinitionBuilder.setInitMethodName("init");// init 方法初始化 私有变量
                 beanFactory.registerBeanDefinition(dataSourceManager.getId(), beanDefinitionBuilder.getBeanDefinition());
                 log.info("datasource register success  id :{},alias:{}", dataSourceManager.getId(), dataSourceManager.getAlias());
             }
         } catch (JsonProcessingException e) {
             e.printStackTrace();
             log.error("json convert  error", e);
         }
     });
     return args -> {
     };
 }

    注入

    由于监听是异步操作 ,故bean应该是懒加载
     @Autowired
 @Lazy
 @Qualifier("61e9107e283fcc36e66213a1")
 private DataSource dataSource;