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

>>强大,10k+点赞的 SpringBoot 后台管理系统竟然出了详细教程!

场景介绍

使用过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来做到热更新。


原文始发于微信公众号(一次编译多处运行):classloader实战:如何不重启校验数据库驱动链接