通过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 ,更多精彩继续。

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