1717
1818package org .apache .dolphinscheduler .plugin .registry .jdbc ;
1919
20- import org .apache .dolphinscheduler .plugin .registry .jdbc .mapper .JdbcRegistryClientHeartbeatMapper ;
21- import org .apache .dolphinscheduler .plugin .registry .jdbc .mapper .JdbcRegistryDataChangeEventMapper ;
22- import org .apache .dolphinscheduler .plugin .registry .jdbc .mapper .JdbcRegistryDataMapper ;
23- import org .apache .dolphinscheduler .plugin .registry .jdbc .mapper .JdbcRegistryLockMapper ;
2420import org .apache .dolphinscheduler .plugin .registry .jdbc .repository .JdbcRegistryClientRepository ;
2521import org .apache .dolphinscheduler .plugin .registry .jdbc .repository .JdbcRegistryDataChangeEventRepository ;
2622import org .apache .dolphinscheduler .plugin .registry .jdbc .repository .JdbcRegistryDataRepository ;
2723import org .apache .dolphinscheduler .plugin .registry .jdbc .repository .JdbcRegistryLockRepository ;
2824import org .apache .dolphinscheduler .plugin .registry .jdbc .server .IJdbcRegistryServer ;
2925import org .apache .dolphinscheduler .plugin .registry .jdbc .server .JdbcRegistryServer ;
3026
31- import org .apache .ibatis .session .SqlSessionFactory ;
32-
3327import lombok .extern .slf4j .Slf4j ;
3428
35- import org .mybatis .spring .SqlSessionTemplate ;
3629import org .mybatis .spring .annotation .MapperScan ;
3730import org .springframework .boot .autoconfigure .AutoConfigureAfter ;
38- import org .springframework .boot .autoconfigure .condition .ConditionalOnMissingBean ;
3931import org .springframework .boot .autoconfigure .condition .ConditionalOnProperty ;
4032import org .springframework .context .annotation .Bean ;
4133import org .springframework .context .annotation .ComponentScan ;
4234import org .springframework .context .annotation .Configuration ;
35+ import org .springframework .transaction .support .TransactionTemplate ;
4336
4437import com .baomidou .mybatisplus .autoconfigure .MybatisPlusAutoConfiguration ;
45- import com .baomidou .mybatisplus .extension .spring .MybatisSqlSessionFactoryBean ;
46- import com .zaxxer .hikari .HikariDataSource ;
4738
4839@ Slf4j
4940@ ComponentScan
@@ -62,13 +53,15 @@ public IJdbcRegistryServer jdbcRegistryServer(JdbcRegistryDataRepository jdbcReg
6253 JdbcRegistryLockRepository jdbcRegistryLockRepository ,
6354 JdbcRegistryClientRepository jdbcRegistryClientRepository ,
6455 JdbcRegistryDataChangeEventRepository jdbcRegistryDataChangeEventRepository ,
65- JdbcRegistryProperties jdbcRegistryProperties ) {
56+ JdbcRegistryProperties jdbcRegistryProperties ,
57+ TransactionTemplate transactionTemplate ) {
6658 return new JdbcRegistryServer (
6759 jdbcRegistryDataRepository ,
6860 jdbcRegistryLockRepository ,
6961 jdbcRegistryClientRepository ,
7062 jdbcRegistryDataChangeEventRepository ,
71- jdbcRegistryProperties );
63+ jdbcRegistryProperties ,
64+ transactionTemplate );
7265 }
7366
7467 @ Bean
@@ -79,44 +72,4 @@ public JdbcRegistry jdbcRegistry(JdbcRegistryProperties jdbcRegistryProperties,
7972 return jdbcRegistry ;
8073 }
8174
82- @ Bean
83- @ ConditionalOnMissingBean
84- public SqlSessionFactory sqlSessionFactory (JdbcRegistryProperties jdbcRegistryProperties ) throws Exception {
85- log .info ("Initialize jdbcRegistrySqlSessionFactory" );
86- MybatisSqlSessionFactoryBean sqlSessionFactoryBean = new MybatisSqlSessionFactoryBean ();
87- sqlSessionFactoryBean .setDataSource (new HikariDataSource (jdbcRegistryProperties .getHikariConfig ()));
88- return sqlSessionFactoryBean .getObject ();
89- }
90-
91- @ Bean
92- @ ConditionalOnMissingBean
93- public SqlSessionTemplate sqlSessionTemplate (SqlSessionFactory jdbcRegistrySqlSessionFactory ) {
94- log .info ("Initialize jdbcRegistrySqlSessionTemplate" );
95- return new SqlSessionTemplate (jdbcRegistrySqlSessionFactory );
96- }
97-
98- @ Bean
99- public JdbcRegistryDataMapper jdbcRegistryDataMapper (SqlSessionTemplate jdbcRegistrySqlSessionTemplate ) {
100- jdbcRegistrySqlSessionTemplate .getConfiguration ().addMapper (JdbcRegistryDataMapper .class );
101- return jdbcRegistrySqlSessionTemplate .getMapper (JdbcRegistryDataMapper .class );
102- }
103-
104- @ Bean
105- public JdbcRegistryLockMapper jdbcRegistryLockMapper (SqlSessionTemplate jdbcRegistrySqlSessionTemplate ) {
106- jdbcRegistrySqlSessionTemplate .getConfiguration ().addMapper (JdbcRegistryLockMapper .class );
107- return jdbcRegistrySqlSessionTemplate .getMapper (JdbcRegistryLockMapper .class );
108- }
109-
110- @ Bean
111- public JdbcRegistryDataChangeEventMapper jdbcRegistryDataChangeEventMapper (SqlSessionTemplate jdbcRegistrySqlSessionTemplate ) {
112- jdbcRegistrySqlSessionTemplate .getConfiguration ().addMapper (JdbcRegistryDataChangeEventMapper .class );
113- return jdbcRegistrySqlSessionTemplate .getMapper (JdbcRegistryDataChangeEventMapper .class );
114- }
115-
116- @ Bean
117- public JdbcRegistryClientHeartbeatMapper jdbcRegistryClientHeartbeatMapper (SqlSessionTemplate jdbcRegistrySqlSessionTemplate ) {
118- jdbcRegistrySqlSessionTemplate .getConfiguration ().addMapper (JdbcRegistryClientHeartbeatMapper .class );
119- return jdbcRegistrySqlSessionTemplate .getMapper (JdbcRegistryClientHeartbeatMapper .class );
120- }
121-
12275}
0 commit comments