D-Net

package eu.dnetlib.services;

import java.io.IOException;

import java.lang.management.ManagementFactory;

import java.lang.management.RuntimeMXBean;

import java.util.ArrayList;

import java.util.Iterator;

import java.util.List;

import java.util.Map;

import java.util.TreeMap;

import java.util.stream.Collectors;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.apache.maven.model.Model;

import org.apache.maven.model.Parent;

import org.apache.maven.model.io.xpp3.MavenXpp3Reader;

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

import org.springframework.context.ResourceLoaderAware;

import org.springframework.core.env.AbstractEnvironment;

import org.springframework.core.env.Environment;

import org.springframework.core.env.MapPropertySource;

import org.springframework.core.env.PropertySource;

import org.springframework.core.io.Resource;

import org.springframework.core.io.ResourceLoader;

import org.springframework.core.io.support.ResourcePatternUtils;

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

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

import com.google.common.collect.Maps;

import eu.dnetlib.conf.DnetGenericApplicationProperties;

import eu.dnetlib.enabling.annotations.DnetService;

import eu.dnetlib.enabling.annotations.DnetServiceType;

import eu.dnetlib.miscutils.datetime.DateUtils;

import eu.dnetlib.miscutils.datetime.HumanTime;

import eu.dnetlib.miscutils.streams.DnetStreamSupport;

@RestController

@RequestMapping("/hcm")

@DnetService(DnetServiceType.hcm)

public class HCMController extends BaseService implements ResourceLoaderAware {

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

	@Autowired

	private Environment env;

	@Autowired

	private DnetGenericApplicationProperties containerConfiguration;

	private ResourceLoader resourceLoader;

	@RequestMapping("properties")

	public Map<String, Object> listProperties() {

		final Iterator<PropertySource<?>> iter = ((AbstractEnvironment) env).getPropertySources().iterator();

		return DnetStreamSupport.generateStreamFromIterator(iter)

				.filter(ps -> ps instanceof MapPropertySource)

				.map(ps -> (MapPropertySource) ps)

				.collect(Collectors.toMap(MapPropertySource::getName, MapPropertySource::getSource));

	}

	@RequestMapping("info")

	public Map<String, Object> info() {

		final RuntimeMXBean mxbean = ManagementFactory.getRuntimeMXBean();

		final Map<String, Object> genInfo = Maps.newLinkedHashMap();

		genInfo.put("Hostname", containerConfiguration.getHost());

		genInfo.put("Port", containerConfiguration.getPort());

		genInfo.put("Uptime", HumanTime.exactly(mxbean.getUptime()));

		genInfo.put("Start Time", DateUtils.calculate_ISO8601(mxbean.getStartTime()));

		return genInfo;

	}

	@RequestMapping("jvm")

	public Map<String, String> jvmInfo() {

		final RuntimeMXBean mxbean = ManagementFactory.getRuntimeMXBean();

		final Map<String, String> jvmInfo = Maps.newLinkedHashMap();

		jvmInfo.put("JVM Name", mxbean.getVmName());

		jvmInfo.put("JVM Vendor", mxbean.getVmVendor());

		jvmInfo.put("JVM Version", mxbean.getVmVersion());

		jvmInfo.put("JVM Spec Name", mxbean.getSpecName());

		jvmInfo.put("JVM Spec Vendor", mxbean.getSpecVendor());

		jvmInfo.put("JVM Spec Version", mxbean.getSpecVersion());

		jvmInfo.put("Running JVM Name", mxbean.getName());

		jvmInfo.put("Management Spec Version", mxbean.getManagementSpecVersion());

		return jvmInfo;

	}

	@RequestMapping("libs")

	public Map<String, String> libInfo() {

		final RuntimeMXBean mxbean = ManagementFactory.getRuntimeMXBean();

		final Map<String, String> libInfo = Maps.newLinkedHashMap();

		libInfo.put("Classpath", mxbean.getClassPath().replaceAll(":", " : "));

		libInfo.put("Boot ClassPath", mxbean.getBootClassPath().replaceAll(":", " : "));

		libInfo.put("Input arguments", mxbean.getInputArguments().toString());

		libInfo.put("Library Path", mxbean.getLibraryPath().replaceAll(":", " : "));

		return libInfo;

	}

	@RequestMapping("maven")

	private Map<String, Map<String, Map<String, List<String>>>> mavenModules() throws IOException {

		final Map<String, Map<String, Map<String, List<String>>>> modules = new TreeMap<>();

		final MavenXpp3Reader reader = new MavenXpp3Reader();

		for (final Resource res : ResourcePatternUtils.getResourcePatternResolver(resourceLoader).getResources("classpath*:/META-INF/**/pom.xml")) {

			try {

				final Model model = reader.read(res.getInputStream());

				final String name = model.getArtifactId();

				String groupId = model.getGroupId();

				for (Parent parent = model.getParent(); (groupId == null) && (model.getParent() != null); parent = model.getParent()) {

					groupId = parent.getGroupId();

				}

				String version = model.getVersion();

				for (Parent parent = model.getParent(); (version == null) && (model.getParent() != null); parent = model.getParent()) {

					version = parent.getVersion();

				}

				if (!modules.containsKey(groupId)) {

					modules.put(groupId, new TreeMap<>());

				}

				if (!modules.get(groupId).containsKey(name)) {

					modules.get(groupId).put(name, new TreeMap<>());

				}

				if (!modules.get(groupId).get(name).containsKey(version)) {

					modules.get(groupId).get(name).put(version, new ArrayList<>());

				}

				modules.get(groupId).get(name).get(version).add(res.getURI().toString());

			} catch (final Exception e) {

				log.warn("Error evaluating pom: " + res.getURI());

				log.debug("-- ERROR --", e);

			}

		}

		return modules;

	}

	@Override

	public void setResourceLoader(final ResourceLoader resourceLoader) {

		this.resourceLoader = resourceLoader;

	}

}