55

我有一个简单的方法标记注释（类似于Effective Java（第 2 版）第 35 项中的第一个示例）：

/**
 * Marker annotation for methods that are called from installer's 
 * validation scripts etc. 
 */
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface InstallerMethod {
}

然后，在一个给定的包（比如com.acme.installer）中，它有几个包含大约 20 个类的子包，我想找到所有用它注释的方法。（因为我想对单元测试中的所有带注释的方法进行一些检查。）

什么（如果有）是最简单的方法？最好不要添加新的 3rd 方库或框架。

编辑：澄清一下，显然method.isAnnotationPresent(InstallerMethod.class)将是检查方法是否具有注释的方法-但此问题包括查找所有方法。

4

3 回答 3

54

如果您想自己实现它，这些方法将找到给定包中的所有类：

/**
 * Scans all classes accessible from the context class loader which belong
 * to the given package and subpackages.
 * 
 * @param packageName
 *            The base package
 * @return The classes
 * @throws ClassNotFoundException
 * @throws IOException
 */
private Iterable<Class> getClasses(String packageName) throws ClassNotFoundException, IOException
{
    ClassLoader classLoader = Thread.currentThread().getContextClassLoader();
    String path = packageName.replace('.', '/');
    Enumeration<URL> resources = classLoader.getResources(path);
    List<File> dirs = new ArrayList<File>();
    while (resources.hasMoreElements())
    {
        URL resource = resources.nextElement();
        URI uri = new URI(resource.toString());
        dirs.add(new File(uri.getPath()));
    }
    List<Class> classes = new ArrayList<Class>();
    for (File directory : dirs)
    {
        classes.addAll(findClasses(directory, packageName));
    }

    return classes;
}

/**
 * Recursive method used to find all classes in a given directory and
 * subdirs.
 * 
 * @param directory
 *            The base directory
 * @param packageName
 *            The package name for classes found inside the base directory
 * @return The classes
 * @throws ClassNotFoundException
 */
private List<Class> findClasses(File directory, String packageName) throws ClassNotFoundException
{
    List<Class> classes = new ArrayList<Class>();
    if (!directory.exists())
    {
        return classes;
    }
    File[] files = directory.listFiles();
    for (File file : files)
    {
        if (file.isDirectory())
        {
            classes.addAll(findClasses(file, packageName + "." + file.getName()));
        }
        else if (file.getName().endsWith(".class"))
        {
            classes.add(Class.forName(packageName + '.' + file.getName().substring(0, file.getName().length() - 6)));
        }
    }
    return classes;
}

然后，您可以使用给定的注释过滤这些类：

for (Method method : testClass.getMethods())
{
    if (method.isAnnotationPresent(InstallerMethod.class))
    {
        // do something
    }
}
于 2009-05-14T08:13:02.647 回答
37

您可能应该看看开源反射库。有了它，你可以用几行代码轻松实现你想要的：

Reflections reflections = new Reflections( 
    new ConfigurationBuilder().setUrls( 
    ClasspathHelper.forPackage( "com.acme.installer" ) ).setScanners(
    new MethodAnnotationsScanner() ) );
Set<Method> methods = reflections.getMethodsAnnotatedWith(InstallerMethod.class);
于 2012-03-05T17:33:09.760 回答
1

如果您乐于使用 Spring，那么它会使用它的 context:component-scan 功能来执行这些操作，其中 Spring 扫描给定包中的注释类。在幕后，它非常可怕，并且涉及在文件系统和 JAR 文件中寻找包中的类。

即使你不能直接使用 Spring，看看它的源代码也可能会给你一些想法。

当然，Java反射API在这里是没有用的，它具体不提供获取包中所有类的方法。

于 2009-05-14T10:04:17.760 回答