D-Net

package eu.dnetlib.datasource.publisher;

import eu.dnetlib.common.rmi.DNetRestDocumentation;

import eu.dnetlib.datasource.publisher.clients.ClientResponse;

import eu.dnetlib.datasource.publisher.clients.DatasourceInfoRetriever;

import eu.dnetlib.datasource.publisher.model.DatasourceResponse;

import eu.dnetlib.datasource.publisher.model.IdentifiersResponse;

import eu.dnetlib.datasource.publisher.model.Response;

import io.swagger.annotations.ApiParam;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.springframework.beans.BeanUtils;

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

import org.springframework.http.HttpStatus;

import org.springframework.http.ResponseEntity;

import org.springframework.stereotype.Controller;

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

@Controller

@DNetRestDocumentation

public class DatasourcesApiController implements DatasourcesApi {

	private static final Log log = LogFactory.getLog(DatasourcesApiController.class);

	@Autowired

	private DatasourceInfoRetriever dsInfoRetriever;

	@Override

    public ResponseEntity<IdentifiersResponse> listIds() {

	    try {

		    return rsp(dsInfoRetriever.listIds(), HttpStatus.OK);

	    } catch (ApiException e) {

		    return rsp(new IdentifiersResponse(), HttpStatus.INTERNAL_SERVER_ERROR);

	    }

    }

    @Override

    public ResponseEntity<DatasourceResponse> getDs(@ApiParam(value = "ID of datasource to fetch", required=true ) @PathVariable("id") String id) {

	    if (log.isDebugEnabled()) {

		    log.debug(String.format("getDatasourceInfo(dsId = %s)", id));

	    }

	    final long start = System.nanoTime();

	    final ResponseEntity<DatasourceResponse> rsp = rsp(new DatasourceResponse(), HttpStatus.OK);

	    final ClientResponse clientResponse = dsInfoRetriever.getInfo(id);

	    BeanUtils.copyProperties(clientResponse.getDatasourceResponse(), rsp.getBody());

	    if (!clientResponse.getErrors().isEmpty()) {

	    	return rsp(clientResponse.getDatasourceResponse(), HttpStatus.INTERNAL_SERVER_ERROR);

	    }

	    final long time = (System.nanoTime() - start) / 1000000;

	    rsp.getBody().getResponseHeader().setDatasourceId(id).setQueryTime(time);

	    log.debug(String.format("%s ms", time));

	    return rsp;

    }

	private <T extends Response> ResponseEntity<T> rsp(final T info, final HttpStatus status) {

		info.getResponseHeader().setStatusCode(status.value());

		return new ResponseEntity<>(info, status);

	}

}