|
@@ -1,109 +1,109 @@
|
|
-package com.jjt.framework.config;
|
|
|
|
-
|
|
|
|
-import com.jjt.common.utils.StringUtils;
|
|
|
|
-import org.apache.ibatis.io.VFS;
|
|
|
|
-import org.apache.ibatis.session.SqlSessionFactory;
|
|
|
|
-import org.mybatis.spring.SqlSessionFactoryBean;
|
|
|
|
-import org.mybatis.spring.boot.autoconfigure.SpringBootVFS;
|
|
|
|
-import org.springframework.context.annotation.Bean;
|
|
|
|
-import org.springframework.context.annotation.Configuration;
|
|
|
|
-import org.springframework.core.env.Environment;
|
|
|
|
-import org.springframework.core.io.DefaultResourceLoader;
|
|
|
|
-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 javax.sql.DataSource;
|
|
|
|
-import java.io.IOException;
|
|
|
|
-import java.util.ArrayList;
|
|
|
|
-import java.util.Arrays;
|
|
|
|
-import java.util.HashSet;
|
|
|
|
-import java.util.List;
|
|
|
|
-
|
|
|
|
-/**
|
|
|
|
- * Mybatis支持*匹配扫描包
|
|
|
|
- *
|
|
|
|
- * @author jjt
|
|
|
|
- */
|
|
|
|
-@Configuration
|
|
|
|
-public class MyBatisConfig {
|
|
|
|
- @javax.annotation.Resource
|
|
|
|
- private Environment env;
|
|
|
|
-
|
|
|
|
- static final String DEFAULT_RESOURCE_PATTERN = "**/*.class";
|
|
|
|
-
|
|
|
|
- public static String setTypeAliasesPackage(String typeAliasesPackage) {
|
|
|
|
- ResourcePatternResolver resolver = (ResourcePatternResolver) new PathMatchingResourcePatternResolver();
|
|
|
|
- MetadataReaderFactory metadataReaderFactory = new CachingMetadataReaderFactory(resolver);
|
|
|
|
- List<String> allResult = new ArrayList<String>();
|
|
|
|
- try {
|
|
|
|
- for (String aliasesPackage : typeAliasesPackage.split(",")) {
|
|
|
|
- List<String> result = new ArrayList<String>();
|
|
|
|
- aliasesPackage = ResourcePatternResolver.CLASSPATH_ALL_URL_PREFIX
|
|
|
|
- + ClassUtils.convertClassNameToResourcePath(aliasesPackage.trim()) + "/" + DEFAULT_RESOURCE_PATTERN;
|
|
|
|
- Resource[] resources = resolver.getResources(aliasesPackage);
|
|
|
|
- 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) {
|
|
|
|
- HashSet<String> hashResult = new HashSet<String>(result);
|
|
|
|
- allResult.addAll(hashResult);
|
|
|
|
- }
|
|
|
|
- }
|
|
|
|
- if (allResult.size() > 0) {
|
|
|
|
- typeAliasesPackage = String.join(",", (String[]) allResult.toArray(new String[0]));
|
|
|
|
- } else {
|
|
|
|
- throw new RuntimeException("mybatis typeAliasesPackage 路径扫描错误,参数typeAliasesPackage:" + typeAliasesPackage + "未找到任何包");
|
|
|
|
- }
|
|
|
|
- } catch (IOException e) {
|
|
|
|
- e.printStackTrace();
|
|
|
|
- }
|
|
|
|
- return typeAliasesPackage;
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- public Resource[] resolveMapperLocations(String[] mapperLocations) {
|
|
|
|
- ResourcePatternResolver resourceResolver = new PathMatchingResourcePatternResolver();
|
|
|
|
- List<Resource> resources = new ArrayList<Resource>();
|
|
|
|
- if (mapperLocations != null) {
|
|
|
|
- for (String mapperLocation : mapperLocations) {
|
|
|
|
- try {
|
|
|
|
- Resource[] mappers = resourceResolver.getResources(mapperLocation);
|
|
|
|
- resources.addAll(Arrays.asList(mappers));
|
|
|
|
- } catch (IOException e) {
|
|
|
|
- // ignore
|
|
|
|
- }
|
|
|
|
- }
|
|
|
|
- }
|
|
|
|
- return resources.toArray(new Resource[resources.size()]);
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- @Bean
|
|
|
|
- public SqlSessionFactory sqlSessionFactory(DataSource dataSource) throws Exception {
|
|
|
|
- String typeAliasesPackage = env.getProperty("mybatis.typeAliasesPackage");
|
|
|
|
- String mapperLocations = env.getProperty("mybatis.mapperLocations");
|
|
|
|
- String configLocation = env.getProperty("mybatis.configLocation");
|
|
|
|
- typeAliasesPackage = setTypeAliasesPackage(typeAliasesPackage);
|
|
|
|
- VFS.addImplClass(SpringBootVFS.class);
|
|
|
|
-
|
|
|
|
- final SqlSessionFactoryBean sessionFactory = new SqlSessionFactoryBean();
|
|
|
|
- sessionFactory.setDataSource(dataSource);
|
|
|
|
- sessionFactory.setTypeAliasesPackage(typeAliasesPackage);
|
|
|
|
- sessionFactory.setMapperLocations(resolveMapperLocations(StringUtils.split(mapperLocations, ",")));
|
|
|
|
- sessionFactory.setConfigLocation(new DefaultResourceLoader().getResource(configLocation));
|
|
|
|
- return sessionFactory.getObject();
|
|
|
|
- }
|
|
|
|
-}
|
|
|
|
|
|
+//package com.jjt.framework.config;
|
|
|
|
+//
|
|
|
|
+//import com.jjt.common.utils.StringUtils;
|
|
|
|
+//import org.apache.ibatis.io.VFS;
|
|
|
|
+//import org.apache.ibatis.session.SqlSessionFactory;
|
|
|
|
+//import org.mybatis.spring.SqlSessionFactoryBean;
|
|
|
|
+//import org.mybatis.spring.boot.autoconfigure.SpringBootVFS;
|
|
|
|
+//import org.springframework.context.annotation.Bean;
|
|
|
|
+//import org.springframework.context.annotation.Configuration;
|
|
|
|
+//import org.springframework.core.env.Environment;
|
|
|
|
+//import org.springframework.core.io.DefaultResourceLoader;
|
|
|
|
+//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 javax.sql.DataSource;
|
|
|
|
+//import java.io.IOException;
|
|
|
|
+//import java.util.ArrayList;
|
|
|
|
+//import java.util.Arrays;
|
|
|
|
+//import java.util.HashSet;
|
|
|
|
+//import java.util.List;
|
|
|
|
+//
|
|
|
|
+///**
|
|
|
|
+// * Mybatis支持*匹配扫描包
|
|
|
|
+// *
|
|
|
|
+// * @author jjt
|
|
|
|
+// */
|
|
|
|
+//@Configuration
|
|
|
|
+//public class MyBatisConfig {
|
|
|
|
+// @javax.annotation.Resource
|
|
|
|
+// private Environment env;
|
|
|
|
+//
|
|
|
|
+// static final String DEFAULT_RESOURCE_PATTERN = "**/*.class";
|
|
|
|
+//
|
|
|
|
+// public static String setTypeAliasesPackage(String typeAliasesPackage) {
|
|
|
|
+// ResourcePatternResolver resolver = (ResourcePatternResolver) new PathMatchingResourcePatternResolver();
|
|
|
|
+// MetadataReaderFactory metadataReaderFactory = new CachingMetadataReaderFactory(resolver);
|
|
|
|
+// List<String> allResult = new ArrayList<String>();
|
|
|
|
+// try {
|
|
|
|
+// for (String aliasesPackage : typeAliasesPackage.split(",")) {
|
|
|
|
+// List<String> result = new ArrayList<String>();
|
|
|
|
+// aliasesPackage = ResourcePatternResolver.CLASSPATH_ALL_URL_PREFIX
|
|
|
|
+// + ClassUtils.convertClassNameToResourcePath(aliasesPackage.trim()) + "/" + DEFAULT_RESOURCE_PATTERN;
|
|
|
|
+// Resource[] resources = resolver.getResources(aliasesPackage);
|
|
|
|
+// 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) {
|
|
|
|
+// HashSet<String> hashResult = new HashSet<String>(result);
|
|
|
|
+// allResult.addAll(hashResult);
|
|
|
|
+// }
|
|
|
|
+// }
|
|
|
|
+// if (allResult.size() > 0) {
|
|
|
|
+// typeAliasesPackage = String.join(",", (String[]) allResult.toArray(new String[0]));
|
|
|
|
+// } else {
|
|
|
|
+// throw new RuntimeException("mybatis typeAliasesPackage 路径扫描错误,参数typeAliasesPackage:" + typeAliasesPackage + "未找到任何包");
|
|
|
|
+// }
|
|
|
|
+// } catch (IOException e) {
|
|
|
|
+// e.printStackTrace();
|
|
|
|
+// }
|
|
|
|
+// return typeAliasesPackage;
|
|
|
|
+// }
|
|
|
|
+//
|
|
|
|
+// public Resource[] resolveMapperLocations(String[] mapperLocations) {
|
|
|
|
+// ResourcePatternResolver resourceResolver = new PathMatchingResourcePatternResolver();
|
|
|
|
+// List<Resource> resources = new ArrayList<Resource>();
|
|
|
|
+// if (mapperLocations != null) {
|
|
|
|
+// for (String mapperLocation : mapperLocations) {
|
|
|
|
+// try {
|
|
|
|
+// Resource[] mappers = resourceResolver.getResources(mapperLocation);
|
|
|
|
+// resources.addAll(Arrays.asList(mappers));
|
|
|
|
+// } catch (IOException e) {
|
|
|
|
+// // ignore
|
|
|
|
+// }
|
|
|
|
+// }
|
|
|
|
+// }
|
|
|
|
+// return resources.toArray(new Resource[resources.size()]);
|
|
|
|
+// }
|
|
|
|
+//
|
|
|
|
+// @Bean
|
|
|
|
+// public SqlSessionFactory sqlSessionFactory(DataSource dataSource) throws Exception {
|
|
|
|
+// String typeAliasesPackage = env.getProperty("mybatis.typeAliasesPackage");
|
|
|
|
+// String mapperLocations = env.getProperty("mybatis.mapperLocations");
|
|
|
|
+// String configLocation = env.getProperty("mybatis.configLocation");
|
|
|
|
+// typeAliasesPackage = setTypeAliasesPackage(typeAliasesPackage);
|
|
|
|
+// VFS.addImplClass(SpringBootVFS.class);
|
|
|
|
+//
|
|
|
|
+// final SqlSessionFactoryBean sessionFactory = new SqlSessionFactoryBean();
|
|
|
|
+// sessionFactory.setDataSource(dataSource);
|
|
|
|
+// sessionFactory.setTypeAliasesPackage(typeAliasesPackage);
|
|
|
|
+// sessionFactory.setMapperLocations(resolveMapperLocations(StringUtils.split(mapperLocations, ",")));
|
|
|
|
+// sessionFactory.setConfigLocation(new DefaultResourceLoader().getResource(configLocation));
|
|
|
|
+// return sessionFactory.getObject();
|
|
|
|
+// }
|
|
|
|
+//}
|