nbcp.myoql.db.mybatis.MybatisDbConfig.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ktmyoql Show documentation
kotlin orm -- mysql,mongo , just like ktorm
The newest version!
package nbcp.myoql.db.mybatis

import nbcp.base.comm.config
import nbcp.base.utils.SpringUtil
import nbcp.myoql.db.db
import org.apache.ibatis.session.SqlSessionFactory
import org.mybatis.spring.SqlSessionFactoryBean
import org.mybatis.spring.SqlSessionTemplate
import org.mybatis.spring.mapper.MapperScannerConfigurer
import org.springframework.boot.autoconfigure.condition.ConditionalOnClass
import org.springframework.boot.autoconfigure.jdbc.DataSourceAutoConfiguration
import org.springframework.boot.context.event.ApplicationPreparedEvent
import org.springframework.context.event.EventListener
import org.springframework.stereotype.Component
import org.springframework.transaction.annotation.EnableTransactionManagement
import javax.sql.DataSource

@EnableTransactionManagement
@Component
@ConditionalOnClass(value = [MapperScannerConfigurer::class])
class MybatisDbConfig {

    @EventListener
    fun prepared(ev: ApplicationPreparedEvent) {
        if (SpringUtil.context.environment.getProperty("spring.datasource.url").isNullOrEmpty() &&
            SpringUtil.context.environment.getProperty("spring.datasource.hikari.jdbc-url").isNullOrEmpty()
        ) {
            return;
        }

        if (config.mybatisPackage.isEmpty()) {
            return;
        }

        if (SpringUtil.containsBean(DataSourceAutoConfiguration::class.java) == false) {
            return;
        }


        SpringUtil.registerBeanDefinition(mybatisMapperScannerConfigurer())
        SpringUtil.registerBeanDefinition(MyBatisTransactionManagementConfig())

//        var sqlSessionFactoryBean = mybatisSessionFactoryBean();
//        if (sqlSessionFactoryBean != null) {
//            SpringUtil.registerBeanDefinition(sqlSessionFactoryBean)
//            SpringUtil.registerBeanDefinition(mybatisSessionTemplate(sqlSessionFactoryBean))
//        }
    }


    val dataSource: DataSource
        get() {
            return db.sql.getScopeDataSource()!!;
        }


    fun mybatisMapperScannerConfigurer(): MapperScannerConfigurer {
        val mapperScannerConfigurer = MapperScannerConfigurer()
        //获取之前注入的beanName为sqlSessionFactory的对象
        mapperScannerConfigurer.setSqlSessionFactoryBeanName("sqlSessionFactory")
        //指定xml配置文件的路径
        mapperScannerConfigurer.setBasePackage(config.mybatisPackage)
        return mapperScannerConfigurer
    }


//    fun mybatisSessionFactoryBean(): SqlSessionFactory {
//        val bean = SqlSessionFactoryBean()
//        bean.setDataSource(dataSource)
//
//        var config = org.apache.ibatis.session.Configuration()
//        config.isCacheEnabled = true
//
////        config.addCache(RedisCacheMyBatis())
//
////        config.addInterceptor(QueryInterceptor())
////        config.addInterceptor(UpdateInterceptor())
//        config.addInterceptor(MyBatisInterceptor())
//        bean.setConfiguration(config)
//
//        return bean.`object`
//    }
//
//    fun mybatisSessionTemplate(sqlSessionFactory: SqlSessionFactory): SqlSessionTemplate {
//        return SqlSessionTemplate(sqlSessionFactory)
//    }
}