ViacheslavKlimov
4 years ago
9 changed files with 28 additions and 144 deletions
@ -1,113 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2022 The Thingsboard Authors |
|
||||
* |
|
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|
||||
* you may not use this file except in compliance with the License. |
|
||||
* You may obtain a copy of the License at |
|
||||
* |
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
|
||||
* |
|
||||
* Unless required by applicable law or agreed to in writing, software |
|
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|
||||
* See the License for the specific language governing permissions and |
|
||||
* limitations under the License. |
|
||||
*/ |
|
||||
package org.thingsboard.monitoring.service; |
|
||||
|
|
||||
import lombok.RequiredArgsConstructor; |
|
||||
import lombok.extern.slf4j.Slf4j; |
|
||||
import org.openqa.selenium.WebDriver; |
|
||||
import org.openqa.selenium.chrome.ChromeOptions; |
|
||||
import org.openqa.selenium.remote.RemoteWebDriver; |
|
||||
import org.openqa.selenium.support.ui.WebDriverWait; |
|
||||
import org.springframework.beans.factory.annotation.Value; |
|
||||
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; |
|
||||
import org.springframework.boot.context.event.ApplicationReadyEvent; |
|
||||
import org.springframework.context.event.EventListener; |
|
||||
import org.springframework.stereotype.Component; |
|
||||
import org.thingsboard.monitoring.data.Latencies; |
|
||||
import org.thingsboard.monitoring.data.MonitoredServiceKey; |
|
||||
import org.thingsboard.monitoring.util.TbStopWatch; |
|
||||
|
|
||||
import java.net.MalformedURLException; |
|
||||
import java.net.URL; |
|
||||
import java.time.Duration; |
|
||||
import java.util.concurrent.ScheduledExecutorService; |
|
||||
import java.util.concurrent.TimeUnit; |
|
||||
|
|
||||
import static org.openqa.selenium.By.xpath; |
|
||||
import static org.openqa.selenium.support.ui.ExpectedConditions.elementToBeClickable; |
|
||||
import static org.openqa.selenium.support.ui.ExpectedConditions.urlContains; |
|
||||
|
|
||||
@Component |
|
||||
@RequiredArgsConstructor |
|
||||
@Slf4j |
|
||||
@ConditionalOnProperty(name = "monitoring.ui.enabled", havingValue = "true") |
|
||||
public class WebUiHealthChecker { |
|
||||
|
|
||||
@Value("${monitoring.ui.url}") |
|
||||
private String url; |
|
||||
@Value("${monitoring.auth.username}") |
|
||||
private String username; |
|
||||
@Value("${monitoring.auth.password}") |
|
||||
private String password; |
|
||||
@Value("${monitoring.ui.monitoring_rate_sec}") |
|
||||
private int monitoringRateSec; |
|
||||
@Value("${monitoring.rest_request_timeout_ms}") |
|
||||
private int timeoutMs; |
|
||||
@Value("${monitoring.ui.remote_webdriver_url}") |
|
||||
private String remoteWebdriverUrl; |
|
||||
|
|
||||
private final MonitoringReporter monitoringReporter; |
|
||||
private final ScheduledExecutorService monitoringExecutor; |
|
||||
private final TbStopWatch stopWatch; |
|
||||
|
|
||||
private static final String EMAIL_FIELD = "//input[@id='username-input']"; |
|
||||
private static final String PASSWORD_FIELD = "//input[@id='password-input']"; |
|
||||
private static final String SUBMIT_BTN = "//button[@type='submit']"; |
|
||||
private static final String DEVICES_BTN = "//mat-toolbar//a[@href='/devices']"; |
|
||||
|
|
||||
@EventListener(ApplicationReadyEvent.class) |
|
||||
public void startMonitoring() { |
|
||||
monitoringExecutor.scheduleWithFixedDelay(() -> { |
|
||||
RemoteWebDriver driver = null; |
|
||||
try { |
|
||||
driver = createDriver(); |
|
||||
WebDriverWait wait = createDriverWait(driver); |
|
||||
driver.manage().window().maximize(); |
|
||||
driver.get(url + "/login"); |
|
||||
|
|
||||
try { |
|
||||
stopWatch.start(); |
|
||||
wait.until(elementToBeClickable(xpath(EMAIL_FIELD))).sendKeys(username); |
|
||||
wait.until(elementToBeClickable(xpath(PASSWORD_FIELD))).sendKeys(password); |
|
||||
wait.until(elementToBeClickable(xpath(SUBMIT_BTN))).click(); |
|
||||
monitoringReporter.reportLatency(Latencies.WEB_UI_LOAD, stopWatch.getTime()); |
|
||||
|
|
||||
wait.until(urlContains("/home")); |
|
||||
wait.until(elementToBeClickable(xpath(DEVICES_BTN))).click(); |
|
||||
} catch (Exception e) { |
|
||||
throw new RuntimeException("Expected web UI elements were not displayed", e); |
|
||||
} |
|
||||
|
|
||||
monitoringReporter.serviceIsOk(MonitoredServiceKey.WEB_UI); |
|
||||
} catch (Exception e) { |
|
||||
monitoringReporter.serviceFailure(MonitoredServiceKey.WEB_UI, e); |
|
||||
} finally { |
|
||||
if (driver != null) driver.quit(); |
|
||||
} |
|
||||
}, 0, monitoringRateSec, TimeUnit.SECONDS); |
|
||||
} |
|
||||
|
|
||||
private RemoteWebDriver createDriver() throws MalformedURLException { |
|
||||
ChromeOptions options = new ChromeOptions(); |
|
||||
options.setPageLoadTimeout(Duration.ofMillis(timeoutMs)); |
|
||||
return new RemoteWebDriver(new URL(remoteWebdriverUrl + "/wd/hub"), options); |
|
||||
} |
|
||||
|
|
||||
private WebDriverWait createDriverWait(WebDriver driver) { |
|
||||
return new WebDriverWait(driver, Duration.ofMillis(timeoutMs)); |
|
||||
} |
|
||||
|
|
||||
} |
|
||||
Loading…
Reference in new issue