Mybatis 自定义SqlSessionFactoryBean扫描通配符typeAliasesPackage

typeAliasesPackage 默认只能扫描某一个路径下，或以逗号等分割的 几个路径下的内容，不支持通配符和正则，采用重写的方式解决

1. package com.xxxx.xxx.util.common;
2. import com.xxxx.xxx.util.LogUtil;
3. import org.apache.commons.lang3.StringUtils;
4. import org.mybatis.spring.SqlSessionFactoryBean;
5. import org.slf4j.Logger;
6. import org.springframework.core.io.Resource;
7. import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
8. import org.springframework.core.io.support.ResourcePatternResolver;
9. import org.springframework.core.type.classreading.CachingMetadataReaderFactory;
10. import org.springframework.core.type.classreading.MetadataReader;
11. import org.springframework.core.type.classreading.MetadataReaderFactory;
12. import org.springframework.util.ClassUtils;
13. import java.io.IOException;
14. import java.util.ArrayList;
15. import java.util.List;
16. /**
17. * Created by Administrator on 2015/10/6.
18. */
19. public class PackagesSqlSessionFactoryBean extends SqlSessionFactoryBean {
20. static final String DEFAULT_RESOURCE_PATTERN = "**/*.class";
21. private static Logger logger = LogUtil.get();
22. @Override
23. public void setTypeAliasesPackage(String typeAliasesPackage) {
24. ResourcePatternResolver resolver = (ResourcePatternResolver) new PathMatchingResourcePatternResolver();
25. MetadataReaderFactory metadataReaderFactory = new CachingMetadataReaderFactory(resolver);
26. typeAliasesPackage = ResourcePatternResolver.CLASSPATH_ALL_URL_PREFIX +
27. ClassUtils.convertClassNameToResourcePath(typeAliasesPackage) + "/" + DEFAULT_RESOURCE_PATTERN;
28. //将加载多个绝对匹配的所有Resource
29. //将首先通过ClassLoader.getResource("META-INF")加载非模式路径部分
30. //然后进行遍历模式匹配
31. try {
32. List<String> result = new ArrayList<String>();
33. Resource[] resources =  resolver.getResources(typeAliasesPackage);
34. if(resources != null && resources.length > 0){
35. MetadataReader metadataReader = null;
36. for(Resource resource : resources){
37. if(resource.isReadable()){
38. metadataReader =  metadataReaderFactory.getMetadataReader(resource);
39. try {
40. result.add(Class.forName(metadataReader.getClassMetadata().getClassName()).getPackage().getName());
41. } catch (ClassNotFoundException e) {
42. e.printStackTrace();
43. }
44. }
45. }
46. }
47. if(result.size() > 0) {
48. super.setTypeAliasesPackage(StringUtils.join(result.toArray(), ","));
49. }else{
50. logger.warn("参数typeAliasesPackage:"+typeAliasesPackage+"，未找到任何包");
51. }
52. //logger.info("d");
53. } catch (IOException e) {
54. e.printStackTrace();
55. }
56. }
57. }

package com.xxxx.xxx.util.common;  

import com.xxxx.xxx.util.LogUtil;
import org.apache.commons.lang3.StringUtils;
import org.mybatis.spring.SqlSessionFactoryBean;
import org.slf4j.Logger;
import org.springframework.core.io.Resource;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.core.io.support.ResourcePatternResolver;
import org.springframework.core.type.classreading.CachingMetadataReaderFactory;
import org.springframework.core.type.classreading.MetadataReader;
import org.springframework.core.type.classreading.MetadataReaderFactory;
import org.springframework.util.ClassUtils;  

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;  

/**
 * Created by Administrator on 2015/10/6.
 */
public class PackagesSqlSessionFactoryBean extends SqlSessionFactoryBean {  

    static final String DEFAULT_RESOURCE_PATTERN = "**/*.class";  

    private static Logger logger = LogUtil.get();  

    @Override
    public void setTypeAliasesPackage(String typeAliasesPackage) {
        ResourcePatternResolver resolver = (ResourcePatternResolver) new PathMatchingResourcePatternResolver();
        MetadataReaderFactory metadataReaderFactory = new CachingMetadataReaderFactory(resolver);
        typeAliasesPackage = ResourcePatternResolver.CLASSPATH_ALL_URL_PREFIX +
                ClassUtils.convertClassNameToResourcePath(typeAliasesPackage) + "/" + DEFAULT_RESOURCE_PATTERN;  

        //将加载多个绝对匹配的所有Resource
        //将首先通过ClassLoader.getResource("META-INF")加载非模式路径部分
        //然后进行遍历模式匹配
        try {
            List<String> result = new ArrayList<String>();
            Resource[] resources =  resolver.getResources(typeAliasesPackage);
            if(resources != null && resources.length > 0){
                MetadataReader metadataReader = null;
                for(Resource resource : resources){
                    if(resource.isReadable()){
                       metadataReader =  metadataReaderFactory.getMetadataReader(resource);
                        try {
                            result.add(Class.forName(metadataReader.getClassMetadata().getClassName()).getPackage().getName());
                        } catch (ClassNotFoundException e) {
                            e.printStackTrace();
                        }
                    }
                }
            }
            if(result.size() > 0) {
                super.setTypeAliasesPackage(StringUtils.join(result.toArray(), ","));
            }else{
                logger.warn("参数typeAliasesPackage:"+typeAliasesPackage+"，未找到任何包");
            }
            //logger.info("d");
        } catch (IOException e) {
            e.printStackTrace();
        }
    }  

}  

然后在mybatis的配置文件中修改

1. <bean id="sqlSession" class="com.xxxx.xxxx.util.common.PackagesSqlSessionFactoryBean">
2. <property name="configLocation" value="classpath:config/sqlmap/sqlmap-config.xml" />
3. <property name="dataSource" ref="dataSource"/>
4. <!--<property name="mapperLocations"-->
5. <!--value="classpath*:com/xxxx/xxxx/merchant/**/domain/mapper/*.xml"/>-->
6. <property name="typeAliasesPackage" value="com.xxxx.xxxx.custom.*.domain"/>
7. <property name="plugins">
8. <array>
9. <ref bean="pageInterceptor"/>
10. </array>
11. </property>
12. </bean>
13. <bean class="org.mybatis.spring.mapper.MapperScannerConfigurer">
14. <property name="basePackage" value="com.xxxx.xxxx.**.dao"/>
15. </bean>

<bean id="sqlSession" class="com.xxxx.xxxx.util.common.PackagesSqlSessionFactoryBean">
        <property name="configLocation" value="classpath:config/sqlmap/sqlmap-config.xml" />
        <property name="dataSource" ref="dataSource"/>
        <!--<property name="mapperLocations"-->
                  <!--value="classpath*:com/xxxx/xxxx/merchant/**/domain/mapper/*.xml"/>-->
        <property name="typeAliasesPackage" value="com.xxxx.xxxx.custom.*.domain"/>
        <property name="plugins">
            <array>
                <ref bean="pageInterceptor"/>
            </array>
        </property>
    </bean>
    <bean class="org.mybatis.spring.mapper.MapperScannerConfigurer">
        <property name="basePackage" value="com.xxxx.xxxx.**.dao"/>
    </bean>  

来源：http://blog.csdn.net/bian1729183741/article/details/52193882

来自为知笔记(Wiz)