11package org .minbox .framework .api .boot .autoconfigure .datasource ;
22
3- import org .minbox .framework .api .boot .datasource .ApiBootDataSource ;
4- import org .minbox .framework .api .boot .datasource .ApiBootDataSourceFactoryBean ;
5- import org .minbox .framework .api .boot .datasource .aop .advistor .ApiBootDataSourceSwitchAdvisor ;
6- import org .minbox .framework .api .boot .datasource .aop .interceptor .ApiBootDataSourceSwitchAnnotationInterceptor ;
7- import org .minbox .framework .api .boot .datasource .config .DataSourceConfig ;
8- import org .minbox .framework .api .boot .datasource .config .DataSourceDruidConfig ;
9- import org .minbox .framework .api .boot .datasource .routing .ApiBootRoutingDataSource ;
10- import org .minbox .framework .api .boot .datasource .support .ApiBootDruidDataSource ;
11- import org .minbox .framework .api .boot .datasource .support .ApiBootHikariDataSource ;
3+ import com .alibaba .druid .pool .DruidDataSource ;
4+ import org .minbox .framework .datasource .DataSourceFactoryBean ;
5+ import org .minbox .framework .datasource .MinBoxDataSource ;
6+ import org .minbox .framework .datasource .aop .advistor .DataSourceSwitchAdvisor ;
7+ import org .minbox .framework .datasource .aop .interceptor .DataSourceSwitchAnnotationInterceptor ;
8+ import org .minbox .framework .datasource .config .DataSourceConfig ;
9+ import org .minbox .framework .datasource .config .DataSourceDruidConfig ;
10+ import org .minbox .framework .datasource .routing .ApiBootRoutingDataSource ;
11+ import org .minbox .framework .datasource .support .MinBoxDruidDataSource ;
12+ import org .minbox .framework .datasource .support .MinBoxHikariDataSource ;
1213import org .springframework .boot .autoconfigure .AutoConfigureBefore ;
1314import org .springframework .boot .autoconfigure .condition .ConditionalOnClass ;
1415import org .springframework .boot .autoconfigure .condition .ConditionalOnMissingBean ;
3031 * @author 恒宇少年
3132 */
3233@ Configuration
33- @ ConditionalOnClass ({ApiBootDataSource .class , AbstractRoutingDataSource .class })
34+ @ ConditionalOnClass ({MinBoxDataSource .class , AbstractRoutingDataSource .class })
3435@ EnableConfigurationProperties (ApiBootDataSourceSwitchProperties .class )
3536@ AutoConfigureBefore (DataSourceAutoConfiguration .class )
3637public class ApiBootDataSourceSwitchAutoConfiguration {
@@ -51,21 +52,21 @@ public ApiBootDataSourceSwitchAutoConfiguration(ApiBootDataSourceSwitchPropertie
5152 */
5253 @ Bean
5354 @ ConditionalOnMissingBean
54- public ApiBootDataSourceFactoryBean apiBootDataSourceFactoryBean () {
55- return new ApiBootDataSourceFactoryBean ();
55+ public DataSourceFactoryBean dataSourceFactoryBean () {
56+ return new DataSourceFactoryBean ();
5657 }
5758
5859 /**
5960 * ApiBoot Routing DataSource
6061 * switch use datasource
6162 * {@link DataSource}
6263 *
63- * @param apiBootDataSourceFactoryBean ApiBoot DataSource FactoryBean
64+ * @param dataSourceFactoryBean ApiBoot DataSource FactoryBean
6465 * @return DataSource
6566 */
6667 @ Bean
6768 @ ConditionalOnMissingBean
68- public DataSource dataSource (ApiBootDataSourceFactoryBean apiBootDataSourceFactoryBean ) {
69+ public DataSource dataSource (DataSourceFactoryBean dataSourceFactoryBean ) {
6970 List <DataSourceConfig > dataSourceConfigList = new LinkedList ();
7071 Map <String , DataSourceConfig > dataSourceConfigMap = new HashMap (1 );
7172
@@ -80,13 +81,13 @@ public DataSource dataSource(ApiBootDataSourceFactoryBean apiBootDataSourceFacto
8081 // set data source pool name
8182 dataSourceConfig .setPoolName (poolName );
8283 // datasource type
83- dataSourceConfig .setDataSourceType (dataSourceConfig instanceof DataSourceDruidConfig ? ApiBootDruidDataSource .class : ApiBootHikariDataSource .class );
84+ dataSourceConfig .setDataSourceType (dataSourceConfig instanceof DataSourceDruidConfig ? MinBoxDruidDataSource .class : MinBoxHikariDataSource .class );
8485
8586 // after convert add to data source list
8687 dataSourceConfigList .add (dataSourceConfig );
8788 });
8889
89- return new ApiBootRoutingDataSource (apiBootDataSourceFactoryBean , apiBootDataSourceSwitchProperties .getPrimary (), dataSourceConfigList );
90+ return new ApiBootRoutingDataSource (dataSourceFactoryBean , apiBootDataSourceSwitchProperties .getPrimary (), dataSourceConfigList );
9091 }
9192
9293 /**
@@ -96,21 +97,21 @@ public DataSource dataSource(ApiBootDataSourceFactoryBean apiBootDataSourceFacto
9697 */
9798 @ Bean
9899 @ ConditionalOnMissingBean
99- public ApiBootDataSourceSwitchAnnotationInterceptor apiBootDataSourceSwitchAnnotationInterceptor () {
100- return new ApiBootDataSourceSwitchAnnotationInterceptor ();
100+ public DataSourceSwitchAnnotationInterceptor dataSourceSwitchAnnotationInterceptor () {
101+ return new DataSourceSwitchAnnotationInterceptor ();
101102 }
102103
103104 /**
104105 * ApiBoot DataSource Switch Advisor
105106 * Used to get @DataSourceSwitch annotation define
106107 *
107- * @param apiBootDataSourceSwitchAnnotationInterceptor ApiBoot DataSource Annotation Interceptor
108+ * @param dataSourceSwitchAnnotationInterceptor ApiBoot DataSource Annotation Interceptor
108109 * @return ApiBootDataSourceSwitchAdvisor
109110 */
110111 @ Bean
111112 @ ConditionalOnMissingBean
112- public ApiBootDataSourceSwitchAdvisor apiBootDataSourceSwitchAdvisor (ApiBootDataSourceSwitchAnnotationInterceptor apiBootDataSourceSwitchAnnotationInterceptor ) {
113- return new ApiBootDataSourceSwitchAdvisor ( apiBootDataSourceSwitchAnnotationInterceptor );
113+ public DataSourceSwitchAdvisor apiBootDataSourceSwitchAdvisor (DataSourceSwitchAnnotationInterceptor dataSourceSwitchAnnotationInterceptor ) {
114+ return new DataSourceSwitchAdvisor ( dataSourceSwitchAnnotationInterceptor );
114115 }
115116
116117}
0 commit comments