您能在Java和Selenium中为显式等待创建一个泛型或可重用的方法吗？

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

public class WaitUtils {
    public static <T> T waitForElement(WebDriver driver, By locator, int timeout) {
        WebDriverWait wait = new WebDriverWait(driver, timeout);
        WebElement element = wait.until(ExpectedConditions.visibilityOfElementLocated(locator));
        return (T) element;
    }
}

WebDriver driver = new ChromeDriver();
driver.get("https://www.example.com");

By searchBoxLocator = By.id("search-box");
WebElement searchBox = WaitUtils.waitForElement(driver, searchBoxLocator, 10);
searchBox.sendKeys("Hello World");