D-Net

package eu.dnetlib.services;

import java.io.File;

import java.lang.annotation.Annotation;

import java.lang.management.ManagementFactory;

import java.lang.management.OperatingSystemMXBean;

import java.util.HashMap;

import java.util.Map;

import javax.annotation.PostConstruct;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

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

import org.springframework.context.Lifecycle;

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

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

import eu.dnetlib.conf.DnetGenericApplicationProperties;

import eu.dnetlib.enabling.annotations.DnetService;

import eu.dnetlib.enabling.annotations.DnetServiceType;

import eu.dnetlib.exceptions.DnetGenericException;

public abstract class BaseService implements Lifecycle {

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

	private boolean started = false;

	private String profileId;

	private String baseUrl;

	private Map<String, String> serviceProperties = new HashMap<>();

	private Map<String, String> extraProtocols = new HashMap<>();

	private DnetServiceType serviceType;

	@Autowired

	private DnetGenericApplicationProperties containerConfiguration;

	@PostConstruct

	public void init() throws DnetGenericException {

		verifyRequiredAnnotations(DnetService.class);

		verifyRequiredAnnotations(RestController.class);

		verifyRequiredAnnotations(RequestMapping.class);

		serviceType = getClass().getAnnotation(DnetService.class).value();

		baseUrl = containerConfiguration.getUrl() + getClass().getAnnotation(RequestMapping.class).value()[0];

		final File dir = new File(containerConfiguration.getBaseDir());

		if (!dir.exists()) {

			dir.mkdirs();

		}

	}

	private void verifyRequiredAnnotations(final Class<? extends Annotation> a) throws DnetGenericException {

		if (!getClass().isAnnotationPresent(a)) {

			final String message = "A required annotation is missing (@" + a.getSimpleName() + ") in class " + getClass();

			log.error(message);

			throw new DnetGenericException(message);

		}

	}

	@Override

	public void start() {

		log.info("Starting service " + serviceType);

		if (started) {

			log.warn("Service " + this + "already started, check bean initializations!");

		}

		started = true;

	}

	@Override

	public boolean isRunning() {

		log.debug("called isRunning " + this);

		return false;

	}

	@Override

	public void stop() {

		log.info("Stopping service " + this);

	}

	@RequestMapping("identify")

	public ServiceRunningInstance identify() {

		final File f = new File(containerConfiguration.getBaseDir());

		final Runtime runtime = Runtime.getRuntime();

		final OperatingSystemMXBean osBean = ManagementFactory.getOperatingSystemMXBean();

		final ServiceRunningInstance res = new ServiceRunningInstance();

		res.setServiceId(profileId);

		res.setBaseUrl(baseUrl);

		res.setFreeDiskSpace(bytesToMegaBytes(f.getFreeSpace()));

		res.setTotalDiskSpace(bytesToMegaBytes(f.getTotalSpace()));

		res.setFreeMemory(bytesToMegaBytes(runtime.freeMemory()));

		res.setTotalMemory(bytesToMegaBytes(runtime.totalMemory()));

		res.setMaxMemory(bytesToMegaBytes(runtime.maxMemory()));

		res.setUsedMemory(bytesToMegaBytes(runtime.totalMemory() - runtime.freeMemory()));

		res.setNumberOfProcessors(osBean.getAvailableProcessors());

		res.setSysLoadAvg(osBean.getSystemLoadAverage());

		return res;

	}

	public String getProfileId() {

		return profileId;

	}

	public void setProfileId(final String profileId) {

		this.profileId = profileId;

	}

	public Map<String, String> getServiceProperties() {

		return serviceProperties;

	}

	public void setServiceProperties(final Map<String, String> serviceProperties) {

		this.serviceProperties = serviceProperties;

	}

	public Map<String, String> getExtraProtocols() {

		return extraProtocols;

	}

	public void setExtraProtocols(final Map<String, String> extraProtocols) {

		this.extraProtocols = extraProtocols;

	}

	public DnetServiceType getServiceType() {

		return serviceType;

	}

	public void setServiceType(final DnetServiceType serviceType) {

		this.serviceType = serviceType;

	}

	public String getBaseUrl() {

		return baseUrl;

	}

	public void setBaseUrl(final String baseUrl) {

		this.baseUrl = baseUrl;

	}

	private static final long bytesToMegaBytes(final long n) {

		return n / 1_048_576;

	}

}