如何借助JUnit或其他工具在测试中检测资源泄漏？

基于JUnit的增强检测 #

• 自定义JUnit 5扩展跟踪资源状态
  写一个JUnit 5扩展，在测试全周期内跟踪AutoCloseable类型资源的创建与关闭。通过代理包装资源实例，记录状态，测试结束后自动校验是否有未关闭的资源。
  示例代码：

  public class ResourceLeakDetector implements AfterTestExecutionCallback {
      private static final ThreadLocal<Set<AutoCloseable>> OPEN_RESOURCES = ThreadLocal.withInitial(HashSet::new);
  
      public static <T extends AutoCloseable> T track(T resource) {
          OPEN_RESOURCES.get().add(resource);
          return resource;
      }
  
      @Override
      public void afterTestExecution(ExtensionContext context) {
          Set<AutoCloseable> unclosed = new HashSet<>();
          for (AutoCloseable res : OPEN_RESOURCES.get()) {
              // 针对FileOutputStream，通过反射检查文件描述符是否有效
              if (res instanceof FileOutputStream) {
                  try {
                      Field fdField = FileOutputStream.class.getDeclaredField("fd");
                      fdField.setAccessible(true);
                      FileDescriptor fd = (FileDescriptor) fdField.get(res);
                      if (fd.valid()) {
                          unclosed.add(res);
                      }
                  } catch (Exception e) {
                      unclosed.add(res);
                  }
              }
          }
  
          if (!unclosed.isEmpty()) {
              throw new AssertionError("检测到未关闭资源: " + unclosed);
          }
          OPEN_RESOURCES.remove();
      }
  }
  

  测试类上添加@ExtendWith(ResourceLeakDetector.class)，创建资源时用ResourceLeakDetector.track(new FileOutputStream(...))即可自动检测。

• @TempDir专属校验逻辑
  针对@TempDir删除失败的场景，在@AfterEach中主动尝试清理目录，失败时触发资源排查：

  @AfterEach
  void verifyTempDirCleanup(@TempDir Path tempDir) {
      try {
          Files.walk(tempDir)
              .sorted(Comparator.reverseOrder())
              .map(Path::toFile)
              .forEach(File::delete);
      } catch (IOException e) {
          // 打印当前进程打开的文件句柄辅助排查
          OperatingSystemMXBean osBean = ManagementFactory.getOperatingSystemMXBean();
          if (osBean instanceof com.sun.management.UnixOperatingSystemMXBean) {
              System.out.println("当前打开文件句柄数: " + 
                  ((com.sun.management.UnixOperatingSystemMXBean) osBean).getOpenFileDescriptorCount());
          }
          throw new AssertionError("临时目录删除失败，疑似资源泄漏", e);
      }
  }
  

第三方工具与JVM层面方案 #

• 静态代码检测：SpotBugs
  用SpotBugs的OS_OPEN_STREAM规则，在编译或测试阶段直接扫描代码中未关闭的流、连接等资源，提前在代码层面发现问题，无需运行时检测。

• 字节码增强自动跟踪
  用Byte Buddy或ASM做字节码拦截，自动给所有AutoCloseable实现类的close()方法添加标记逻辑，测试结束后统计未关闭实例，无需修改业务代码。

• JVM原生监控
  测试中通过JDK自带API监控资源使用：

  @AfterEach
  void checkResourceUsage() {
      OperatingSystemMXBean osBean = ManagementFactory.getOperatingSystemMXBean();
      if (osBean instanceof com.sun.management.UnixOperatingSystemMXBean) {
          long openFiles = ((com.sun.management.UnixOperatingSystemMXBean) osBean).getOpenFileDescriptorCount();
          if (openFiles > 150) { // 自定义阈值
              throw new AssertionError("打开文件句柄数超标，疑似资源泄漏");
          }
      }
  }
  

  也可以用jcmd或jconsole在测试运行时实时查看进程的文件句柄、连接等资源状态。

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