调用AutoIt脚本完成DB重置后，如何将控制权交还给Selenium？

我之前帮不少开发者解决过类似的问题——当用Selenium调用外部桌面程序时，经常会因为进程没正常收尾或者缓冲区阻塞导致控制权卡在外边。结合你的场景，给你几个针对性的解决办法：

1. 必须等待外部进程完全终止 #

你之前直接调用Runtime.getRuntime().exec()是异步的，Selenium不会自动等这个程序跑完就继续往下走，甚至会因为进程没退出而卡住。正确的做法是获取进程对象，等待它执行完毕：

// 启动外部程序并获取进程实例
Process dbResetProcess = Runtime.getRuntime().exec("E:\\AutoIT\\DBReset.exe");

// 关键：等待进程执行完成，拿到退出码
int exitCode = dbResetProcess.waitFor();

// 根据退出码判断是否成功，0一般表示正常结束
if (exitCode == 0) {
    System.out.println("DB重置搞定，控制权回到Selenium啦");
    // 这里继续你的Selenium测试逻辑
} else {
    System.err.println("DB重置出问题了，退出码：" + exitCode);
}

⚠️ 注意：如果这个DBReset.exe会输出大量日志，一定要同时读取它的标准输出和错误输出，不然缓冲区满了会导致进程挂起，永远等不到结束。可以加两个线程来处理：

Process dbResetProcess = Runtime.getRuntime().exec("E:\\AutoIT\\DBReset.exe");

// 启动线程读取标准输出，避免阻塞
new Thread(() -> {
    try (BufferedReader reader = new BufferedReader(new InputStreamReader(dbResetProcess.getInputStream()))) {
        String line;
        while ((line = reader.readLine()) != null) {
            System.out.println("DBReset日志：" + line);
        }
    } catch (IOException e) {
        e.printStackTrace();
    }
}).start();

// 启动线程读取错误输出
new Thread(() -> {
    try (BufferedReader reader = new BufferedReader(new InputStreamReader(dbResetProcess.getErrorStream()))) {
        String line;
        while ((line = reader.readLine()) != null) {
            System.err.println("DBReset错误日志：" + line);
        }
    } catch (IOException e) {
        e.printStackTrace();
    }
}).start();

// 等待进程结束
int exitCode = dbResetProcess.waitFor();
// 后续逻辑...

2. 检查你的AutoIT脚本是否收尾干净 #

很多时候问题出在AutoIT脚本上：你点击了重置按钮，但程序窗口没关闭、进程没退出，导致Selenium一直等不到信号。在AutoIT脚本最后加上收尾逻辑：

; 假设重置完成后会弹出提示窗口，先等它出现再关闭
WinWaitActive("DB重置完成")
WinClose("DB重置完成")

; 保险起见，直接结束DBReset.exe进程
ProcessClose("DBReset.exe")

这样外部程序会彻底退出，Selenium的waitFor()就能收到退出信号，顺利拿回控制权。

3. 改用ProcessBuilder替代Runtime.exec #

ProcessBuilder是Java 5+推荐的进程启动方式，比Runtime.exec更灵活，能自动处理输出流合并，减少阻塞概率：

ProcessBuilder pb = new ProcessBuilder("E:\\AutoIT\\DBReset.exe");
// 将错误输出合并到标准输出，不用单独处理错误流
pb.redirectErrorStream(true);

Process process = pb.start();

// 读取合并后的输出
try (BufferedReader reader = new BufferedReader(new InputStreamReader(process.getInputStream()))) {
    String line;
    while ((line = reader.readLine()) != null) {
        System.out.println("DBReset输出：" + line);
    }
}

// 等待进程结束
int exitCode = process.waitFor();
// 继续Selenium测试...

4. 处理窗口焦点问题（可选） #

如果进程结束后Selenium还是操作不了浏览器，可能是外部程序抢占了焦点。可以让Selenium重新获取浏览器窗口的控制权：

// 切换回当前浏览器窗口
driver.switchTo().window(driver.getWindowHandle());
// 或者最大化窗口确保焦点在浏览器上
driver.manage().window().maximize();

按上面的步骤调整后，应该就能解决控制权无法返回的问题了。

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