75 lines
2.9 KiB
Java
75 lines
2.9 KiB
Java
package eu.openaire.pdf_aggregation_statistics;
|
|
|
|
import eu.openaire.pdf_aggregation_statistics.util.UriBuilder;
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
import org.springframework.boot.CommandLineRunner;
|
|
import org.springframework.boot.SpringApplication;
|
|
import org.springframework.boot.autoconfigure.SpringBootApplication;
|
|
import org.springframework.boot.web.servlet.context.ServletWebServerApplicationContext;
|
|
import org.springframework.context.ConfigurableApplicationContext;
|
|
import org.springframework.context.annotation.Bean;
|
|
import org.springframework.core.env.Environment;
|
|
import org.springframework.scheduling.annotation.EnableScheduling;
|
|
import org.springframework.web.cors.CorsConfiguration;
|
|
import org.springframework.web.cors.CorsConfigurationSource;
|
|
import org.springframework.web.cors.UrlBasedCorsConfigurationSource;
|
|
|
|
import javax.annotation.PreDestroy;
|
|
import java.util.Arrays;
|
|
import java.util.Collections;
|
|
|
|
@SpringBootApplication
|
|
@EnableScheduling
|
|
public class PdfAggregationStatisticsApplication {
|
|
|
|
private static final Logger logger = LoggerFactory.getLogger(PdfAggregationStatisticsApplication.class);
|
|
|
|
|
|
private static ConfigurableApplicationContext context;
|
|
|
|
public static void main(String[] args) {
|
|
context = SpringApplication.run(PdfAggregationStatisticsApplication.class, args);
|
|
}
|
|
|
|
|
|
@Bean
|
|
public CorsConfigurationSource corsConfigurationSource() {
|
|
CorsConfiguration configuration = new CorsConfiguration();
|
|
configuration.setAllowedOrigins(Collections.singletonList("*"));
|
|
configuration.setAllowedMethods(Collections.singletonList("GET"));
|
|
configuration.setAllowedHeaders(Arrays.asList("authorization", "content-type", "x-auth-token"));
|
|
configuration.setExposedHeaders(Collections.singletonList("x-auth-token"));
|
|
UrlBasedCorsConfigurationSource source = new UrlBasedCorsConfigurationSource();
|
|
source.registerCorsConfiguration("/**", configuration);
|
|
return source;
|
|
}
|
|
|
|
|
|
public static void gentleAppShutdown()
|
|
{
|
|
int exitCode = 0;
|
|
try {
|
|
exitCode = SpringApplication.exit(context, () -> 0); // The "PreDestroy" method will be called. (the "context" will be closed automatically (I checked it))
|
|
} catch (IllegalArgumentException iae) {
|
|
logger.error(iae.getMessage()); // This will say "Context must not be null", in case the "gentleAppShutdown()" was called too early in the app's lifetime. But it's ok.
|
|
}
|
|
System.exit(exitCode);
|
|
}
|
|
|
|
|
|
@PreDestroy
|
|
public void preDestroy() {
|
|
// TODO - Add any future shut-down logic here (e.g. shutting down threads).
|
|
logger.info("Exiting..");
|
|
}
|
|
|
|
|
|
@Bean
|
|
public CommandLineRunner setServerBaseUrl(Environment environment, ServletWebServerApplicationContext webServerAppCtxt)
|
|
{
|
|
return args -> new UriBuilder(environment, webServerAppCtxt);
|
|
}
|
|
|
|
}
|