D-Net

package eu.dnetlib.datasource.publisher;

import java.util.ArrayList;

import java.util.List;

import com.google.common.collect.Lists;

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.DatasourceInfo;

import io.swagger.annotations.*;

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;

    public ResponseEntity<List<String>> listIds() {

	    try {

		    final List<String> res = dsInfoRetriever.listIds();

		    return new ResponseEntity<>(res, HttpStatus.OK);

	    } catch (ApiException e) {

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

		    return new ResponseEntity<>(res, HttpStatus.INTERNAL_SERVER_ERROR);

	    }

    }

    public ResponseEntity<DatasourceInfo> 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<DatasourceInfo> rsp = rsp(new DatasourceInfo(), HttpStatus.OK);

	    final ClientResponse clientResponse = dsInfoRetriever.getInfo(id);

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

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

	    	return rsp(clientResponse.getDatasourceInfo(), 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 ResponseEntity<DatasourceInfo> rsp(final DatasourceInfo info, final HttpStatus status) {

		return new ResponseEntity<>(info, status);

	}

}