122 lines
4.6 KiB
Java
122 lines
4.6 KiB
Java
package eu.dnetlib.broker;
|
|
|
|
import java.io.IOException;
|
|
import java.io.InputStreamReader;
|
|
|
|
import javax.annotation.PostConstruct;
|
|
|
|
import org.apache.commons.logging.Log;
|
|
import org.apache.commons.logging.LogFactory;
|
|
import org.apache.maven.model.Model;
|
|
import org.apache.maven.model.io.xpp3.MavenXpp3Reader;
|
|
import org.codehaus.plexus.util.xml.pull.XmlPullParserException;
|
|
import org.elasticsearch.client.RestHighLevelClient;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.beans.factory.annotation.Value;
|
|
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 org.springframework.core.io.ClassPathResource;
|
|
import org.springframework.data.elasticsearch.client.ClientConfiguration;
|
|
import org.springframework.data.elasticsearch.client.RestClients;
|
|
import org.springframework.data.elasticsearch.config.AbstractElasticsearchConfiguration;
|
|
import org.springframework.data.elasticsearch.repository.config.EnableElasticsearchRepositories;
|
|
import org.springframework.scheduling.annotation.EnableScheduling;
|
|
import org.springframework.transaction.annotation.EnableTransactionManagement;
|
|
|
|
import com.google.common.collect.Lists;
|
|
|
|
import eu.dnetlib.broker.common.elasticsearch.Event;
|
|
import eu.dnetlib.broker.common.elasticsearch.Notification;
|
|
import eu.dnetlib.broker.common.properties.ElasticSearchProperties;
|
|
import io.micrometer.core.instrument.ImmutableTag;
|
|
import io.micrometer.core.instrument.Metrics;
|
|
import io.prometheus.client.exporter.MetricsServlet;
|
|
import io.prometheus.client.hotspot.DefaultExports;
|
|
import springfox.documentation.builders.ApiInfoBuilder;
|
|
import springfox.documentation.builders.RequestHandlerSelectors;
|
|
import springfox.documentation.service.ApiInfo;
|
|
import springfox.documentation.service.Tag;
|
|
import springfox.documentation.spi.DocumentationType;
|
|
import springfox.documentation.spring.web.plugins.Docket;
|
|
import springfox.documentation.swagger2.annotations.EnableSwagger2;
|
|
|
|
@Configuration
|
|
@EnableSwagger2
|
|
@EnableScheduling
|
|
@EnableTransactionManagement
|
|
@EnableElasticsearchRepositories(basePackageClasses = {
|
|
Event.class, Notification.class
|
|
})
|
|
public class BrokerConfiguration extends AbstractElasticsearchConfiguration {
|
|
|
|
private static final Log log = LogFactory.getLog(BrokerConfiguration.class);
|
|
|
|
@Autowired
|
|
private ElasticSearchProperties elasticSearchProperties;
|
|
|
|
@Value("pom.xml")
|
|
private ClassPathResource pom;
|
|
|
|
public static final String OA_PUBLIC_APIS = "Openaire Broker Public API";
|
|
|
|
@PostConstruct
|
|
public void init() {
|
|
final MavenXpp3Reader reader = new MavenXpp3Reader();
|
|
try {
|
|
final Model model = reader.read(new InputStreamReader(pom.getInputStream()));
|
|
|
|
log.info(String.format("registering metric for %s", model.getArtifactId()));
|
|
|
|
final ImmutableTag tag1 = new ImmutableTag("component", model.getGroupId() + ":" + model.getArtifactId());
|
|
// final ImmutableTag tag2 = new ImmutableTag("version", model.getVersion());
|
|
// final ImmutableTag tag3 = new ImmutableTag("scmtag", model.getScm().getTag());
|
|
|
|
Metrics.gauge("micrometer_info", Lists.newArrayList(tag1), 1);
|
|
} catch (IOException | XmlPullParserException e) {
|
|
log.error("Error registering metric", e);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
@Bean
|
|
public RestHighLevelClient elasticsearchClient() {
|
|
final ClientConfiguration clientConfiguration = ClientConfiguration.builder()
|
|
.connectedTo(elasticSearchProperties.getClusterNodes().split(","))
|
|
.withConnectTimeout(elasticSearchProperties.getConnectionTimeout())
|
|
.withSocketTimeout(elasticSearchProperties.getSocketTimeout())
|
|
.build();
|
|
return RestClients.create(clientConfiguration).rest();
|
|
}
|
|
|
|
@Bean
|
|
public static Docket newSwaggerDocket() {
|
|
log.info("Initializing SWAGGER...");
|
|
|
|
return new Docket(DocumentationType.SWAGGER_2)
|
|
.select()
|
|
.apis(RequestHandlerSelectors.any())
|
|
.paths(p -> p.startsWith("/"))
|
|
.build()
|
|
.tags(new Tag(OA_PUBLIC_APIS, OA_PUBLIC_APIS))
|
|
.apiInfo(new ApiInfoBuilder()
|
|
.title("OpenAIRE Public Broker API")
|
|
.description("APIs documentation")
|
|
.version("1.1")
|
|
.contact(ApiInfo.DEFAULT_CONTACT)
|
|
.license("Apache 2.0")
|
|
.licenseUrl("http://www.apache.org/licenses/LICENSE-2.0")
|
|
.build());
|
|
|
|
}
|
|
|
|
@Bean
|
|
@ConditionalOnMissingBean(name = "prometheusMetricsServletRegistrationBean")
|
|
ServletRegistrationBean<MetricsServlet> prometheusMetricsServletRegistrationBean(@Value("${prometheus.metrics.path:/metrics}") final String metricsPath) {
|
|
DefaultExports.initialize();
|
|
return new ServletRegistrationBean<>(new MetricsServlet(), metricsPath);
|
|
}
|
|
|
|
}
|