2021-03-16 14:25:15 +01:00
|
|
|
package eu.openaire.urls_controller;
|
|
|
|
|
2023-05-11 02:07:55 +02:00
|
|
|
import eu.openaire.urls_controller.services.FullTextsServiceImpl;
|
2023-02-24 22:49:04 +01:00
|
|
|
import eu.openaire.urls_controller.services.UrlsServiceImpl;
|
2022-12-15 17:34:28 +01:00
|
|
|
import eu.openaire.urls_controller.util.FileUtils;
|
2022-02-22 12:54:16 +01:00
|
|
|
import eu.openaire.urls_controller.util.UriBuilder;
|
2022-02-04 13:48:22 +01:00
|
|
|
import org.slf4j.Logger;
|
|
|
|
import org.slf4j.LoggerFactory;
|
2022-02-22 12:54:16 +01:00
|
|
|
import org.springframework.boot.CommandLineRunner;
|
2021-03-16 14:25:15 +01:00
|
|
|
import org.springframework.boot.SpringApplication;
|
|
|
|
import org.springframework.boot.autoconfigure.SpringBootApplication;
|
2022-09-12 16:04:05 +02:00
|
|
|
import org.springframework.boot.web.servlet.context.ServletWebServerApplicationContext;
|
2021-03-16 14:25:15 +01:00
|
|
|
import org.springframework.context.annotation.Bean;
|
|
|
|
import org.springframework.core.env.Environment;
|
2021-12-21 14:55:27 +01:00
|
|
|
import org.springframework.scheduling.annotation.EnableScheduling;
|
2021-03-16 14:25:15 +01:00
|
|
|
import org.springframework.web.cors.CorsConfiguration;
|
|
|
|
import org.springframework.web.cors.CorsConfigurationSource;
|
|
|
|
import org.springframework.web.cors.UrlBasedCorsConfigurationSource;
|
|
|
|
|
2022-02-04 13:48:22 +01:00
|
|
|
import javax.annotation.PreDestroy;
|
2021-03-16 14:25:15 +01:00
|
|
|
import java.util.Arrays;
|
|
|
|
import java.util.Collections;
|
2022-12-15 17:34:28 +01:00
|
|
|
import java.util.concurrent.ExecutorService;
|
2022-02-04 13:48:22 +01:00
|
|
|
import java.util.concurrent.TimeUnit;
|
2021-03-16 14:25:15 +01:00
|
|
|
|
|
|
|
@SpringBootApplication
|
|
|
|
@EnableScheduling
|
|
|
|
public class Application {
|
|
|
|
|
2022-02-04 13:48:22 +01:00
|
|
|
private static final Logger logger = LoggerFactory.getLogger(Application.class);
|
|
|
|
|
2021-03-16 14:25:15 +01:00
|
|
|
public static void main(String[] args) {
|
|
|
|
SpringApplication.run(Application.class, args);
|
|
|
|
}
|
|
|
|
|
|
|
|
@Bean
|
|
|
|
public CorsConfigurationSource corsConfigurationSource() {
|
|
|
|
CorsConfiguration configuration = new CorsConfiguration();
|
|
|
|
configuration.setAllowedOrigins(Collections.singletonList("*"));
|
|
|
|
configuration.setAllowedMethods(Arrays.asList("GET", "POST", "PUT", "PATCH", "DELETE", "OPTIONS"));
|
|
|
|
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;
|
|
|
|
}
|
2022-02-04 13:48:22 +01:00
|
|
|
|
|
|
|
|
|
|
|
@PreDestroy
|
2022-12-15 17:34:28 +01:00
|
|
|
public void preDestroy() {
|
2022-02-04 13:48:22 +01:00
|
|
|
logger.info("Shutting down the threads..");
|
2022-12-15 17:34:28 +01:00
|
|
|
|
2023-02-24 22:49:04 +01:00
|
|
|
shutdownThreads(UrlsServiceImpl.insertsExecutor);
|
2022-12-15 17:34:28 +01:00
|
|
|
shutdownThreads(FileUtils.hashMatchingExecutor);
|
2023-05-11 02:07:55 +02:00
|
|
|
shutdownThreads(FullTextsServiceImpl.backgroundExecutor);
|
|
|
|
shutdownThreads(FullTextsServiceImpl.bulkImportExecutor);
|
2022-12-15 17:34:28 +01:00
|
|
|
|
|
|
|
logger.info("Exiting..");
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
private void shutdownThreads(ExecutorService executorService)
|
|
|
|
{
|
|
|
|
executorService.shutdown(); // Define that no new tasks will be scheduled.
|
2022-02-04 13:48:22 +01:00
|
|
|
try {
|
2022-12-15 17:34:28 +01:00
|
|
|
if ( ! executorService.awaitTermination(1, TimeUnit.MINUTES) ) {
|
2022-02-04 13:48:22 +01:00
|
|
|
logger.warn("The working threads did not finish on time! Stopping them immediately..");
|
2022-12-15 17:34:28 +01:00
|
|
|
executorService.shutdownNow();
|
2022-02-04 13:48:22 +01:00
|
|
|
}
|
|
|
|
} catch (SecurityException se) {
|
|
|
|
logger.error("Could not shutdown the threads in any way..!", se);
|
|
|
|
} catch (InterruptedException ie) {
|
|
|
|
try {
|
2022-12-15 17:34:28 +01:00
|
|
|
executorService.shutdownNow();
|
2022-02-04 13:48:22 +01:00
|
|
|
} catch (SecurityException se) {
|
|
|
|
logger.error("Could not shutdown the threads in any way..!", se);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2022-02-22 12:54:16 +01:00
|
|
|
@Bean
|
2022-09-12 16:04:05 +02:00
|
|
|
public CommandLineRunner setServerBaseUrl(Environment environment, ServletWebServerApplicationContext webServerAppCtxt)
|
|
|
|
{
|
|
|
|
return args -> new UriBuilder(environment, webServerAppCtxt);
|
2022-02-22 12:54:16 +01:00
|
|
|
}
|
2021-03-16 14:25:15 +01:00
|
|
|
|
|
|
|
}
|