1
1
package org .minbox .framework .api .boot .autoconfigure .datasource ;
2
2
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 ;
12
13
import org .springframework .boot .autoconfigure .AutoConfigureBefore ;
13
14
import org .springframework .boot .autoconfigure .condition .ConditionalOnClass ;
14
15
import org .springframework .boot .autoconfigure .condition .ConditionalOnMissingBean ;
30
31
* @author 恒宇少年
31
32
*/
32
33
@ Configuration
33
- @ ConditionalOnClass ({ApiBootDataSource .class , AbstractRoutingDataSource .class })
34
+ @ ConditionalOnClass ({MinBoxDataSource .class , AbstractRoutingDataSource .class })
34
35
@ EnableConfigurationProperties (ApiBootDataSourceSwitchProperties .class )
35
36
@ AutoConfigureBefore (DataSourceAutoConfiguration .class )
36
37
public class ApiBootDataSourceSwitchAutoConfiguration {
@@ -51,21 +52,21 @@ public ApiBootDataSourceSwitchAutoConfiguration(ApiBootDataSourceSwitchPropertie
51
52
*/
52
53
@ Bean
53
54
@ ConditionalOnMissingBean
54
- public ApiBootDataSourceFactoryBean apiBootDataSourceFactoryBean () {
55
- return new ApiBootDataSourceFactoryBean ();
55
+ public DataSourceFactoryBean dataSourceFactoryBean () {
56
+ return new DataSourceFactoryBean ();
56
57
}
57
58
58
59
/**
59
60
* ApiBoot Routing DataSource
60
61
* switch use datasource
61
62
* {@link DataSource}
62
63
*
63
- * @param apiBootDataSourceFactoryBean ApiBoot DataSource FactoryBean
64
+ * @param dataSourceFactoryBean ApiBoot DataSource FactoryBean
64
65
* @return DataSource
65
66
*/
66
67
@ Bean
67
68
@ ConditionalOnMissingBean
68
- public DataSource dataSource (ApiBootDataSourceFactoryBean apiBootDataSourceFactoryBean ) {
69
+ public DataSource dataSource (DataSourceFactoryBean dataSourceFactoryBean ) {
69
70
List <DataSourceConfig > dataSourceConfigList = new LinkedList ();
70
71
Map <String , DataSourceConfig > dataSourceConfigMap = new HashMap (1 );
71
72
@@ -80,13 +81,13 @@ public DataSource dataSource(ApiBootDataSourceFactoryBean apiBootDataSourceFacto
80
81
// set data source pool name
81
82
dataSourceConfig .setPoolName (poolName );
82
83
// datasource type
83
- dataSourceConfig .setDataSourceType (dataSourceConfig instanceof DataSourceDruidConfig ? ApiBootDruidDataSource .class : ApiBootHikariDataSource .class );
84
+ dataSourceConfig .setDataSourceType (dataSourceConfig instanceof DataSourceDruidConfig ? MinBoxDruidDataSource .class : MinBoxHikariDataSource .class );
84
85
85
86
// after convert add to data source list
86
87
dataSourceConfigList .add (dataSourceConfig );
87
88
});
88
89
89
- return new ApiBootRoutingDataSource (apiBootDataSourceFactoryBean , apiBootDataSourceSwitchProperties .getPrimary (), dataSourceConfigList );
90
+ return new ApiBootRoutingDataSource (dataSourceFactoryBean , apiBootDataSourceSwitchProperties .getPrimary (), dataSourceConfigList );
90
91
}
91
92
92
93
/**
@@ -96,21 +97,21 @@ public DataSource dataSource(ApiBootDataSourceFactoryBean apiBootDataSourceFacto
96
97
*/
97
98
@ Bean
98
99
@ ConditionalOnMissingBean
99
- public ApiBootDataSourceSwitchAnnotationInterceptor apiBootDataSourceSwitchAnnotationInterceptor () {
100
- return new ApiBootDataSourceSwitchAnnotationInterceptor ();
100
+ public DataSourceSwitchAnnotationInterceptor dataSourceSwitchAnnotationInterceptor () {
101
+ return new DataSourceSwitchAnnotationInterceptor ();
101
102
}
102
103
103
104
/**
104
105
* ApiBoot DataSource Switch Advisor
105
106
* Used to get @DataSourceSwitch annotation define
106
107
*
107
- * @param apiBootDataSourceSwitchAnnotationInterceptor ApiBoot DataSource Annotation Interceptor
108
+ * @param dataSourceSwitchAnnotationInterceptor ApiBoot DataSource Annotation Interceptor
108
109
* @return ApiBootDataSourceSwitchAdvisor
109
110
*/
110
111
@ Bean
111
112
@ ConditionalOnMissingBean
112
- public ApiBootDataSourceSwitchAdvisor apiBootDataSourceSwitchAdvisor (ApiBootDataSourceSwitchAnnotationInterceptor apiBootDataSourceSwitchAnnotationInterceptor ) {
113
- return new ApiBootDataSourceSwitchAdvisor ( apiBootDataSourceSwitchAnnotationInterceptor );
113
+ public DataSourceSwitchAdvisor apiBootDataSourceSwitchAdvisor (DataSourceSwitchAnnotationInterceptor dataSourceSwitchAnnotationInterceptor ) {
114
+ return new DataSourceSwitchAdvisor ( dataSourceSwitchAnnotationInterceptor );
114
115
}
115
116
116
117
}
0 commit comments