demetra.cli.tsproviders.Uri2Ts.java Source code

Introduction

Here is the source code for demetra.cli.tsproviders.Uri2Ts.java

Source

/*
 * Copyright 2015 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 demetra.cli.tsproviders;

import com.google.common.annotations.VisibleForTesting;
import be.nbb.cli.util.joptsimple.JOptSimpleArgsParser;
import com.google.common.collect.Iterables;
import be.nbb.cli.util.BasicCliLauncher;
import static be.nbb.cli.util.joptsimple.ComposedOptionSpec.newOutputOptionsSpec;
import static be.nbb.cli.util.joptsimple.ComposedOptionSpec.newStandardOptionsSpec;
import be.nbb.cli.util.OutputOptions;
import be.nbb.cli.util.StandardOptions;
import ec.tss.ITsProvider;
import ec.tss.TsCollectionInformation;
import ec.tss.TsInformationType;
import ec.tss.tsproviders.IFileLoader;
import ec.tss.xml.XmlTsCollection;
import java.net.URI;
import java.util.ServiceLoader;
import joptsimple.OptionSet;
import joptsimple.OptionSpec;
import be.nbb.cli.util.BasicCommand;
import be.nbb.cli.util.proc.CommandRegistration;
import be.nbb.cli.util.joptsimple.ComposedOptionSpec;
import demetra.cli.helpers.XmlUtil;

/**
 * Retrieves time series from an URI.
 *
 * @author Philippe Charles
 */
public final class Uri2Ts implements BasicCommand<Uri2Ts.Parameters> {

    @CommandRegistration
    public static void main(String[] args) {
        BasicCliLauncher.run(args, Parser::new, Uri2Ts::new, o -> o.so);
    }

    public static final class Parameters {

        StandardOptions so;
        public URI uri;
        public OutputOptions output;
    }

    @Override
    public void exec(Parameters params) throws Exception {
        Iterable<ITsProvider> providers = ServiceLoader.load(ITsProvider.class);
        for (IFileLoader o : Iterables.filter(providers, IFileLoader.class)) {
            ProviderTool.getDefault().applyWorkingDir(o);
        }
        TsCollectionInformation result = ProviderTool.getDefault().getTsCollection(providers, params.uri,
                TsInformationType.All);
        XmlUtil.writeValue(params.output, XmlTsCollection.class, result);
        for (ITsProvider o : providers) {
            o.dispose();
        }
    }

    @VisibleForTesting
    static final class Parser extends JOptSimpleArgsParser<Parameters> {

        private final ComposedOptionSpec<StandardOptions> so = newStandardOptionsSpec(parser);
        private final OptionSpec<URI> uri = parser.nonOptions("uri").ofType(URI.class);
        private final ComposedOptionSpec<OutputOptions> output = newOutputOptionsSpec(parser);

        @Override
        protected Parameters parse(OptionSet o) {
            Parameters result = new Parameters();
            result.uri = o.has(uri) ? uri.value(o) : null;
            result.output = output.value(o);
            result.so = so.value(o);
            return result;
        }
    }
}