io.github.mivek.provider.airport.impl.DefaultAirportProvider Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of metarParser-spi Show documentation
There is a newer version: 2.16.0
package io.github.mivek.provider.airport.impl;

import io.github.mivek.model.Airport;
import io.github.mivek.model.Country;
import io.github.mivek.provider.airport.AirportProvider;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.nio.charset.StandardCharsets;
import java.util.HashMap;
import java.util.Map;
import java.util.Objects;
import org.apache.commons.csv.CSVFormat;
import org.apache.commons.csv.CSVParser;
import org.apache.commons.csv.CSVRecord;

/**
 * Default implementation of the AiportProvider using local files to build the airport map.
 *
 * @author mivek
 */
public final class DefaultAirportProvider implements AirportProvider {
    /** Path of airport file. */
    private final InputStream airportsFile = DefaultAirportProvider.class.getClassLoader().getResourceAsStream("data/airports.dat");
    /** Path of countries file. */
    private final InputStream countriesFile = DefaultAirportProvider.class.getClassLoader().getResourceAsStream("data/countries.dat");
    /** Map of countries. */
    private Map countries;
    /** Map of airports. */
    private Map airports;

    /**
     * Default constructor.
     */
    public DefaultAirportProvider() {
        initCountries();
        initAirports();
    }

    /**
     * Initiate countries map.
     */
    private void initCountries() {
        Objects.requireNonNull(countriesFile);
        countries = new HashMap<>();
        try (CSVParser parser = CSVFormat.DEFAULT.parse(new InputStreamReader(countriesFile, StandardCharsets.UTF_8))) {
            for (CSVRecord line : parser) {
                Country country = new Country();
                country.setName(line.get(0));
                countries.put(country.getName(), country);
            }
        } catch (IOException exception) {
            throw new IllegalStateException(exception.getMessage());
        }
    }

    /**
     * Initiate airports map.
     */
    private void initAirports() {
        Objects.requireNonNull(airportsFile);
        airports = new HashMap<>();
        try (CSVParser parser = CSVFormat.DEFAULT.parse(new InputStreamReader(airportsFile, StandardCharsets.UTF_8))) {
            for (CSVRecord line : parser) {
                Airport airport = new Airport();
                airport.setName(line.get(1));
                airport.setCity(line.get(2));
                airport.setCountry(countries.get(line.get(3)));
                airport.setIata(line.get(4));
                airport.setIcao(line.get(5));
                airport.setLatitude(Double.parseDouble(line.get(6)));
                airport.setLongitude(Double.parseDouble(line.get(7)));
                airport.setAltitude(Integer.parseInt(line.get(8)));
                airport.setTimezone(line.get(9));
                airport.setDst(line.get(10));
                airports.put(airport.getIcao(), airport);
            }
        } catch (IOException exception) {
            throw new IllegalStateException(exception.getMessage());
        }
    }

    @Override
    public Map getAirports() {
        return airports;
    }
}