通过JAVA反射修改JDK1.6*当中DNS缓存内容

时间 2012-05-28 17:09:03 Taobao QA Team原文 http://qa.taobao.com/?p=15523

为了实现性能压测时的域名动态绑定功能，尝试通过java反射修改JDK1.6×当中的DNS缓存，感谢在此过程中林轩同学的大力帮助。

网上也存在着修改DNS缓存的方法，但是都是基于jdk1.5的，无法应用。另外，大部分都是修改的缓存过期时间，而没有真正去尝试修改dns 的cache内容，所以尝试了很多种方法，并且查看了jdk的源代码，终于实现了修改dns缓存内容和时间，如下，欢迎大家一起探讨

修改缓存时间可以采用修改java security manage文件和增加jvm参数的方法，以下是在代码中动态修改dns的缓存时间：

public boolean addressCache(Map<String, String> addressMap) {

// 以下内容修改缓存失效时间,单位秒(-1,永久缓存;0,不缓存;其它>0的值为缓存的秒数)
Class<?> iacp = sun.net.InetAddressCachePolicy.class;

Field cachePolicyFiled = null;
Field setEnable = null;
try {
cachePolicyFiled = iacp.getDeclaredField(“cachePolicy”);
setEnable = iacp.getDeclaredField(“set”);
} catch (NoSuchFieldException e) {
logger.error(“Get cachePolicyFiled or setEnable failed!”, e);
}
cachePolicyFiled.setAccessible(true);
setEnable.setAccessible(true);

try {
cachePolicyFiled.set(null, -1);
setEnable.set(null, true);
} catch (IllegalArgumentException e) {
logger.error(“Set cachePolicyFiled and setEnable failed!”, e);
} catch (IllegalAccessException e) {
logger.error(“Access cachePolicyFiled and setEnable failed!”, e);
}
// 以下内容修改DNS的缓存数据，实现域名动态绑定
Class<?> inetAddressClass = java.net.InetAddress.class;
Field cacheField = null;
Object addressCache = null;
InetAddress ia = null;
try {
cacheField = inetAddressClass.getDeclaredField(“addressCache”);
} catch (SecurityException e) {
logger.error(“Get addressCache security failed!”, e);
} catch (NoSuchFieldException e) {
logger.error(“Get addressCache field failed!”, e);
}

cacheField.setAccessible(true);

try {
addressCache = cacheField.get(inetAddressClass);
} catch (IllegalArgumentException e) {
logger.error(“Get cacheField IllegalArgument failed!”, e);
} catch (IllegalAccessException e) {
logger.error(“Get cacheField Access failed!”, e);
}
Class<?> cacheClazz = addressCache.getClass();

Method m_put = cacheClazz.getMethod(“put”, new Class<?>[] {
String.class, Object.class });
m_put.setAccessible(true);

try {
for (String domain : addressMap.keySet()) {
ia = Inet4Address.getByName(addressMap.get(domain));
m_put.invoke(addressCache, new Object[] { domain,
new InetAddress[] { ia } });
}

} catch (IllegalArgumentException e) {
logger.error(“Put addressCache IllegalArgument failed!”, e);
} catch (IllegalAccessException e) {
logger.error(“Put addressCache IllegalAccess failed!”, e);
} catch (InvocationTargetException e) {
logger.error(“Put addressCache InvocationTarget failed!”, e);
} catch (UnknownHostException e) {
logger.error(“Put addressCache UnknownHost failed!”, e);
}
// 修改缓存数据结束
return true;
}

在性能压测时，采用JAVA的DNS缓存有以下好处

1.无需关心dns解析时间，dns解析消耗0毫秒，几乎无消耗，这样使得压测更加关注服务器响应，压测更充分

2.可以动态修改DNS缓存，无需修改hosts文件和http链接等不灵活的方式。

3.一个jvm进程可以对应一套域名绑定，相互之间不影响，可以实现多场景，多域名绑定的需求压测。

最后，做一下广告：欢迎支持淘宝自主研发的高并发集群性能压测工具 Trunner ,更多精彩继续。

云帅 技术质量部–测试平台与发展–压测工具