com.sghd.common.utils.reflect.PackageScanner Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of common-utils Show documentation
The newest version!
package com.sghd.common.utils.reflect;

import java.io.File;
import java.io.FileFilter;
import java.io.IOException;
import java.net.JarURLConnection;
import java.net.URL;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.Enumeration;
import java.util.HashSet;
import java.util.Stack;
import java.util.jar.JarEntry;
import java.util.jar.JarFile;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class PackageScanner {
	
	private static final Logger LOG = LoggerFactory.getLogger(PackageScanner.class);
	
	private final Collection> clazzCollection = new HashSet>();

	public PackageScanner(final String... packageNames) {
		for (String packageName : packageNames) {
			try {
				final String packageDirectory = packageName.replace('.', '/');
				final Enumeration urls = Thread.currentThread().getContextClassLoader().getResources(packageDirectory);
				while (urls.hasMoreElements()) {
					final URL url = urls.nextElement();
					if("file".equals(url.getProtocol())) {
						final File directory = new File(url.getPath());
						if (!directory.isDirectory()) {
							throw new RuntimeException("package:[" + packageName + "] is not directory");
						}
						clazzCollection.addAll(PackageScanner.scanClassFromDirectory(packageName, directory));
					} else if("jar".equals(url.getProtocol())) {
						final JarFile jar = ((JarURLConnection)url.openConnection()).getJarFile();
						clazzCollection.addAll(PackageScanner.scanClassFromJar(packageName, jar));
					}
				}
			} catch (IOException exception) {
				throw new RuntimeException(exception);
			}
		}
	}
	
	public PackageScanner(final Collection packageNames) {
		this(packageNames.toArray(new String[packageNames.size()]));
	}
	
	public Collection> getClazzCollection() {
		return this.clazzCollection;
	}
	
	public static Collection> scanClassFromJar(final String packageName, final JarFile jar) {
		final Enumeration jarEntries = jar.entries();
		final Pattern pattern = Pattern.compile("("+packageName.replace('.', '/')+".*)\\.class");
		final Collection> clazzCollection = new HashSet>();
		
        while (jarEntries.hasMoreElements()) {
        	final JarEntry entry = jarEntries.nextElement();
            final String name = entry.getName();
            final Matcher matcher = pattern.matcher(name.replace(File.separatorChar, '/'));
			if(matcher.find()) {
				final String className = matcher.group(1).replace('/', '.');
				try {
					final Class clazz = Class.forName(className);
					clazzCollection.add(clazz);
				} catch (ClassNotFoundException e) {
					LOG.error("无法加载类[{}]", className, e);
				}
			}
        }
        return clazzCollection;
	}

	public static Collection> scanClassFromDirectory(final String packageName, final File directory) {
		final Stack scanDirectories = new Stack();
		final Collection classFiles = new ArrayList();
		final FileFilter fileFilter = new FileFilter() {
			@Override
			public boolean accept(File file) {
				if (file.isDirectory()) {
					scanDirectories.push(file);
					return false;
				}
				return file.getName().matches(".*\\.class$");
			}
		};

		scanDirectories.push(directory);

		while (!scanDirectories.isEmpty()) {
			final File scanDirectory = scanDirectories.pop();
			Collections.addAll(classFiles, scanDirectory.listFiles(fileFilter));
		}
		final Pattern pattern = Pattern.compile("("+packageName.replace('.', '/')+".*)\\.class");
		final Collection> clazzCollection = new HashSet>();
		for (File file : classFiles) {
			final Matcher matcher = pattern.matcher(file.getAbsolutePath().replace(File.separatorChar, '/'));
			if(matcher.find()) {
				final String className = matcher.group(1).replace('/', '.');
				try {
					final Class clazz = Class.forName(className);
					clazzCollection.add(clazz);
				} catch (ClassNotFoundException e) {
					LOG.error("无法加载类[{}]", className, e);
				}
			}
		}
		
		return clazzCollection;
	}
	
}