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

原创
2017/04/21 22:18
阅读数 655

场景介绍  

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

解决方案

        java的热部署方案可以解决这个问题。java利用classloader的双亲委托机制可以解决这个问题。

        classloader的介绍请看:https://my.oschina.net/xpbob/blog/603868

        思路就是用新的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);
	}

}

        不知道为什么要覆写getPackage的,请看https://my.oschina.net/xpbob/blog/761436

        当有新的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的驱动类。

        这里已经获取到驱动类了,再通过这个驱动类就能直接获取链接。

Connection con = aDriver.driver.connect(url, info);

        对应数据库驱动获取链接的规范可以参考https://my.oschina.net/xpbob/blog/774722

总结

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

展开阅读全文
加载中
点击引领话题📣 发布并加入讨论🔥
0 评论
27 收藏
3
分享
返回顶部
顶部