package com.match;

import java.io.*;

/**

* @ClassName: LoadClass

* @Description TODO

* @Date 2019/1/23 17:04

* @Created by chenzhuo

**/

public class LoadClass extends ClassLoader {

public static String rootDir = "/Users/mac/Desktop/JavaCode/src/main/java/";

@Override

public Class<?> loadClass(String name) throws ClassNotFoundException {

return super.loadClass(name);

}

@Override

protected Class<?> findClass(String name) throws ClassNotFoundException {

byte[] bytes = getClassData(name);

if (null == bytes) {

throw new ClassNotFoundException();

} else {

return defineClass(name, bytes, 0, bytes.length);

}

}

private byte[] getClassData(String className) {

String path = classNameToPath(className);

try {

InputStream ins = new FileInputStream(path);

ByteArrayOutputStream baos = new ByteArrayOutputStream();

int bufferSize = 4096;

byte[] buffer = new byte[bufferSize];

int bytesNumRead = 0;

// 读取类文件的字节码

while ((bytesNumRead = ins.read(buffer)) != -1) {

baos.write(buffer, 0, bytesNumRead);

}

return baos.toByteArray();

} catch (IOException e) {

e.printStackTrace();

}

return null;

}

private String classNameToPath(String className) {

return rootDir + File.separatorChar + className.replace('.', File.separatorChar)

+ ".class";

}

}