CrestDataHandler.java

package net.troja.eve.crest;

/*
 * ========================================================================
 * Library for the Eve Online CREST API
 * ------------------------------------------------------------------------
 * Copyright (C) 2014 - 2015 Jens Oberender <j.obi@troja.net>
 * ------------------------------------------------------------------------
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * ========================================================================
 */

import java.util.EnumMap;
import java.util.Map;
import java.util.Map.Entry;

import net.troja.eve.crest.beans.Status;
import net.troja.eve.crest.processors.CrestApiProcessor;
import net.troja.eve.crest.processors.StatusProcessor;
import net.troja.eve.crest.utils.ProcessorConfiguration;

import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;

/**
 * Backend class that handles the communication and the processors.
 * <p>
 * Do not use this class, use {@link CrestHandler} instead as it includes the caching of the data.
 */
public abstract class CrestDataHandler {
    private static final Logger LOGGER = LogManager.getLogger(CrestDataHandler.class);

    public enum DataType {
        ITEM_TYPE, MARKET_PRICE, INDUSTRY_SYSTEM, INDUSTRY_FACILITY
    }

    private CrestDataProcessor processor = new CrestDataProcessor();
    private final Map<DataType, ProcessorConfiguration<?>> dataProcessors = new EnumMap<>(DataType.class);
    private final Map<DataType, Long> lastUpdates = new EnumMap<>(DataType.class);
    private final StatusProcessor statusProcessor = new StatusProcessor();

    protected CrestDataHandler() {
        super();
    }

    protected void addProcessorConfiguration(final DataType type, final ProcessorConfiguration<?> config) {
        dataProcessors.put(type, config);
    }

    /**
     * Enable fetching of the data in the background for the given types.
     *
     * @param types
     *            DataType(s) to enable
     */
    public void enableDataPrefetching(final DataType... types) {
        for (final DataType type : types) {
            dataProcessors.get(type).setEnabled(true);
        }
    }

    /**
     * Enable fetching of the data in the background for the given types.
     *
     * @param types
     *            DataType(s) to enable
     */
    public void disableDataPrefetching(final DataType... types) {
        for (final DataType type : types) {
            dataProcessors.get(type).setEnabled(false);
        }
    }

    public void setProcessor(final CrestDataProcessor processor) {
        this.processor = processor;
    }

    public Status getServerStatus() {
        return processor.downloadAndProcessData(statusProcessor);
    }

    /**
     * Update the cached data manually. Use this function of you don't want it fetched regularly! Otherwise use {@link CrestHandler#init()
     * CrestHandler.init()}
     */
    public void updateData() {
        if (LOGGER.isInfoEnabled()) {
            LOGGER.info("Updating data");
        }
        for (final Entry<DataType, ProcessorConfiguration<?>> entry : dataProcessors.entrySet()) {
            final ProcessorConfiguration<?> procConfig = entry.getValue();
            Long lastUpdate = lastUpdates.get(entry.getKey());
            if (lastUpdate == null) {
                lastUpdate = 0L;
            }
            final int refreshInterval = procConfig.getProcessor().getRefreshInterval();
            if (procConfig.isEnabled() && (lastUpdate < (System.currentTimeMillis() - refreshInterval))) {
                final long updateDate = updateData(procConfig);
                if (updateDate > 0) {
                    lastUpdates.put(entry.getKey(), updateDate);
                }
            }
        }
        if (LOGGER.isInfoEnabled()) {
            LOGGER.info("Finished updating data");
        }
    }

    private <T> long updateData(final ProcessorConfiguration<T> config) {
        final CrestApiProcessor<T> typeProcessor = config.getProcessor();
        if (LOGGER.isInfoEnabled()) {
            LOGGER.info("Updating: " + typeProcessor.getClass().getSimpleName().replace("Processor", ""));
        }
        long updateDate = -1;
        final CrestContainer<T> result = processor.downloadAndProcessContainerData(typeProcessor);

        if (result != null) {
            config.getConsumer().accept(result.getEntries());
            updateDate = result.getTimestamp();
        }
        return updateDate;
    }
}