扫描注解,添加到权限表中
本文所写的是一个扫描注解的工具类,减少了更改注解时数据库权限表中的数据修改。避免了很多不必要的麻烦。
本文主要用到了反射,通过反射得到每个注解的值,得到注解的值后,保存到数据库中,节省了很多时间。
// 使用说明:传入一个类似于“com.xxx.controller”的包名,就会搜索这个包下的所有类的RequestMapping注解里的值,然后放到集合中输出
// 代码如下
import java.io.File;
import java.io.FileFilter;
import java.io.IOException;
import java.lang.reflect.Method;
import java.net.URL;
import java.net.URLDecoder;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.List;
import org.springframework.web.bind.annotation.RequestMapping;
/**
*
* Title: GetAnnotationValueUtil.java
* Description: 获取注解的值
* @author 一身傲骨
* @created 2016年6月14日 下午7:17:09
*/
public class GetAnnotationValueUtil {
/**
*
* @description 获取注解的值
* @author :一身傲骨
* @created 2016年6月14日 下午7:17:27
* @param packageName
* @return
*/
public static List<String> getRequestMappingValue(String packageName) {
GetAnnotationValueUtil getAnnotationValueUtil = new GetAnnotationValueUtil();
//第一个class类的集合
List<Class<?>> classes = new ArrayList<Class<?>>();
//是否循环迭代
boolean recursive = true;
//获取包的名字 并进行替换
String packageDirName = packageName.replace('.', '/');
//定义一个枚举的集合 并进行循环来处理这个目录下的文件
Enumeration<URL> dirs;
try {
//读取指定package下的所有class
dirs = Thread.currentThread().getContextClassLoader().getResources(packageDirName);
while (dirs.hasMoreElements()){
URL url = dirs.nextElement();
//得到协议的名称
String protocol = url.getProtocol();
//判断是否以文件的形式保存在服务器上
if ("file".equals(protocol)) {
//获取包的物理路径
String filePath = URLDecoder.decode(url.getFile(), "UTF-8");
//以文件的方式扫描整个包下的文件 并添加到集合中
getAnnotationValueUtil.findAndAddClassesInPackageByFile(packageName, filePath, recursive, classes);
}
}
} catch (IOException e) {
e.printStackTrace();
}
List<String> stringList = new ArrayList<String>();
for (Class<?> clazz : classes) {
//循环获取所有的类
Class<?> c = clazz;
//获取类的所有方法
Method[] methods = c.getMethods();
for (Method method : methods) {
//获取RequestMapping注解
RequestMapping annotation = method.getAnnotation(RequestMapping.class);
if (annotation != null) {
//获取注解的value值
String[] value = annotation.value();
for (String string : value) {
//放入List集合
stringList.add(string);
}
}
}
}
return stringList;
}
/**
*
* @description 获取包下的所有类
* @author 一身傲骨
* @created 2016年6月14日 下午5:17:40
* @param packageName
* @param packagePath
* @param recursive
* @param classes
*/
public void findAndAddClassesInPackageByFile(String packageName, String packagePath, final boolean recursive, List<Class<?>> classes){
//获取此包的目录 建立一个File
File dir = new File(packagePath);
//如果不存在或者 也不是目录就直接返回
if (!dir.exists() || !dir.isDirectory()) {
return;
}
//如果存在 就获取包下的所有文件 包括目录
File[] dirfiles = dir.listFiles(new FileFilter() {
//自定义过滤规则 如果可以循环(包含子目录) 或则是以.class结尾的文件(编译好的java类文件)
public boolean accept(File file) {
return (recursive && file.isDirectory()) || (file.getName().endsWith(".class"));
}
});
//循环所有文件
for (File file : dirfiles) {
//如果是目录 则继续扫描
if (file.isDirectory()) {
findAndAddClassesInPackageByFile(packageName + "." + file.getName(),
file.getAbsolutePath(),
recursive,
classes);
}
else {
//如果是java类文件 去掉后面的.class 只留下类名
String className = file.getName().substring(0, file.getName().length() - 6);
try {
//添加到集合中去
classes.add(Thread.currentThread().getContextClassLoader().loadClass(packageName + "." + className));
} catch (ClassNotFoundException e) {
e.printStackTrace();
}
}
}
}
private GetAnnotationValueUtil() {
}
}