78 lines
3.6 KiB
Java
78 lines
3.6 KiB
Java
|
package com.ruoyi.common.utils;
|
||
|
|
|
||
|
|
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.Arrays;
|
||
|
|
import java.util.HashSet;
|
||
|
|
import java.util.List;
|
||
|
|
|
||
|
|
public class MybatisUtils {
|
||
|
|
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 static 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()]);
|
||
|
|
}
|
||
|
|
}
|