D-Net

package eu.dnetlib.enabling.resultSetManager;

import java.util.List;

import org.springframework.beans.factory.annotation.Required;

import org.springframework.web.bind.annotation.PathVariable;

import com.google.common.collect.Lists;

import com.google.gson.Gson;

import eu.dnetlib.rmi.objects.resultSet.ResultSetResponse;

import eu.dnetlib.rmi.soap.exceptions.ResultSetException;

public class ResultSetServiceCore {

	private ResultSetRegistry registry;

	public ResultSetResponse getNextResults(@PathVariable final String rsId, @PathVariable final int n) throws ResultSetException {

		final ResultSetIterator<?> iter = registry.findIterator(rsId);

		final int from = iter.position();

		final Gson gson = new Gson();

		final List<String> elements = Lists.newArrayList();

		for (Object o : iter.getNextElements(n)) {

			elements.add(gson.toJson(o));

		}

		final int to = iter.position();

		final boolean end = iter.isFinished();

		final int total = iter.getTotal();

		return new ResultSetResponse(from, to, end, total, elements);

	}

	public ResultSetRegistry getRegistry() {

		return registry;

	}

	@Required

	public void setRegistry(final ResultSetRegistry registry) {

		this.registry = registry;

	}

}