我正在为在网页上查找WebElement的重试机制实现一个 Package 器方法，我的要求是如果元素不可用，在失败之前最多重试3次我如何实现这个方法，下面是我的代码，它正确吗？

public static WebElement findElementWithRetry(WebDriver driver, By by, int retryCount){
            WebElement element = null;
            try {
                wait = new WebDriverWait(driver, 30);
                element = wait.until(ExpectedConditions.visibilityOfElementLocated(by));
            } catch (Exception e) {
                //TODO: handle exception
                for(int i=0; i<retryCount; i++){
                    element = driver.findElement(by);
                    if(element.isDisplayed())
                        return element;
                }
            }
            return element;
        }