classloader实战：如何不重启校验数据库驱动链接

场景介绍  

使用过was的时候，我们常见的建立数据源时有一个验证数据库信息的正确性的按钮。但是如果没有相应的驱动包的时候，校验是失败的，如果想校验成功，那就加入对应的数据库驱动包即可，但是was本身并不是热部署的，要想验证那就必须重启was。这个在服务器还是可以接受的，因为你建立数据源肯定是是先有规划的，驱动包都是放入指定地点的，weblogic11g was本身还提供了很多数据库的Jar包以备使用。但是换成一个配置系统的话，那这样的操作就不能忍受了。因为要验证一下数据库连接是否正确还要去重启远程的机器，这听起来就比较麻烦。

解决方案

java的热部署方案可以解决这个问题。java利用classloader的双亲委托机制可以解决这个问题。思路就是用新的classloader去加载类，然后去做校验，打破双亲委托。类加载的目录就是我们指定的目录，每次可以上传jar包到固定目录。然后用新的classloader去加载。

代码实现

首先要写一个破坏双亲委托的classloader。

package com.xp.classloader;

import java.net.URL;
import java.net.URLClassLoader;
import java.util.HashMap;
import java.util.Map;

public class DriverLoader extends URLClassLoader {

    // class cache
    Map<String, Class<?>> loadedClasses = new HashMap<>();

    public DriverLoader(URL[] urls) {
        super(urls);
    }

    @Override
    public Class<?> loadClass(String name) throws ClassNotFoundException {
        if (loadedClasses.containsKey(name)) {
            return loadedClasses.get(name);
        }

        try {
            Class<?> findClass = this.findClass(name);
            loadedClasses.put(name, findClass);
            return findClass;
        } catch (ClassNotFoundException e) {
            // ignore it
        }
        return super.loadClass(name);
    }

    @Override
    protected Package getPackage(String name) {
        return super.getPackage(name);
    }

}

当有新的jar包上传的时候，就需要重新加载类，这里的类是可以配置的。然后根据配置读取加载的类。

package com.xp.classloader;

import java.io.File;
import java.io.FilenameFilter;
import java.net.URL;

public class ClassManager {
    public static final String DIR = "d:/jarFile/";

    public static Class<?> getClass(String className) throws Exception {
        File jarDir = new File(DIR);
        File[] listFiles = jarDir.listFiles(new FilenameFilter() {

            @Override
            public boolean accept(File dir, String name) {
                // check jar file
                if (name.endsWith(".jar")) {
                    return true;
                }
                return false;
            }
        });
        URL[] urls;
        if (listFiles == null) {
            urls = new URL[0];
        } else {
            int count = 0;
            urls = new URL[listFiles.length];
            for (File listFile : listFiles) {
                urls[count++] = listFile.toURI().toURL();
            }
        }
        ClassLoader loader = new DriverLoader(urls);
        return loader.loadClass(className);
    }
}

为了发现文件上传就可以出发这个操作，所以此时需要使用java7的新特性---目录监控。

package com.xp.dir;

import java.io.IOException;
import java.nio.file.FileSystems;
import java.nio.file.Paths;
import java.nio.file.StandardWatchEventKinds;
import java.nio.file.WatchEvent;
import java.nio.file.WatchEvent.Kind;
import java.nio.file.WatchKey;
import java.nio.file.WatchService;

public class DirWatcher {

    private WatchService service;

    public DirWatcher(String filePath) throws IOException {
        service = FileSystems.getDefault().newWatchService();
        // register create event
        Paths.get(filePath).register(service, StandardWatchEventKinds.ENTRY_CREATE);

    }

    public void watch(WatcherHandler handler) throws InterruptedException {
        while (true) {
            WatchKey key = service.take();
            for (WatchEvent<?> event : key.pollEvents()) {
                Kind<?> kind = event.kind();
                if (kind == StandardWatchEventKinds.ENTRY_CREATE) {
                    handler.handler();
                }
            }
            key.reset();
        }
    }

}

为了完成功能代码简洁，这里使用了回调。增加一个接口。

package com.xp.dir;

public interface WatcherHandler {
    public void handler();
}

接下来就是调用的主体。

package com.xp;

import com.xp.classloader.ClassManager;
import com.xp.dir.DirWatcher;
import com.xp.dir.WatcherHandler;

public class Main {

    public static void main(String[] args) throws Exception {
        new DirWatcher(ClassManager.DIR).watch(new WatcherHandler() {

            @Override
            public void handler() {
                Class<?> class1;
                try {
                    class1 = ClassManager.getClass("com.mysql.jdbc.Driver");
                    if (class1 != null) {
                        System.out.println("success");
                    }
                } catch (Exception e) {
                    e.printStackTrace();
                }

            }
        });
    }

}

 开启目录监控，然后直接使用回调去加载mysql的驱动类。这里已经获取到驱动类了，再通过这个驱动类就能直接获取链接。

总结

classloader让Java的操作更加灵活，很多需要重启应用的问题，都可以考虑加入classloader来做到热更新。