sdmxdl.provider.ri.monitors.UpptimeMonitor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sdmx-dl-provider-ri Show documentation
Easily download official statistics - RI
The newest version!
package sdmxdl.provider.ri.monitors;

import nbbrd.io.http.HttpClient;
import lombok.NonNull;
import nbbrd.design.DirectImpl;
import nbbrd.design.VisibleForTesting;
import nbbrd.io.text.Parser;
import nbbrd.service.ServiceProvider;
import sdmxdl.ext.Cache;
import sdmxdl.provider.ri.drivers.RiHttpUtils;
import sdmxdl.provider.web.WebMonitors;
import sdmxdl.web.MonitorReport;
import sdmxdl.web.MonitorReports;
import sdmxdl.web.MonitorStatus;
import sdmxdl.web.WebSource;
import sdmxdl.web.spi.Monitor;
import sdmxdl.web.spi.WebContext;

import java.io.IOException;
import java.text.NumberFormat;
import java.time.Clock;
import java.time.Duration;
import java.util.Locale;

@DirectImpl
@ServiceProvider
public final class UpptimeMonitor implements Monitor {

    @Override
    public @NonNull String getMonitorId() {
        return "UPPPTIME";
    }

    @Override
    public @NonNull String getMonitorUriScheme() {
        return UpptimeId.URI_SCHEME;
    }

    @Override
    public @NonNull MonitorReport getReport(@NonNull WebSource source, @NonNull WebContext context) throws IOException, IllegalArgumentException {
        WebMonitors.checkMonitor(source.getMonitor(), getMonitorUriScheme());

        UpptimeId id = UpptimeId.parse(source.getMonitor());

        Cache cache = context.getMonitorCache(source);
        String key = id.toSummaryURL().toString();

        MonitorReports reports = cache.get(key);

        if (reports == null) {
            reports = createReports(RiHttpUtils.newClient(source, context), id, cache.getClock());
            cache.put(key, reports);
        }

        return reports.getReports()
                .stream()
                .filter(item -> item.getSource().equals(id.getSite()))
                .findFirst()
                .orElseThrow(IOException::new);
    }

    private MonitorReports createReports(HttpClient client, UpptimeId base, Clock clock) throws IOException {
        MonitorReports.Builder result = MonitorReports.builder().uriScheme(getMonitorUriScheme());
        for (UpptimeSummary summary : UpptimeSummary.request(client, base.toSummaryURL())) {
            result.report(getReport(summary));
        }
        return result.ttl(clock.instant(), Duration.ofMinutes(5)).build();
    }

    @VisibleForTesting
    static MonitorReport getReport(UpptimeSummary summary) {
        return MonitorReport
                .builder()
                .source(summary.getName())
                .status(parseStatus(summary.getStatus()))
                .uptimeRatio(parseUptimeRatio(summary.getUptime()))
                .averageResponseTime(summary.getTime())
                .build();
    }

    private static MonitorStatus parseStatus(String status) {
        switch (status) {
            case "up":
                return MonitorStatus.UP;
            case "down":
                return MonitorStatus.DOWN;
            default:
                return MonitorStatus.UNKNOWN;
        }
    }

    private static Double parseUptimeRatio(String uptime) {
        // onNumberFormat parser is NOT thread-safe!
        return Parser
                .onNumberFormat(NumberFormat.getPercentInstance(Locale.ROOT))
                .parseValue(uptime)
                .map(Number::doubleValue)
                .orElse(null);
    }
}