Xeraph@NCHOVY

아주 간단하게 JarClassLoader를 직접 구현해서 끝장봤다.
그냥 URLClassLoader 쓰려고 했는데 JAR 파일 한 번 물면 죽어라 안 놔줘서..

이제 로딩만 끝나고 나면 언로딩을 하든 말든 그냥 파일 날릴 수 있다.
진작 이렇게 나올 것이지 (..)

내일부터는 문서화하고 버그 찾고 코드 좀 다듬어 줘야지.
완전 대충 막 찍어놓은 코드 천지라서 -_-
그러고보니 리소스 처리도 빼먹었네;

package kr.nchovy.kraken.common;

import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;

import java.util.Enumeration;
import java.util.LinkedList;
import java.util.List;
import java.util.jar.JarEntry;
import java.util.jar.JarFile;

public class JarClassLoader extends ClassLoader {
    private List<String> loadedClasses;

    public JarClassLoader(String jarPath) throws FileNotFoundException,
            IOException {
        loadedClasses = new LinkedList<String>();
        loadClasses(jarPath);
    }

    public List<String> getLoadedClasses() {
        return loadedClasses;
    }

    private void loadClasses(String path) throws FileNotFoundException,
            IOException {
        JarFile jarFile = null;
        try {
            jarFile = new JarFile(path);

            Enumeration<JarEntry> entries = jarFile.entries();
            JarEntry entry = null;
            while (entries.hasMoreElements()) {
                entry = entries.nextElement();

                if (entry.getName().endsWith(".class")) {
                    String className = getClassName(entry);

                    InputStream jarInputStream = jarFile.getInputStream(entry);
                    byte[] data = new byte[(int) entry.getSize()];

                    jarInputStream.read(data);
                    defineClass(className, data, 0, data.length);
                    loadedClasses.add(className);
                }
            }

        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if (jarFile != null)
                jarFile.close();
        }
    }

    private String getClassName(JarEntry classEntry) {
        return classEntry.getName().replace(".class", "").replace("/", ".");
    }
}