D-Net

package eu.dnetlib.data.mapreduce.hbase.statsExport.drivers;

import eu.dnetlib.data.mapreduce.hbase.statsExport.daos.*;

import eu.dnetlib.data.mapreduce.hbase.statsExport.utils.HdfsWriter;

import org.apache.log4j.Logger;

import java.text.SimpleDateFormat;

import java.util.Date;

public class DBDriver {

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

    private StatsDAO statsDao;

    /**

     * Prepares shadow schema in the database that will be used to import data

     *

     * @throws Exception

     */

    public DBDriver(String dbUrl, String dbUser, String dbPassword, String dbDriver) {

        statsDao = new StatsDAO(dbUrl, dbUser, dbPassword, dbDriver);

    }

    public void prepareDB(String user) throws Exception {

        try {

            statsDao.setSearchPathDB(user);

            statsDao.createSchema();

        } catch (Exception e) {

            log.error("Error Preparing Schema for DB import. ", e);

            throw new Exception("Error Preparing Schema for DB import.", e);

        }

    }

    public void finalizedDB() throws Exception {

        try {

            statsDao.buildIndexes();

            statsDao.buildViews();

            statsDao.executeExtraInserts();

        } catch (Exception e) {

            log.error("Error while finalizing SQL DB", e);

            throw new Exception("Error while finalizing SQL DB", e);

        }

    }

    public void renameSchema() throws Exception {

        try {

            statsDao.renameSchema();

        } catch (Exception e) {

            log.error("Error while renaming database schema", e);

            throw new Exception("Error while renaming database schema", e);

        }

    }

    public StatsDAO getStatsDao() {

        return statsDao;

    }

    public void setStatsDao(StatsDAO statsDao) {

        this.statsDao = statsDao;

    }

}