自定义ClassLoader，用于加载用户JAR包

最近在考虑C/S结构程序的软件自动升级的实现方式，比如QQ、飞信等都自动升级的功能。

 

自动升级模块虽然还没有编码完成，但是思路还是比较清晰的。

 

自动升级过程中，升级文件的JAR包是专门加载到程序中去的，因此，自定义一个ClassLoader，用于加载用户JAR包，就非常的重要了。

 

应用程序ClassLoader只提供了一个public Class<?> loadClass(String name) throws ClassNotFoundException 方法，没有提供加载JAR的方法。

 

URLClassLoader提供了一个protected void addURL(URL url)的方法，倒是可以加载JAR包，但苦于非public的。

 

经测试发现，AppClassLoader是URLClassLoader的子类。因此，我们完全可以利用URLClassLoader了哦。

URLClassLoader system = (URLClassLoader) ClassLoader.getSystemClassLoader();

 

这样，我们可以通过反射得到addURL方法，在程序中加载我们自己的JAR包了。

 

整个源代码如下所示：

/**
 * Copyright (c) YMCN Team
 * All rights reserved.
 */
package com.aboy.toolkit.util;

import java.io.File;
import java.lang.reflect.Method;
import java.net.URL;
import java.net.URLClassLoader;
import java.util.ArrayList;
import java.util.List;

/**
 * @author obullxl
 *
 * email: [e-mail]  MSN: [e-mail]  QQ: 303630027
 *
 * Blog: http://obullxl.iteye.com
 */
public final class ClassLoaderUtil {
    /** URLClassLoader的addURL方法 */
    private static Method addURL = initAddMethod();
    
    /** 初始化方法 */
    private static final Method initAddMethod() {
        try {
            Method add = URLClassLoader.class
                .getDeclaredMethod("addURL", new Class[] { URL.class });
            add.setAccessible(true);
            return add;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }

    private static URLClassLoader system = (URLClassLoader) ClassLoader.getSystemClassLoader();

    /**
     * 循环遍历目录，找出所有的JAR包
     */
    private static final void loopFiles(File file, List<File> files) {
        if (file.isDirectory()) {
            File[] tmps = file.listFiles();
            for (File tmp : tmps) {
                loopFiles(tmp, files);
            }
        } else {
            if (file.getAbsolutePath().endsWith(".jar") || file.getAbsolutePath().endsWith(".zip")) {
                files.add(file);
            }
        }
    }

    /**
     * <pre>
     * 加载JAR文件
     * </pre>
     *
     * @param file
     */
    public static final void loadJarFile(File file) {
        try {
            addURL.invoke(system, new Object[] { file.toURI().toURL() });
            System.out.println("加载JAR包：" + file.getAbsolutePath());
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * <pre>
     * 从一个目录加载所有JAR文件
     * </pre>
     *
     * @param path
     */
    public static final void loadJarPath(String path) {
        List<File> files = new ArrayList<File>();
        File lib = new File(path);
        loopFiles(lib, files);
        for (File file : files) {
            loadJarFile(file);
        }
    }
}

 

在程序中，只要使用上面最后两个方法，就可以加载自定义JAR包和一个目录中的所有JAR包了。

 

经本人测试，上面代码运行无误，能正常加载自定义JAR包。

 

欢迎指正~~~~~~~ 

评论

6 楼 xsz0606 2011-08-17  

我想问一下,为什么要使用反射这么迂回的方式,直接使用system.addURL不行吗?

5 楼 strawren 2010-07-06  

我把它改成了如下，主要用在web app里：
public class ClassLoaderUtils {
protected static Log log = LogFactory.getLog(ClassLoaderUtils.class);

private ClassLoaderUtils() {

}

public static void addURL(String url, ClassLoader classLoader) {
log.debug("addURL(),url->" + url);

if (StringUtils.isBlank(url)) {
return;
}
ClassLoader threadContextClassLoader = classLoader;
if (threadContextClassLoader == null) {
threadContextClassLoader = Thread.currentThread().getContextClassLoader();
}
boolean is = threadContextClassLoader instanceof URLClassLoader;
if (is) {
try {
URL realUrl = new URL(url);
URLClassLoader curr = (URLClassLoader) threadContextClassLoader;
Method add = URLClassLoader.class.getDeclaredMethod("addURL", new Class[] { URL.class });
add.setAccessible(true);
add.invoke(curr, new Object[] { realUrl });
}
catch (Exception e) {
log.warn("WARN", e);
throw new IllegalArgumentException(e);
}
} else {
log.warn("the classloader is not a URLClassLoader type!");
throw new IllegalArgumentException("the classloader is not a URLClassLoader type!");
}
}

public static ClassLoader getDefaultClassLoader() {
ClassLoader cl = null;
try {
cl = Thread.currentThread().getContextClassLoader();
}
catch (Throwable e) {
log.info("INFO", e);
}
if (cl == null) {
cl = ClassLoaderUtils.class.getClassLoader();
}
return cl;
}
----------------
当然也可以自己通过继承URLClassLoader自己实现类加载。

4 楼 obullxl 2010-04-23  

ivywjhua 写道

建议参考
http://www.ibm.com/developerworks/cn/java/j-lo-hotswapcls/index.html

看了这篇文章，升级方法值得参考，但和本文的目的有点不同。
我想做的是：升级单位是JAR包，不是一个类文件；升级下载一个JAR包后，并不是立即加载运行，而把它放在一个tmp目录，下次系统运行时再加载。

3 楼 ivywjhua 2010-04-23  

写一个  CustomizeClassloader， 继承自URLClassloader, 把addURL暴露出来

2 楼 ivywjhua 2010-04-23  

建议参考
http://www.ibm.com/developerworks/cn/java/j-lo-hotswapcls/index.html

1 楼 obullxl 2010-04-23  

说明下，AppClassLoader是URLClassLoader的子类：
System.out.println(ClassLoader.getSystemClassLoader() instanceof URLClassLoader);
输出结果为：true