Selenium WebDriver：构建健壮的元素查找重试策略

import org.openqa.selenium.By;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.TimeoutException;
import org.openqa.selenium.support.ui.ExpectedConditions;
import org.openqa.selenium.support.ui.WebDriverWait;
import java.time.Duration; // 适用于 Selenium 4 及以上版本

public class ElementFinder {

    /**
     * 在指定重试次数内查找Web元素。
     * 每次尝试都会使用显式等待来确保元素可见。
     *
     * @param driver     WebDriver实例
     * @param by         元素的定位器 (By对象)
     * @param retryCount 最大重试次数
     * @param waitTimeoutPerAttempt 每次尝试的显式等待超时时间 (秒)
     * @return 找到的WebElement
     * @throws org.openqa.selenium.TimeoutException 如果在所有重试后仍未找到元素
     */
    public static WebElement findElementWithRetry(WebDriver driver, By by, int retryCount, int waitTimeoutPerAttempt) {
        for (int i = 1; i <= retryCount; i++) {
            try {
                // 每次尝试都使用WebDriverWait进行显式等待
                // Selenium 4+ 推荐使用 Duration.ofSeconds
                WebDriverWait wait = new WebDriverWait(driver, Duration.ofSeconds(waitTimeoutPerAttempt));
                WebElement element = wait.until(ExpectedConditions.visibilityOfElementLocated(by));

                // 检查元素是否可见，虽然visibilityOfElementLocated已经包含了这个含义，
                // 但作为额外的确认或针对更复杂的ExpectedConditions，此检查仍有价值。
                if (element.isDisplayed()) {
                    System.out.println(String.format("尝试 %d 次后成功找到元素: %s", i, by.toString()));
                    return element;
                }
            } catch (TimeoutException e) {
                // 如果当前尝试超时，则忽略异常，继续下一次重试
                System.out.println(String.format("尝试 %d 次查找元素 %s 超时，进行下一次重试...", i, by.toString()));
                // 可选：在这里添加短暂的Thread.sleep() 来模拟用户行为或等待页面稳定
                // try { Thread.sleep(500); } catch (InterruptedException ie) { Thread.currentThread().interrupt(); }
            } catch (Exception e) {
                // 捕获其他可能的异常，例如StaleElementReferenceException等，并记录
                System.err.println(String.format("尝试 %d 次查找元素 %s 时发生非超时异常: %s", i, by.toString(), e.getMessage()));
                // 如果是致命错误，可能需要在此处重新抛出或采取其他措施
                throw e; // 如果是其他意外异常，可能需要立即抛出
            }
        }
        // 所有重试失败后，抛出异常
        throw new TimeoutException(String.format("元素 %s 在 %d 次重试后仍未找到。", by.toString(), retryCount));
    }

    // 示例用法
    public static void main(String[] args) {
        // 实际使用时，请初始化您的WebDriver实例，例如：
        // WebDriver driver = new ChromeDriver();
        // driver.get("http://example.com"); // 导航到目标网页

        // 模拟 driver 和 by 对象，实际运行时需要替换为真实的WebDriver和By对象
        WebDriver driver = null; 
        By exampleBy = By.id("someElementId"); // 替换为实际的定位器，例如 By.xpath("//button[@id='submit']")

        try {
            // 尝试查找元素，最多重试3次，每次等待10秒
            WebElement element = findElementWithRetry(driver, exampleBy, 3, 10);
            // 对找到的元素执行操作，例如：
            // element.click();
            System.out.println("元素成功找到并可操作。");
        } catch (TimeoutException e) {
            System.err.println("错误：元素查找超时 - " + e.getMessage());
        } catch (Exception e) {
            System.err.println("发生其他错误：" + e.getMessage());
        } finally {
            // 在实际应用中，确保在测试结束后关闭WebDriver实例
            // if (driver != null) {
            //     driver.quit();
            // }
        }
    }
}