D-Net

package eu.dnetlib.data.claims.migration.handler;

import eu.dnetlib.data.claims.migration.JsonldBuilder;

import eu.dnetlib.data.claims.migration.entity.Claim;

import eu.dnetlib.data.claims.migration.entity.Project;

import eu.dnetlib.data.claimsDemo.QueryGenerator;

import eu.dnetlib.data.claimsDemo.SqlDAO;

import org.apache.log4j.Logger;

import java.sql.Array;

import java.sql.ResultSet;

import java.text.DateFormat;

import java.text.SimpleDateFormat;

import java.util.ArrayList;

import java.util.Date;

import java.util.List;

import java.util.UUID;

/**

 * Created by konstantina on 10/4/2017.

 */

public class FetchProjectHandler {

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

    SqlDAO sqlDAO = null;

    QueryGenerator queryGenerator = null;

    public FetchProjectHandler(){

    }

    public Project fetchProjectById(String projectId) throws Exception {

        //System.out.println(queryGenerator.generateSelectProjectByIdQuery(projectId));

        ResultSet rs = sqlDAO.executePreparedQuery(queryGenerator.generateSelectProjectByIdQuery(projectId));

        Project project = fetchProjectByResultSet(rs);

        if(project == null) {

            log.info("No project with id : "+ projectId+"\n");

        }

        return project;

    }

    public List<String> fetchContactEmailsByProjectId(String projectId) throws Exception {

        ResultSet rs = sqlDAO.executePreparedQuery(queryGenerator.generateSelectContactEmailsByProjectIdQuery(projectId));

        if(rs.next()) {

//            Array array = ((String[])rs.getArray(1));

            Array array = rs.getArray(1);

            String[] str_array = (String[])array.getArray();

            List<String> mails = new ArrayList<String>();

            for (int i=0; i<str_array.length; i++) {

                mails.add(str_array[i]);

            }

            return mails;

        }

        return null;

    }

    public void updateTokenByProjectId(String projectId, UUID newToken) throws Exception {

        sqlDAO.executeUpdateQuery(queryGenerator.generateUpdateTokenByProjectId(projectId, newToken));

    }

    public void updateContactEmailsByProjectId(String projectId, List<String> contactEmails) throws Exception {

        sqlDAO.executeUpdateQuery(queryGenerator.generateUpdateContactEmailsByProjectIdByProjectId(projectId, contactEmails));

    }

    public Project fetchProjectByResultSet(ResultSet rs) throws Exception {

        Project project = null;

        while(rs.next()) {

            project = new Project();

            project.setOpenaireId(rs.getString(1));

            project.setName(rs.getString(2));

            project.setAcronym(rs.getString(3));

            project.setFunderId(rs.getString(4));

            project.setFunderName(rs.getString(5));

            project.setFunderShortName(rs.getString(6));

            //project.setContactEmails((List<String>)rs.getArray(7));

        }

        return project;

    }

    public SqlDAO getSqlDAO() {

        return sqlDAO;

    }

    public void setSqlDAO(SqlDAO sqlDAO) {

        this.sqlDAO = sqlDAO;

    }

    public QueryGenerator getQueryGenerator() {

        return queryGenerator;

    }

    public void setQueryGenerator(QueryGenerator queryGenerator) {

        this.queryGenerator = queryGenerator;

    }

}