D-Net

package eu.dnetlib.enabling.app;

import java.util.Collection;

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

import org.springframework.boot.actuate.endpoint.PublicMetrics;

import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;

import org.springframework.boot.web.servlet.ServletRegistrationBean;

import org.springframework.context.annotation.Bean;

import org.springframework.context.annotation.Configuration;

import eu.dnetlib.services.ApiDocUtils;

import io.prometheus.client.exporter.MetricsServlet;

import io.prometheus.client.hotspot.DefaultExports;

import io.prometheus.client.spring.boot.SpringBootMetricsCollector;

import springfox.documentation.spring.web.plugins.Docket;

@Configuration

public class MainConfiguration {

	@Bean

	public Docket api() {

		return ApiDocUtils.newSwaggerDocket();

	}

	@Bean

	@ConditionalOnMissingBean(SpringBootMetricsCollector.class)

	SpringBootMetricsCollector springBootMetricsCollector(final Collection<PublicMetrics> publicMetrics) {

		final SpringBootMetricsCollector springBootMetricsCollector = new SpringBootMetricsCollector(publicMetrics);

		springBootMetricsCollector.register();

		return springBootMetricsCollector;

	}

	@Bean

	@ConditionalOnMissingBean(name = "prometheusMetricsServletRegistrationBean")

	ServletRegistrationBean prometheusMetricsServletRegistrationBean(@Value("${prometheus.metrics.path:/prometheus}") final String metricsPath) {

		DefaultExports.initialize();

		return new ServletRegistrationBean(new MetricsServlet(), metricsPath);

	}

}