sdmxdl.cli.CheckStatusCommand Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sdmx-dl-cli Show documentation
Easily download official statistics - CLI
The newest version!
/*
 * Copyright 2018 National Bank of Belgium
 *
 * Licensed under the EUPL, Version 1.1 or - as soon they will be approved
 * by the European Commission - subsequent versions of the EUPL (the "Licence");
 * You may not use this work except in compliance with the Licence.
 * You may obtain a copy of the Licence at:
 *
 * http://ec.europa.eu/idabc/eupl
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the Licence is distributed on an "AS IS" basis,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the Licence for the specific language governing permissions and
 * limitations under the Licence.
 */
package sdmxdl.cli;

import internal.sdmxdl.cli.SortOptions;
import internal.sdmxdl.cli.WebSourcesOptions;
import internal.sdmxdl.cli.ext.CsvTable;
import internal.sdmxdl.cli.ext.RFC4180OutputOptions;
import lombok.AccessLevel;
import lombok.NonNull;
import nbbrd.io.text.Formatter;
import org.checkerframework.checker.nullness.qual.Nullable;
import picocli.CommandLine;
import sdmxdl.web.MonitorReport;
import sdmxdl.web.SdmxWebManager;
import sdmxdl.web.WebSource;

import java.io.IOException;
import java.util.Comparator;
import java.util.concurrent.Callable;
import java.util.stream.Stream;

/**
 * @author Philippe Charles
 */
@CommandLine.Command(name = "status")
@SuppressWarnings("FieldMayBeFinal")
public final class CheckStatusCommand implements Callable {

    @CommandLine.Mixin
    private WebSourcesOptions web;

    @CommandLine.Mixin
    private final RFC4180OutputOptions csv = new RFC4180OutputOptions();

    @CommandLine.Mixin
    private SortOptions sort;

    @Override
    public Void call() throws Exception {
        getTable().write(csv, getRows());
        return null;
    }

    private CsvTable getTable() {
        return CsvTable
                .builderOf(Status.class)
                .columnOf("Source", Status::getSource)
                .columnOf("Status", Status::getStatus, Formatter.onObjectToString())
                .columnOf("UptimeRatio", Status::getUptimeRatio, Formatter.onObjectToString())
                .columnOf("AverageResponseTime", Status::getAverageResponseTime, Formatter.onObjectToString())
                .columnOf("ErrorMessage", Status::getCause)
                .build();
    }

    private Stream getRows() throws IOException {
        SdmxWebManager manager = web.loadManager();
        Stream sources = web.isAllSources() ? WebSourcesOptions.getAllSourceNames(manager) : web.getSources().stream();
        return sort.applySort(web.applyParallel(sources).map(sourceName -> Status.of(manager, sourceName)), BY_SOURCE);
    }

    private static final Comparator BY_SOURCE = Comparator.comparing(Status::getSource);

    @lombok.AllArgsConstructor(access = AccessLevel.PRIVATE)
    @lombok.Value
    private static class Status {

        static @NonNull Status of(@NonNull SdmxWebManager manager, @NonNull String sourceName) {
            WebSource source = manager.getSources().get(sourceName);
            if (source == null) {
                return failure(sourceName, "Cannot find source");
            }
            if (source.getMonitor() == null) {
                return failure(sourceName, "No monitor defined");
            }
            try {
                return success(manager.getMonitorReport(source));
            } catch (IOException ex) {
                return failure(sourceName, ex);
            }
        }

        static @NonNull Status success(@NonNull MonitorReport report) {
            return new Status(report, null);
        }

        static @NonNull Status failure(@NonNull String source, @NonNull IOException cause) {
            return new Status(MonitorReport.builder().source(source).build(), cause.getMessage());
        }

        static @NonNull Status failure(@NonNull String source, @NonNull String cause) {
            return new Status(MonitorReport.builder().source(source).build(), cause);
        }

        @lombok.NonNull
        @lombok.experimental.Delegate
        MonitorReport report;

        @Nullable
        String cause;
    }
}