사용자 정의 주석 프로세서 - 주석을 사용하여 메서드 검색

Question

@PrintMethod 주석을 사용하여 주석이 달린 메서드를 검색하려면 주석 Procssor를 작성하려고합니다. 아래의 테스트 클래스 예를 들어, 테스트 메서드 내에서 코드를 인쇄 할 싶습니다. 그것을 할 수있는 방법이 있습니까?

아래 명시된 AnnotationProcessor 클래스에서 메서드 이름 만 가져올 수 있지만 메서드의 세부 정보는 가져올 수 없습니다.

테스트 클래스

public class test { 

    public static void main(String[] args) { 
     System.out.println("Args"); 
    } 

    @PrintMethod 
    private boolean testMethod(String input) { 
     if(input!=null) { 
      return true; 
     } 
     return false; 
    } 
} 

주석 프로세서 클래스 이것에 대해 궁금

public class AnnotationProcessor extends AbstractProcessor { 
//...... 
    @Override 
    public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) { 
     //retrieve test Anntoation 
     Set<? extends Element> ann =roundEnv.getElementsAnnotatedWith(PrintMethod.class); 

     //Print the Method Name 
     for(Element e: ann) { 
      String msg="Element ee :"+ee.getSimpleName().toString(); 
      processingEnv.getMessager().printMessage(javax.tools.Diagnostic.Kind.ERROR, msg, e); 
     } 
    } 
} 

Answer 1

너무 그래서 나는 시도하고 그것을 알아낼하기로 결정했다. 내가 기대했던 것보다 쉬워 지네. 독점적 인 tools.jar 라이브러리에서 Trees API를 활용하면됩니다.

@SupportedSourceVersion(SourceVersion.RELEASE_6) 
@SupportedAnnotationTypes("org.printMethod.PrintMethod") 
public class PrintMethodAnnotationProcessor extends AbstractProcessor { 

    private Trees trees; 

    @Override 
    public synchronized void init(ProcessingEnvironment processingEnv) { 
     super.init(processingEnv); 
     trees = Trees.instance(processingEnv); //initialize the Trees api. 
    } 

    @Override 
    public boolean process(Set<? extends TypeElement> typeElements, RoundEnvironment roundEnvironment) { 

     MethodPrintScanner visitor = new MethodPrintScanner(); 

     for (Element e : roundEnvironment.getElementsAnnotatedWith(PrintMethod.class)) { 
      TreePath tp = trees.getPath(e); 
      // visit the annotated methods 
      visitor.scan(tp, trees); 
     } 
     return true; 
    } 

    @Override 
    public SourceVersion getSupportedSourceVersion() { 
     return SourceVersion.latestSupported(); 
    } 
} 

그리고 MethodPrintScanner :

public class MethodPrintScanner extends TreePathScanner { 

    @Override 
    public Object visitMethod(MethodTree methodTree, Object o) { 
     System.out.println(methodTree); 
     return null; 
    } 
} 

당신은 우리가 할 수있는 것을 알 수 있습니다 https://github.com/johncarl81/printMethod

여기에 그것의 고기입니다 : 여기이 라인을 따라 빠른 주석 프로세서이 만든 주어진 주석이 달린 요소와 연결된 TreePath을 방문하십시오. 각 메서드에 대해 우리는 단순히 println()methodTree 메서드의 내용을 제공합니다. 귀하의 예제를 사용

, 여기에 컴파일시 프로그램의 출력입니다 :

@PrintMethod() 
private boolean testMethod(String input) { 
    if (input != null) { 
     return true; 
    } 
    return false; 
} 

Answer 2

그것은 당신의 IDE에서 작동하도록 한 것입니다. 하지만 일단 코드가 jar 파일 안에 압축되면이를 감지하는 것이 또 다른 방법입니다. 다음 코드는이 두 가지를 모두 관리 할 수 ​​있습니다.

public static List<Class> getPackageClassListHavingAnnotation(String pPackageName, 
                   Class<? extends Annotation> pAnnotation) throws Exception 
{ 
    try 
    { 
    List<Class> classList = getPackageClassList(pPackageName); 
    if ((pAnnotation == null) || (classList == null)) return classList; 

    List<Class> resultList = new ArrayList<Class>(classList.size()); 

    outerLoop: 
    for (Class clazz : classList) 
    { 
     try 
     { 
     for (Method method : clazz.getMethods()) 
     { 
      if (method.isAnnotationPresent(pAnnotation)) 
      { 
      resultList.add(clazz); 
      continue outerLoop; 
      } 
     } 
     } 
     catch (Throwable e) 
     { 
     } 
    } 
    return (resultList.isEmpty()) ? null : resultList; 
    } 
    catch (Exception e) 
    { 
    return null; 
    } 
} 

은 다음 도우미 메서드가 필요합니다

public static List<Class> getPackageClassList(String pPackageName) throws Exception 
{ 
    try 
    { 
    ClassLoader classLoader = Thread.currentThread().getContextClassLoader(); 
    String path = pPackageName.replace('.', '/'); 

    List<File> dirs = new ArrayList<File>(); 
    List<JarFile> jars = new ArrayList<JarFile>(); 
    Enumeration<URL> resources = classLoader.getResources(path); 
    if (resources != null) 
    { 
     String fileName; 
     URL resource; 
     File file; 
     while (resources.hasMoreElements()) 
     { 
     resource = resources.nextElement(); 
     fileName = resource.getFile(); 

     if (fileName.contains("!")) 
     { 
      // jar file 
      resource = new URL(StringUtil.getArrayFromString(fileName, "!")[0]); 
      file = urlToFile(resource); 
      if (!file.exists()) continue; 
      jars.add(new JarFile(file)); 
     } 
     else 
     { 
      // class file that is not in a jar file 
      file = urlToFile(resource); 
      if (!file.exists()) continue; 
      dirs.add(file); 
     } 
     } 
    } 

    List<Class> resultList = new ArrayList<Class>(1000); 
    List<Class> tmpClassList; 
    for (File directory : dirs) 
    { 
     tmpClassList = getPckDirClassList(directory, pPackageName); 
     if (tmpClassList != null) resultList.addAll(tmpClassList); 
    } 

    for (JarFile jar : jars) 
    { 
     tmpClassList = getPckJarClassList(jar, pPackageName); 
     if (tmpClassList != null) resultList.addAll(tmpClassList); 
    } 

    return (resultList.isEmpty()) ? null : resultList; 
    } 
    catch (Exception e) 
    { 
    return null; 
    } 
} 

private static List<Class> getPckJarClassList(JarFile pJar, String pPackageName) 
{ 
    if ((pJar == null) || (pPackageName == null)) return null; 

    List<Class> resultList = new ArrayList<Class>(100); 

    Enumeration<JarEntry> jarEntries = (pJar.entries()); 
    JarEntry jarEntry; 
    String fullClassName; 
    while (jarEntries.hasMoreElements()) 
    { 
    jarEntry = jarEntries.nextElement(); 
    fullClassName = jarEntry.getName().replaceAll("/", "."); 
    if (!fullClassName.startsWith(pPackageName)) continue; 
    if (!fullClassName.endsWith(".class")) continue; 

    // do not do a Class.forName for the following path, this can crash the server 
    try 
    { 
     resultList.add(Class.forName(fullClassName.substring(0, fullClassName.length() - 6))); 
    } 
    catch (Throwable e) 
    { 
    } 
    } 

    return (resultList.isEmpty()) ? null : resultList; 
} 

/** 
* Recursive method to find all classes in a package directory tree. 
*/ 
private static List<Class> getPckDirClassList(File pDirectory, String pPackageName) throws ClassNotFoundException 
{ 
    try 
    { 
    if ((pDirectory == null) || (pPackageName == null)) return null; 

    if (!pDirectory.exists()) return null; 
    File[] files = pDirectory.listFiles(); 
    if ((files == null) || (files.length == 0)) return null; 

    List<Class> resultList = new ArrayList<Class>(100); 
    List<Class> tmpClassList; 
    for (File file : files) 
    { 
     if (file.isDirectory()) 
     { 
     tmpClassList = getPckDirClassList(file, pPackageName + "." + file.getName()); 
     if (tmpClassList != null) resultList.addAll(tmpClassList); 
     } 
     else if (file.getName().endsWith(".class")) 
     { 
     try 
     { 
      resultList.add(Class.forName(pPackageName + '.' + file.getName().substring(0, file.getName().length() - 6))); 
     } 
     catch (Throwable e) 
     { 
     } 
     } 
    } 
    return (resultList.isEmpty()) ? null : resultList; 
    } 
    catch (Exception e) 
    { 
    return null; 
    } 
} 

이 코드는 Windows 및 UNIX 시스템 모두에 .jar 파일을 테스트되었습니다. Windows의 IntelliJ에있는 파일도 .java으로 테스트되었습니다.