D-Net

package eu.dnetlib.administration.uis.modules.services;

import java.io.IOException;

import java.net.HttpURLConnection;

import java.net.URL;

import java.util.ArrayList;

import java.util.Collection;

import java.util.List;

import java.util.Map;

import java.util.stream.Collectors;

import javax.servlet.http.HttpServletRequest;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

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

import org.springframework.ui.ModelMap;

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

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

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

import org.springframework.web.client.RestTemplate;

import com.google.common.collect.Maps;

import eu.dnetlib.administration.uis.annotations.Authorization;

import eu.dnetlib.administration.uis.annotations.MenuEntry;

import eu.dnetlib.administration.uis.annotations.MenuGroup;

import eu.dnetlib.administration.uis.annotations.PermissionLevel;

import eu.dnetlib.administration.uis.modules.UIModule;

import eu.dnetlib.clients.is.InformationServiceClient;

import eu.dnetlib.clients.locators.ServiceClientFactory;

import eu.dnetlib.clients.msro.MsroWorkerClient;

import eu.dnetlib.clients.msro.MsroWorkerRunningInstance;

import eu.dnetlib.miscutils.datetime.DateUtils;

@RestController

@RequestMapping("/ajax/services")

@MenuEntry(value = "Services Inspector", urlSection = "services", order = 3, group = MenuGroup.TOOLS)

@Authorization(PermissionLevel.SUPER_ADMIN)

public class ServicesInspectorModule extends UIModule {

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

	@Autowired

	private InformationServiceClient isClient;

	@Autowired

	private ServiceClientFactory serviceClientFactory;

	@Override

	public void populateModelMap(final ModelMap map, final HttpServletRequest request) {}

	@RequestMapping("")

	public Collection<ServiceGrouperDesc> listServices() throws Exception {

		final String xq = "for $x in collection('/db/DRIVER/dnetService') return concat ($x//RESOURCE_IDENTIFIER/@value, ' @ ', $x//RESOURCE_URI/@value)";

		final List<ServiceDesc> list = isClient.find(xq).stream()

				.map(e -> e.split("@"))

				.filter(arr -> arr.length == 2)

				.filter(arr -> arr[0].split("/").length == 3)

				.map(arr -> new ServiceDesc(arr[0].trim(), arr[0].split("/")[1], arr[1].trim()))

				.collect(Collectors.toList());

		final Map<String, ServiceGrouperDesc> map = Maps.newHashMap();

		for (final ServiceDesc s : list) {

			final URL url = new URL(s.getUrl());

			final String host = url.getHost();

			final int port = url.getPort();

			final String tmpKey = host + ":" + port;

			if (!map.containsKey(tmpKey)) {

				map.put(tmpKey, new ServiceGrouperDesc(host, port, new ArrayList<ServiceDesc>()));

			}

			map.get(tmpKey).getServices().add(s);

		}

		return map.values();

	}

	@RequestMapping("ping")

	public long pingUrl(@RequestParam("url") final String url,

			@RequestParam("timeout") final int timeout)

			throws IOException {

		final long start = DateUtils.now();

		final HttpURLConnection urlConn = (HttpURLConnection) new URL(url).openConnection();

		urlConn.setConnectTimeout(timeout);

		urlConn.setReadTimeout(timeout);

		if (urlConn.getResponseCode() == HttpURLConnection.HTTP_OK) {

			return DateUtils.now() - start;

		} else {

			throw new IllegalArgumentException("Invalid Url");

		}

	}

	@RequestMapping("workers")

	public List<MsroWorkerRunningInstance> listWorkers() throws Exception {

		final String xq =

				"for $x in collection('/db/DRIVER/dnetService/msroWorker') return $x//RESOURCE_URI/@value/string()";

		return isClient.find(xq)

				.stream()

				.map(url -> serviceClientFactory.getClient(MsroWorkerClient.class, url))

				.map(MsroWorkerClient::getStatus)

				.collect(Collectors.toList());

	}

	@RequestMapping("workers/pause")

	public MsroWorkerRunningInstance pause(@RequestParam final String baseUrl) {

		return (new RestTemplate()).getForObject(baseUrl + "/pause", MsroWorkerRunningInstance.class);

	}

	@RequestMapping("workers/resume")

	public MsroWorkerRunningInstance resume(@RequestParam final String baseUrl) throws Exception {

		return (new RestTemplate()).getForObject(baseUrl + "/resume", MsroWorkerRunningInstance.class);

	}

}