D-Net

package eu.dnetlib.data.search.solr;

import eu.dnetlib.domain.EPR;

import eu.dnetlib.domain.SecureDriverResource;

import gr.uoa.di.driver.enabling.resultset.ResultSet;

import gr.uoa.di.driver.enabling.resultset.ResultSetFactory;

import org.apache.log4j.Logger;

import org.apache.solr.client.solrj.impl.CloudSolrServer;

import java.util.HashMap;

import java.util.Map;

/**

 * Created by antleb on 2/4/14.

 */

public class SolrResultSetFactory implements ResultSetFactory {

    private Logger logger = Logger.getLogger(getClass());

    private Map<String, CloudSolrServer> clients = new HashMap<String, CloudSolrServer>();

    //private static int Rsets = 0;

    @Override

    public ResultSet<String> createResultSet(EPR epr) {

        try {

            CloudSolrServer solrClient = null;

            synchronized (clients) {

                solrClient =  clients.get(epr.getAddress());

            }

            if (solrClient == null) {

                solrClient = new CloudSolrServer(epr.getAddress());

                clients.put(epr.getAddress(), solrClient);

            }

            logger.debug("Address " + epr.getAddress());

            //Rsets++;

            //System.out.println("Created Results sets " + Rsets);

            return new SolrResultSet(epr, solrClient);

        } catch (Exception e) {

            logger.error("Error creating solr client", e);

        }

        return null;

    }

    @Override

    public <D> ResultSet<D> createResultSet(EPR epr, Class<D> resourceClass) {

        return (ResultSet<D>) createResultSet(epr);

    }

    @Override

    public <D> ResultSet<D> createResultSet(ResultSet<?> rs, Class<D> resourceClass) {

        throw new UnsupportedOperationException();

    }

    @Override

    public <D extends SecureDriverResource> ResultSet<D> createSecurityAwareRS(EPR epr, Class<D> resourceClass) {

        throw new UnsupportedOperationException();

    }

}