D-Net

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

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

import eu.dnetlib.data.claimsDemo.QueryGenerator;

import eu.dnetlib.data.claimsDemo.SQLStoreException;

import eu.dnetlib.data.claimsDemo.SqlDAO;

import org.apache.log4j.Logger;

import java.sql.ResultSet;

import java.util.ArrayList;

public class FetchContextHandler {

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

    SqlDAO sqlDAO = null;

    QueryGenerator queryGenerator = null;

    public FetchContextHandler(){

    }

    public Context fetchFirstContextByCommunityId(String communityId) throws Exception, SQLStoreException {

        ArrayList<Object> params = new ArrayList<>();

        String query = queryGenerator.generateSelectFirstContextByCommunityIdQuery(communityId, params);

        ResultSet rs = sqlDAO.executePreparedQuery(query, params);

        Context context = fetchContextByResultSet(rs);

        if(context == null) {

            log.info("No context with community id : "+ communityId+"\n");

        }

        return context;

    }

    public Context fetchContextByResultSet(ResultSet rs) throws Exception {

        Context context = null;

        while(rs.next()) {

            context = new Context();

            context.setOpenaireId(rs.getString("openaire_id"));

            context.setTitle(rs.getString("context_title"));

        }

        return context;

    }

    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;

    }

}