SpringMVC的註解機制:Java中利用反射查找使用指定註解的類---找到指定包下的指定註解類

先掃描指定包下的全部的class文件,再找註解是@Controller文件的類文件,再找RequestMapping是/about的方法,並執行

應用:java項目裏用純servlet寫的項目 ,就能夠作本身的註解映射路由了

定義註解

Controller.javahtml

@Target({ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface Controller {
 
}

RequestMapping.javajava

@Target({ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface RequestMapping {
 
  String value();
 
}

使用註解

@Controller
public class IndexController {
 
  @RequestMapping("/")
  public void index() {
    System.out.println("index method")
  }
 
  @RequestMapping("/about")
  public void about(String args) {
    System.out.println("about method")
  }
 
}
@Controller
public class TwoController {

   @RequestMapping("/")
   public void index() {
      System.out.println("index method");
   }

   @RequestMapping("/about")
   public void about(String args) {
      System.out.println("two  about method");
   }

}

掃描包下全部類

public class Scanner {
   /**
    * 從包package中獲取全部的Class
    *
    * @param packageName
    * @return
    */
   public  Set<Class<?>> getClasses(String packageName) throws Exception{

      // 第一個class類的集合
      //List<Class<?>> classes = new ArrayList<Class<?>>();
      Set<Class<?>> classes = new HashSet<>();
      // 是否循環迭代
      boolean recursive = true;
      // 獲取包的名字 並進行替換
      String packageDirName = packageName.replace('.', '/');
      // 定義一個枚舉的集合 並進行循環來處理這個目錄下的things
      Enumeration<URL> dirs;
      try {
         dirs = Thread.currentThread().getContextClassLoader().getResources(packageDirName); // 循環迭代下去
         while (dirs.hasMoreElements()) {
            // 獲取下一個元素,URL的解釋 http://www.javashuo.com/article/p-gnmmjjzx-dg.html  https://www.runoob.com/java/java-url-processing.html
            URL url = dirs.nextElement();
            // 獲得協議的名稱
            String protocol = url.getProtocol();
            // 若是是以文件的形式保存在服務器上
            if ("file".equals(protocol)) {
               // 獲取包的物理路徑
               String filePath = URLDecoder.decode(url.getFile(), "UTF-8");
               // 以文件的方式掃描整個包下的文件 並添加到集合中,如下倆種方法均可以
               //網上的第一種方法,
               findAndAddClassesInPackageByFile(packageName, filePath, recursive, classes);
               //網上的第二種方法
               //addClass(classes,filePath,packageName);
            } else if ("jar".equals(protocol)) {
               // 若是是jar包文件
               // 定義一個JarFile
               JarFile jar;
               try {
                  // 獲取jar
                  jar = ((JarURLConnection) url.openConnection()).getJarFile();
                  // 今後jar包 獲得一個枚舉類
                  Enumeration<JarEntry> entries = jar.entries();
                  // 一樣的進行循環迭代
                  while (entries.hasMoreElements()) {
                     // 獲取jar裏的一個實體 能夠是目錄 和一些jar包裏的其餘文件 如META-INF等文件
                     JarEntry entry = entries.nextElement();
                     String name = entry.getName();
                     // 若是是以/開頭的
                     if (name.charAt(0) == '/') {
                        // 獲取後面的字符串
                        name = name.substring(1);
                     }
                     // 若是前半部分和定義的包名相同
                     if (name.startsWith(packageDirName)) {
                        int idx = name.lastIndexOf('/');
                        // 若是以"/"結尾 是一個包
                        if (idx != -1) {
                           // 獲取包名 把"/"替換成"."
                           packageName = name.substring(0, idx).replace('/', '.');
                        }
                        // 若是能夠迭代下去 而且是一個包
                        if ((idx != -1) || recursive) {
                           // 若是是一個.class文件 並且不是目錄
                           if (name.endsWith(".class") && !entry.isDirectory()) {
                              // 去掉後面的".class" 獲取真正的類名
                              String className = name.substring(packageName.length() + 1, name.length() - 6);
                              try {
                                 // 添加到classes
                                 classes.add(Class.forName(packageName + '.' + className));
                              } catch (ClassNotFoundException e) {
                                 e.printStackTrace();
                              }
                           }
                        }
                     }
                  }
               } catch (IOException e) {
                  e.printStackTrace();
               }
            }
         }
      } catch (IOException e) {
         e.printStackTrace();
      }

      return classes;
   }

   public   void addClass(Set<Class<?>> classes, String filePath, String packageName) throws Exception{
      File[] files=new File(filePath).listFiles(new FileFilter() {
         @Override
         public boolean accept(File file) {
            return (file.isFile()&&file.getName().endsWith(".class"))||file.isDirectory();
         }
      });
      for(File file:files){
         String fileName=file.getName();
         if(file.isFile()){
            String classsName=fileName.substring(0,fileName.lastIndexOf("."));
            if(! StringUtil.isBlank(packageName)){
                       classsName=packageName+"."+classsName;
            }
            doAddClass(classes,classsName);
         }

      }
   }

   public   void doAddClass(Set<Class<?>> classes, final String classsName) throws Exception{
      ClassLoader classLoader=new ClassLoader() {
         @Override
         public Class<?> loadClass(String name) throws ClassNotFoundException {
            return super.loadClass(name);
         }
      };
      //Class<?> cls= ClassLoader.loadClass(classsName);
      classes.add(classLoader.loadClass(classsName));
   }

   //找也用了Controller註解的類
   private Set<Class<?>> controllers;

   public Set<Class<?>> getControllers() throws Exception{
      if (controllers == null) {
         controllers = new HashSet<>();
         Set<Class<?>> clsList = getClasses("aop_zdh.zhujie");
         if (clsList != null && clsList.size() > 0) {
            for (Class<?> cls : clsList) {
               if (cls.getAnnotation(Controller.class) != null) {
                  Map<Class<?>, Object> map = new HashMap<>();
                  controllers.add(cls);
               }
            }
         }
      }
      return controllers;
   }
   public void getMapping() throws Exception{
      for (Class<?> cls : getControllers()) {
         Method[] methods = cls.getMethods();
         for (Method method : methods) {
            RequestMapping annotation = method.getAnnotation(RequestMapping.class);
            if (annotation != null) {
               String value = annotation.value();//找到RequestMapping的注入value值
               if (value.equals("/about")) {//判斷是否是/about,是的話,就調用about(args)方法
                  method.invoke(cls.newInstance(), "args"); //第二個參數是方法裏的參數
               }
            }
         }
      }
   }
   /**
    * 以文件的形式來獲取包下的全部Class
    *
    * @param packageName
    * @param packagePath
    * @param recursive
    * @param classes
    */
   public static void findAndAddClassesInPackageByFile(String packageName,
                                          String packagePath, final boolean recursive, Set<Class<?>> classes) {
      // 獲取此包的目錄 創建一個File
      File dir = new File(packagePath);
      // 若是不存在或者 也不是目錄就直接返回
      if (!dir.exists() || !dir.isDirectory()) {
         // log.warn("用戶定義包名 " + packageName + " 下沒有任何文件");
         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(Class.forName(packageName + '.' + className));
               //通過回覆同窗的提醒,這裏用forName有一些很差,會觸發static方法,沒有使用classLoader的load乾淨
               classes.add(Thread.currentThread().getContextClassLoader().loadClass(packageName + '.' + className));
            } catch (ClassNotFoundException e) {
               // log.error("添加用戶自定義視圖類錯誤 找不到此類的.class文件");
               e.printStackTrace();
            }
         }
      }
   }

   public static void main(String[] args)throws Exception {
      new Scanner().getMapping();
   }
}

運行結果顯示以下:服務器

about method
two  about method

 涉及到的URL知識點見以下兩個大佬縮寫:app

file協議與http協議的區別ide

https://www.cnblogs.com/lishanlei/p/9327646.html
這個是專門介紹URL的
https://www.runoob.com/java/java-url-processing.html
相關文章
相關標籤/搜索