集册 Java实例教程 查找包中的类

查找包中的类

欢马劈雪     最近更新时间:2020-01-02 10:19:05

548
查找包中的类


import java.io.File;

import java.io.FileFilter;

import java.io.FilenameFilter;// from 时 代      J a v a   公   众 号 - nowjava.com

import java.io.IOException;

import java.net.JarURLConnection;

import java.net.URL;

import java.net.URLDecoder;

import java.util.ArrayList;

import java.util.Enumeration;

import java.util.List;

import java.util.jar.JarEntry;

import java.util.jar.JarFile;

import java.util.regex.Pattern;


public class Main{

    /**

     * 

     * @param packageName

     * @return The list of all the classes inside this package

     * @throws IOException

     */

    public static String[] findClassesInPackage(String packageName,

            List<String> included, List<String> excluded)

            throws IOException {

        String packageOnly = packageName;

        String packageNameSave = packageName.replace('/', '.');/*from 时代Java - nowjava.com*/

        boolean recursive = false;

        if (packageName.endsWith(".*")) {

            packageOnly = packageName.substring(0,

                    packageName.lastIndexOf(".*"));

            recursive = true;

        }


        List<String> vResult = new ArrayList<String>();

        String packageDirName = packageOnly.replace('.', '/');

        Enumeration<URL> dirs = PackageUtil.class.getClassLoader()

                .getResources(packageDirName);

        while (dirs.hasMoreElements()) {

            URL url = dirs.nextElement();

            String protocol = url.getProtocol();

            if ("file".equals(protocol)) {

                findClassesInDirPackage(packageOnly, included, excluded,

                        URLDecoder.decode(url.getFile(), "UTF-8"),

                        recursive, vResult);

            } else if ("jar".equals(protocol)) {

                JarFile jar = ((JarURLConnection) url.openConnection())

                        .getJarFile();

                Enumeration<JarEntry> entries = jar.entries();

                while (entries.hasMoreElements()) {

                    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(

                                    '/', '.');

                        }


                        //PackageUtils.log("Package name is " + packageName);

                        if ((idx != -1) || recursive) {

                            // it's not inside a deeper dir

                            if (name.endsWith(".class")

                                    && !entry.isDirectory()

                                    && packageName.equals(packageNameSave)) {

                                String className = name.substring(

                                        packageName.length() + 1,

                                        name.length() - 6);

                                // PackageUtils.log("Found class " + className +

                                // ", seeing it if it's included or excluded");

                                includeOrExcludeClass(packageName,

                                        className, included, excluded,

                                        vResult);

                            }

                        }

                    }

                }

            }

        }

        String[] result = vResult.toArray(new String[vResult.size()]);

        return result;

    }

    private static void findClassesInDirPackage(String packageName,

            List<String> included, List<String> excluded,

            String packagePath, final boolean recursive,

            List<String> classes) {

        File dir = new File(packagePath);


        if (!dir.exists() || !dir.isDirectory()) {

            return;

        }


        File[] dirfiles = dir.listFiles(new FileFilter() {

            public boolean accept(File file) {

                return (recursive && file.isDirectory())

                        || (file.getName().endsWith(".class"));

            }

        });


        // PackageUtils.log("Looking for test classes in the directory: " +

        // dir);

        for (File file : dirfiles) {

            if (file.isDirectory() && recursive) {

                findClassesInDirPackage(packageName + "." + file.getName(),

                        included, excluded, file.getAbsolutePath(),

                        recursive, classes);

            } else {

                String className = file.getName().substring(0,

                        file.getName().length() - 6);

                // PackageUtils.log("Found class " + className + ", seeing it if

                // it's included or excluded");

                includeOrExcludeClass(packageName, className, included,

                        excluded, classes);

            }

        }

    }

    private static void includeOrExcludeClass(String packageName,

            String className, List<String> included, List<String> excluded,

            List<String> classes) {

        if (isIncluded(className, included, excluded)) {

            //Utils.log("... Including class " + className);

            //classes.add(packageName + '.' + className);

            classes.add(className);

        }

        //        }       else {

        //Utils.log("... Excluding class " + className);

        //        }

    }

    /**

     * @return true if name should be included.

     */

    private static boolean isIncluded(String name, List<
展开阅读全文