D-Net

package eu.dnetlib.iis.core.workflows.usage_stats;

import eu.dnetlib.iis.core.java.PortBindings;

import eu.dnetlib.iis.core.java.Process;

import eu.dnetlib.iis.core.java.porttype.PortType;

import eu.dnetlib.usagestats.export.UsageStatsExporter;

import org.apache.hadoop.conf.Configuration;

import org.apache.log4j.Logger;

import java.util.HashMap;

import java.util.Map;

import java.util.Properties;

public class UsageStatsWrapper implements Process {

    private Logger log = Logger.getLogger(this.getClass());

    private UsageStatsExporter usageStatsExporter;

    @Override

    public void run(PortBindings portBindings, Configuration context, Map<String, String> parameters) throws Exception {

        Properties properties = new Properties();

        for (String key:parameters.keySet()) {

            properties.setProperty(key, parameters.get(key));

        }

        usageStatsExporter = new UsageStatsExporter(properties);

//        usageStatsExporter.setUsername(parameters.get("piwik_username"));

//        usageStatsExporter.setPassword(parameters.get("piwik_pass"));

//        usageStatsExporter.setHttpProtocol(parameters.get("piwik_httpProtocol"));

//        usageStatsExporter.setPiwikUrl(parameters.get("piwik_url"));

//        usageStatsExporter.setSiteId(parameters.get("piwik_siteId"));

//        usageStatsExporter.setStartDate(parameters.get("piwik_startDate"));

//        usageStatsExporter.setStartDate(parameters.get("piwik_startDate"));

//        usageStatsExporter.setFinalDate(parameters.get("piwik_finalDate"));

//        usageStatsExporter.setLogsPath(parameters.get("piwik_logsPath"));

//        usageStatsExporter.setFilterOffset(parameters.get("piwik_filterOffset"));

//        //usageStatsExporter.setDbSchema(parameters.get("piwik_schema"));

//        usageStatsExporter.setDbUrl(parameters.get("Stats_db_Url"));

//        usageStatsExporter.setDbUsername(parameters.get("Stats_db_User"));

//        usageStatsExporter.setDbPassword(parameters.get("Stats_db_Pass"));

//        usageStatsExporter.setTokenAuth(parameters.get("token_auth"));

//        log.info("Parameters: " + usageStatsExporter.getDbUrl() + usageStatsExporter.getUsername() + usageStatsExporter.getDbPassword());

//        log.info("Parameters: " + usageStatsExporter.getDbSchema() + usageStatsExporter.getLogsPath());

        for (Map.Entry<String, String> e : parameters.entrySet()) {

            log.info(e.getKey() + ":" + e.getValue());

        }

        try {

            usageStatsExporter.export();

            log.info("Usage Stats Exported!...");

        } catch (Exception e){

            log.error("Usage Stats Export failed: ", e);

            throw new Exception("Usage Stats Export failed: " + e.toString(), e);

        }

    }

    @Override

    public Map<String, PortType> getInputPorts() {

        return createInputPorts();

    }

    @Override

    public Map<String, PortType> getOutputPorts() {

        return createOutputPorts();

    }

    private static HashMap<String, PortType> createInputPorts() {

        HashMap<String, PortType> inputPorts = new HashMap<String, PortType>();

        return inputPorts;

    }

    private static HashMap<String, PortType> createOutputPorts() {

        HashMap<String, PortType> outputPorts = new HashMap<String, PortType>();

        return outputPorts;

    }

}