Hibernate中HttpURLConnection使用位置及Java Agent注入异常排查

这问题我之前碰到过类似的，核心原因大概率是你的Java Agent无差别地给所有HttpURLConnection添加请求头，干扰了Hibernate加载配置文件时的内部HTTP请求——具体来说是dom4j解析hibernate.cfg.xml时，会自动请求外部DTD文件（比如Hibernate官方的DTD），这时候你的Agent修改了这个请求的头，导致DTD下载失败，最终触发XML解析异常。

咱们一步步拆解解决：

先确认问题根源 #

检查你的hibernate.cfg.xml开头，是不是有类似这样的外部DTD引用：

<!DOCTYPE hibernate-configuration PUBLIC
        "-//Hibernate/Hibernate Configuration DTD 3.0//EN"
        "http://www.hibernate.org/dtd/hibernate-configuration-3.0.dtd">

如果是的话，dom4j在解析XML时会自动发起HTTP请求去下载这个DTD文件，而你的Agent恰好给这个HttpURLConnection加了自定义请求头——有些服务器会拒绝带未知头的请求，或者插桩时机不对导致连接状态异常，最终引发DocumentException，连锁导致Hibernate配置解析失败。

而你说只打印日志就正常，是因为打印操作不会修改连接状态，自然不会干扰DTD的下载流程。

针对性解决方案 #

1. 限制插桩范围，只给业务请求加头 #

不要给所有HttpURLConnection都插桩，只针对你自己业务代码发起的请求。可以通过检查调用栈来过滤：

// 修改你的Agent插桩代码，添加调用栈判断
ctConstructor.insertAfter("" +
    "// 只给业务包下的请求加头\n" +
    "boolean isBusinessRequest = false;\n" +
    "for(StackTraceElement elem : Thread.currentThread().getStackTrace()) {\n" +
    "    // 替换成你的业务包名前缀\n" +
    "    if(elem.getClassName().startsWith(\"com.javapapers\")) {\n" +
    "        isBusinessRequest = true;\n" +
    "        break;\n" +
    "    }\n" +
    "}\n" +
    "if(isBusinessRequest) {\n" +
    "    setRequestProperty(\"testHeader\",\"NewHeader\");\n" +
    "}\n" +
    "System.out.println(10);");

这样就能避开Hibernate、dom4j等框架内部的HttpURLConnection调用，只处理你自己的业务请求。

2. 改用本地DTD文件，避免外部HTTP请求 #

把Hibernate的DTD文件下载到本地（可以从Maven仓库或者官方网站获取），然后修改hibernate.cfg.xml的DTD引用为本地路径：

<!DOCTYPE hibernate-configuration PUBLIC
        "-//Hibernate/Hibernate Configuration DTD 3.0//EN"
        "classpath:/hibernate-configuration-3.0.dtd">

把下载好的hibernate-configuration-3.0.dtd放到项目的resources目录下（确保在classpath中），这样dom4j就不会发起外部HTTP请求，自然不会被你的Agent干扰。

3. 调整插桩时机，在connect()前设置请求头 #

HttpURLConnection的请求头应该在connect()方法调用之前设置才是规范的时机。你之前在构造函数后插桩，可能连接内部状态还没准备好，导致设置头时出错。可以改为插桩connect()方法：

// 找到HttpURLConnection的connect方法
CtMethod connectMethod = ctClass.getMethod("connect", new CtClass[]{});
// 在connect前设置请求头
connectMethod.insertBefore("setRequestProperty(\"testHeader\",\"NewHeader\");");

这种方式更符合HttpURLConnection的使用规范，也能避免构造阶段的状态异常。

验证方法 #

先尝试方案2（本地DTD），如果异常消失，就说明确实是外部DTD请求被干扰导致的问题；再根据业务需求选择方案1或3来完善你的Agent。

内容的提问来源于stack exchange，提问作者s s