Selenium中实现健壮的元素查找重试机制

public static WebElement findElementWithRetry(WebDriver driver, By by, int retryCount){
    WebElement element = null;
    try {
        WebDriverWait wait = new WebDriverWait(driver, 30);
        element = wait.until(ExpectedConditions.visibilityOfElementLocated(by));
    } catch (Exception e) {
        // 错误处理：在这里进行重试
        for(int i=0; i<retryCount; i++){
            // 注意：这里直接使用 findElement，没有再次等待
            element = driver.findElement(by);
            if(element.isDisplayed())
                return element;
        }
    }
    return element;
}

import org.openqa.selenium.By;
import org.openqa.selenium.TimeoutException;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.support.ui.ExpectedConditions;
import org.openqa.selenium.support.ui.WebDriverWait;

import java.time.Duration; // For Selenium 4+

public class ElementFinder {

    /**
     * 在指定次数内尝试查找并返回可见的WebElement。
     * 如果在所有重试后仍未找到元素，则抛出运行时异常。
     *
     * @param driver     WebDriver实例。
     * @param by         用于定位元素的By策略。
     * @param retryCount 重试的最大次数（包括第一次尝试）。
     * @param timeoutInSeconds 每次尝试的显式等待超时时间（秒）。
     * @return 成功找到并可见的WebElement。
     * @throws RuntimeException 如果在所有重试后仍未找到元素。
     */
    public static WebElement findElementWithRetry(WebDriver driver, By by, int retryCount, int timeoutInSeconds) {
        for (int i = 1; i <= retryCount; i++) {
            try {
                // 每次重试都使用 WebDriverWait 进行显式等待
                // For Selenium 4+ use Duration.ofSeconds()
                WebDriverWait wait = new WebDriverWait(driver, Duration.ofSeconds(timeoutInSeconds));
                WebElement element = wait.until(ExpectedConditions.visibilityOfElementLocated(by));

                // visibilityOfElementLocated 已经确保了元素是可见的，因此 isDisplayed() 检查通常是冗余的
                return element; // 元素已找到且可见，直接返回

            } catch (TimeoutException e) {
                // 捕获 TimeoutException，表示当前尝试未能在指定时间内找到元素
                System.out.printf("尝试查找元素 '%s' 失败 (第 %d/%d 次重试).%n", by.toString(), i, retryCount);
                // 继续下一次重试
            } catch (Exception e) {
                // 捕获其他可能的异常，例如 NoSuchElementException（虽然visibilityOfElementLocated会处理）
                // 或StaleElementReferenceException等，但TimeoutException是最常见的失败原因。
                System.out.printf("尝试查找元素 '%s' 遇到未知错误 (第 %d/%d 次重试): %s%n", by.toString(), i, retryCount, e.getMessage());
                // 通常，对于元素查找，TimeoutException是唯一需要重试的。
            }
        }
        // 所有重试均失败，抛出异常
        throw new RuntimeException(String.format("在 %d 次重试后，元素 '%s' 仍未找到或不可见。", retryCount, by.toString()));
    }

    // 示例用法
    public static void main(String[] args) {
        // 假设 driver 已经被初始化并导航到某个页面
        // WebDriver driver = new ChromeDriver();
        // driver.get("http://example.com");

        // try {
        //     WebElement element = findElementWithRetry(driver, By.id("someElementId"), 3, 10);
        //     System.out.println("成功找到元素: " + element.getText());
        // } catch (RuntimeException e) {
        //     System.err.println("查找元素失败: " + e.getMessage());
        // } finally {
        //     // driver.quit();
        // }
    }
}