java查找反复类/jar包/普通文件

时间:2022-07-15 20:50:18

开发web应用时,有时更新了类却没有生效,事实上是由于jboss/tomcat中其它公布包下有同名类(包含全路径都同样)。

于是萌发了做个程序来检查指定文件夹是否存在反复类(通过asm从类文件里取类的全路径),扩展开来,还支持查找反复的文件(按文件md5进行比較),反复的jar文件。

主要代码例如以下:

package cn.jerryhouse.util.dup_files;

import java.io.File;

public abstract class FileProcessor {
private long totalFileCount = 0;
private long processedFileCount = 0;
public void processFiles(File[] dirs) throws Exception {
for (File file : dirs) {
processFile(file);
}
} public void processFile(File file) throws Exception {
if (file.isFile()) {
if (isFileAccepted(file)) {
handleFile(file);
processedFileCount++;
}
totalFileCount++;
} else {
File[] files = file.listFiles();
for (File fileInDir : files) {
processFile(fileInDir);
}
}
} protected boolean isFileAccepted(File file) throws Exception {
return true;
} protected abstract void handleFile(File file); public long getTotalFileCount() {
return totalFileCount;
} public long getProcessedFileCount() {
return processedFileCount;
}
}

package cn.jerryhouse.util.dup_files;

import java.io.File;
import java.io.IOException;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.concurrent.ConcurrentHashMap; public class FileDupFinder extends FileProcessor {
protected Map<String, List<String>> fileMap = new ConcurrentHashMap<String, List<String>>();
public void analyseResult() {
boolean foundDup = false;
for (Entry<String, List<String>> entry : fileMap.entrySet()) {
List<String> classPathList = entry.getValue();
if (classPathList.size() > 1) {
System.out.println(" paths:"
+ classPathList);
foundDup = true;
}
}
if (foundDup == false) {
System.out.println("No duplicate file found.");
}
} protected void handleFile(final File file) {
try {
String fileMD5 = FileDigest.getFileMD5(file);
try {
List<String> list;
if (fileMap.get(fileMD5) != null) {
list = fileMap.get(fileMD5);
list.add(file.getCanonicalPath());
} else {
list = new LinkedList<String>();
list.add(file.getCanonicalPath());
fileMap.put(fileMD5, list);
}
} catch (IOException e) {
e.printStackTrace();
}
} catch (Exception e) {
e.printStackTrace();
}
} }

package cn.jerryhouse.util.dup_files;

import java.io.File;

public class JarFileDupFinder extends FileDupFinder {

	protected boolean isFileAccepted(final File file) throws Exception
{
if(file.getCanonicalPath().endsWith(".jar"))
{
return true;
}
else
{
return false;
}
} }
package cn.jerryhouse.util.dup_files;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.LinkedList;
import java.util.List;
import java.util.Map.Entry; import org.objectweb.asm.AnnotationVisitor;
import org.objectweb.asm.Attribute;
import org.objectweb.asm.ClassReader;
import org.objectweb.asm.ClassVisitor;
import org.objectweb.asm.FieldVisitor;
import org.objectweb.asm.MethodVisitor; public class ClassDupFinder extends FileDupFinder { protected boolean isFileAccepted(final File file) throws Exception {
if (file.getCanonicalPath().endsWith(".class")) {
return true;
} else {
return false;
}
} public void analyseResult() {
boolean foundDup = false;
for (Entry<String, List<String>> entry : fileMap.entrySet()) {
List<String> classPathList = entry.getValue();
if (classPathList.size() > 1) {
System.out.println("class:" + entry.getKey() + " paths:"
+ classPathList);
foundDup = true;
}
}
if (foundDup == false) {
System.out.println("No duplicate class found.");
}
} protected void handleFile(final File file) {
try {
String filePath = file.getCanonicalPath();
ClassVisitor visitor = new ClassVisitor() {
@Override
public void visit(int version, int access, String name,
String signature, String superName, String[] interfaces) {
try {
List<String> list;
if (fileMap.get(name) != null) {
list = fileMap.get(name);
list.add(file.getCanonicalPath());
} else {
list = new LinkedList<String>();
list.add(file.getCanonicalPath());
fileMap.put(name, list);
}
} catch (IOException e) {
e.printStackTrace();
}
} @Override
public void visitSource(String source, String debug) {
} @Override
public void visitOuterClass(String owner, String name,
String desc) {
} @Override
public AnnotationVisitor visitAnnotation(String desc,
boolean visiable) {
return null;
} @Override
public void visitAttribute(Attribute attr) {
} @Override
public void visitInnerClass(String name, String outerName,
String innerName, int access) {
} @Override
public FieldVisitor visitField(int access, String name,
String desc, String signature, Object value) {
return null;
} @Override
public MethodVisitor visitMethod(int access, String name,
String desc, String signature, String[] exceptions) {
return null;
} @Override
public void visitEnd() {
}
};
ClassReader cr = new ClassReader(new FileInputStream(filePath));
cr.accept(visitor, 0);
} catch (Exception e) {
e.printStackTrace();
}
} }
package cn.jerryhouse.util.dup_files;

import java.io.File;
import java.io.FileInputStream;
import java.math.BigInteger;
import java.security.MessageDigest;
import java.util.HashMap;
import java.util.Map; public class FileDigest {
/**
* 获取单个文件的MD5值
* @param file
*/
public static String getFileMD5(File file) {
if (!file.isFile()) {
return null;
}
MessageDigest digest = null;
FileInputStream in = null;
byte buffer[] = new byte[1024];
int len;
try {
digest = MessageDigest.getInstance("MD5");
in = new FileInputStream(file);
while ((len = in.read(buffer, 0, 1024)) != -1) {
digest.update(buffer, 0, len);
}
in.close();
} catch (Exception e) {
e.printStackTrace();
return null;
}
BigInteger bigInt = new BigInteger(1, digest.digest());
return bigInt.toString(16);
} /**
* 获取文件夹中文件的MD5值
* @param file
* @param listChild 为true时,递归子文件夹中的文件;否则不递归
*/
public static Map<String, String> getDirMD5(File file, boolean listChild) {
if (!file.isDirectory()) {
return null;
}
Map<String, String> map = new HashMap<String, String>();
String md5;
File files[] = file.listFiles();
for (int i = 0; i < files.length; i++) {
File f = files[i];
if (f.isDirectory() && listChild) {
map.putAll(getDirMD5(f, listChild));
} else {
md5 = getFileMD5(f);
if (md5 != null) {
map.put(f.getPath(), md5);
}
}
}
return map;
}
}
package cn.jerryhouse.util.dup_files;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader; public abstract class LineReaderProcessor extends FileProcessor {
private long totalFileCount = 0;
private long processedFileCount = 0;
public void processFiles(File[] dirs) throws Exception {
for (File file : dirs) {
processFile(file);
}
} public void processFile(File file) throws Exception {
if (file.isFile()) {
if (isFileAccepted(file)) {
handleFile(file);
}
} else {
File[] files = file.listFiles();
for (File fileInDir : files) {
processFile(fileInDir);
}
}
} protected boolean isFileAccepted(File file) throws Exception {
return true;
} protected void handleFile(File file)
{
try {
BufferedReader br = new BufferedReader(new FileReader(file));
String line;
while((line=br.readLine())!=null)
{
readLine(file,line);
}
br.close();
} catch (Exception e) {
e.printStackTrace();
}
}
protected abstract void readLine(File file,String line); public long getTotalFileCount() {
return totalFileCount;
} public long getProcessedFileCount() {
return processedFileCount;
}
}
package cn.jerryhouse.util.dup_files;

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileReader;
import java.io.FileWriter;
import java.util.UUID; public abstract class LineUpdateProcessor extends FileProcessor {
private long totalFileCount = 0;
private long processedFileCount = 0;
private String NEW_LINE = System.getProperty("line.separator");
public void processFiles(File[] dirs) throws Exception {
for (File file : dirs) {
processFile(file);
}
} public void processFile(File file) throws Exception {
if (file.isFile()) {
if (isFileAccepted(file)) {
handleFile(file);
}
} else {
File[] files = file.listFiles();
for (File fileInDir : files) {
processFile(fileInDir);
}
}
} protected boolean isFileAccepted(File file) throws Exception {
return true;
} protected void handleFile(File file)
{
try {
BufferedReader br = new BufferedReader(new FileReader(file));
File tmpFile = new File(tmpFilePath(file));
BufferedWriter bw = new BufferedWriter(new FileWriter(tmpFile));
String line;
while((line=br.readLine())!=null)
{
String updatedLine = updateLine(file,line);
bw.write(updatedLine+NEW_LINE);
}
br.close();
bw.close();
file.delete();
tmpFile.renameTo(file);
} catch (Exception e) {
e.printStackTrace();
}
} private String tmpFilePath(File file)
{
String dir = file.getParent();
String filePath = dir+""+getUniqFileName();
return filePath;
} private String getUniqFileName()
{
return UUID.randomUUID().toString();
}
protected abstract String updateLine(File file,String line); public long getTotalFileCount() {
return totalFileCount;
} public long getProcessedFileCount() {
return processedFileCount;
}
}

简单測试代码:

package cn.jerryhouse.util.dup_files.test;

import java.io.File;

import org.junit.Test;

import cn.jerryhouse.util.dup_files.ClassDupFinder;
import cn.jerryhouse.util.dup_files.FileDupFinder;
import cn.jerryhouse.util.dup_files.JarFileDupFinder; public class DupTest { @Test
public void testJarFiles() {
try {
File[] files = new File[1];
files[0] = new File("E:\\workspace\\yinxing");
JarFileDupFinder dupFinder = new JarFileDupFinder();
dupFinder.processFiles(files);
dupFinder.analyseResult();
} catch (Exception e) {
e.printStackTrace();
}
}
@Test
public void testFileDup() {
try {
File[] files = new File[1];
files[0] = new File("E:\\workspace\\yinxing");
FileDupFinder classDupFinder = new FileDupFinder();
classDupFinder.processFiles(files);
classDupFinder.analyseResult();
} catch (Exception e) {
e.printStackTrace();
}
} @Test
public void testClassDup() {
try {
File[] files = new File[1];
files[0] = new File("E:\\workspace\\yinxing");
ClassDupFinder classDupFinder = new ClassDupFinder();
classDupFinder.processFiles(files);
classDupFinder.analyseResult();
} catch (Exception e) {
e.printStackTrace();
}
} }

注:依赖jar包asm.jar。