sdmxdl.file.SdmxFileManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sdmx-dl-api Show documentation
Easily download official statistics - API
There is a newer version: 3.0.0-beta.13
/*
 * Copyright 2017 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.file;

import lombok.AccessLevel;
import org.checkerframework.checker.nullness.qual.NonNull;
import sdmxdl.LanguagePriorityList;
import sdmxdl.SdmxManager;
import sdmxdl.ext.SdmxCache;
import sdmxdl.ext.SdmxException;
import sdmxdl.ext.spi.SdmxDialect;
import sdmxdl.ext.spi.SdmxDialectLoader;
import sdmxdl.file.spi.SdmxFileContext;
import sdmxdl.file.spi.SdmxFileReader;
import sdmxdl.file.spi.SdmxFileReaderLoader;

import java.io.IOException;
import java.util.List;
import java.util.Objects;
import java.util.Optional;

/**
 * @author Philippe Charles
 */
@lombok.Value
@lombok.Builder(toBuilder = true)
public class SdmxFileManager implements SdmxManager {

    @NonNull
    public static SdmxFileManager ofServiceLoader() {
        return builder()
                .dialects(SdmxDialectLoader.load())
                .readers(SdmxFileReaderLoader.load())
                .build();
    }

    @lombok.NonNull
    @lombok.Builder.Default
    LanguagePriorityList languages = LanguagePriorityList.ANY;

    @lombok.NonNull
    @lombok.Builder.Default
    SdmxCache cache = SdmxCache.noOp();

    @lombok.NonNull
    @lombok.Builder.Default
    SdmxFileListener eventListener = SdmxFileListener.getDefault();

    @lombok.NonNull
    @lombok.Singular
    List dialects;

    @lombok.NonNull
    @lombok.Singular
    List readers;

    @lombok.NonNull
    @lombok.Getter(lazy = true, value = AccessLevel.PRIVATE)
    SdmxFileContext context = initContext();

    @Override
    public SdmxFileConnection getConnection(String name) throws IOException {
        Objects.requireNonNull(name);

        SdmxFileSource source = lookupSource(name)
                .orElseThrow(() -> SdmxException.missingSource(name, SdmxFileSource.class));

        return getConnection(source);
    }

    @NonNull
    public SdmxFileConnection getConnection(@NonNull SdmxFileSource source) throws IOException {
        Objects.requireNonNull(source);

        SdmxFileReader reader = lookupReader(source)
                .orElseThrow(() -> new IOException("cannot find reader for source '" + source + "'"));

        return reader.read(source, getContext());
    }

    private SdmxFileContext initContext() {
        return SdmxFileContext
                .builder()
                .languages(languages)
                .eventListener(eventListener)
                .cache(cache)
                .dialects(dialects)
                .build();
    }

    private Optional lookupSource(String name) {
        return readers.stream()
                .map(reader -> reader.getSource(name))
                .filter(Objects::nonNull)
                .findFirst();
    }

    private Optional lookupReader(SdmxFileSource source) {
        return readers.stream()
                .filter(reader -> reader.canRead(source))
                .findFirst();
    }
}