java - 在运行时查找包中的所有类(并调用静态方法)

Question

我有一个包，其中有很多类，这些类都有一个名为 onLoad 的静态方法。我想在程序启动时调用 onLoad 方法，但是我不想硬连线每个方法，即 classA.onLoad(); classB.onLoad(); 等

如何列出 com.foo.bar.asd 包中的所有类并对所有类运行 onLoad()？

提前致谢

Answer 1

我发现这个问题很有趣，所以我想出了一个解决方案。这里棘手的部分是实际找到给定包中的所有类。

在此示例中，我假设所有类都位于类 C 所在的同一个包中，并且除 C 之外的所有类都有可能无法访问的 onLoad 方法(即 private)。然后您可以使用以下示例代码。

public final class C {

    public static void main(String[] args) throws Exception {
        for (Class<?> cls : getClasses(C.class)) {
            if (cls != C.class) {
                Method onLoad = cls.getDeclaredMethod("onLoad");
                onLoad.setAccessible(true);
                onLoad.invoke(null);
            }
        }
    }

    private static List<Class<?>> getClasses(Class<?> caller)
            throws IOException, URISyntaxException {
        return Files.walk(getPackagePath(caller))
                .filter(Files::isRegularFile)
                .filter(file -> file.toString().endsWith(".class"))
                .map(path -> mapPathToClass(path, caller.getPackage().getName()))
                .collect(Collectors.toList());
    }

    private static Class<?> mapPathToClass(Path clsPath, String packageName) {
        String className = clsPath.toFile().getName();
        className = className.substring(0, className.length() - 6);
        return loadClass(packageName + "." + className);
    }

    private static Path getPackagePath(Class<?> caller)
            throws IOException, URISyntaxException {
        String packageName = createPackageName(caller);
        Enumeration<URL> resources = caller.getClassLoader()
                .getResources(packageName);
        return Paths.get(resources.nextElement().toURI());
    }

    private static String createPackageName(Class<?> caller) {
        return caller.getPackage().getName().replace(".", "/");
    }

    private static Class<?> loadClass(String name) {
        try {
            return Class.forName(name);
        } catch (ClassNotFoundException e) {
            return null;
        }
    }
}

我省略了所有异常检查，像 resources.nextElement() 这样的语句甚至可能引发异常。因此，如果您想涵盖这些情况，则需要在这里或那里添加一些检查。