pmproducer/src/main/java/org/oran/pmproducer/Application.java

   1 /*-
   2  * ========================LICENSE_START=================================
   3  * O-RAN-SC
   4  * %%
   5  * Copyright (C) 2023 Nordix Foundation
   6  * %%
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ========================LICENSE_END===================================
  19  */
  20
  21 package org.oran.pmproducer;
  22
  23 import java.io.File;
  24
  25 import org.slf4j.Logger;
  26 import org.slf4j.LoggerFactory;
  27 import org.springframework.beans.factory.annotation.Value;
  28 import org.springframework.boot.ApplicationArguments;
  29 import org.springframework.boot.SpringApplication;
  30 import org.springframework.boot.autoconfigure.SpringBootApplication;
  31 import org.springframework.boot.context.properties.EnableConfigurationProperties;
  32 import org.springframework.context.ConfigurableApplicationContext;
  33 import org.springframework.scheduling.annotation.EnableScheduling;
  34 import org.springframework.scheduling.annotation.Scheduled;
  35
  36 @SpringBootApplication
  37 @EnableConfigurationProperties
  38 @EnableScheduling
  39 public class Application {
  40
  41     private static final Logger logger = LoggerFactory.getLogger(Application.class);
  42
  43     @Value("${app.configuration-filepath}")
  44     private String localConfigurationFilePath;
  45
  46     private long configFileLastModification = 0;
  47     private static ConfigurableApplicationContext applicationContext;
  48
  49     public static void main(String[] args) {
  50         applicationContext = SpringApplication.run(Application.class);
  51         Runtime.getRuntime().addShutdownHook(new Thread() {
  52             @Override
  53             public void run() {
  54                 logger.warn("Shutting down, received signal SIGTERM");
  55                 SpringApplication.exit(applicationContext);
  56                 applicationContext = null;
  57             }
  58         });
  59     }
  60
  61     @Scheduled(fixedRate = 10 * 1000)
  62     public void checkConfigFileChanges() {
  63         long timestamp = new File(localConfigurationFilePath).lastModified();
  64         if (configFileLastModification != 0 && timestamp != configFileLastModification) {
  65             logger.info("Restarting due to change in the file {}", localConfigurationFilePath);
  66             restartApplication();
  67         }
  68         configFileLastModification = timestamp;
  69     }
  70
  71     private static void restartApplication() {
  72         if (applicationContext == null) {
  73             logger.info("Cannot restart in unittest");
  74             return;
  75         }
  76         ApplicationArguments args = applicationContext.getBean(ApplicationArguments.class);
  77
  78         Thread thread = new Thread(() -> {
  79             applicationContext.close();
  80             applicationContext = SpringApplication.run(Application.class, args.getSourceArgs());
  81         });
  82
  83         thread.setDaemon(false);
  84         thread.start();
  85     }
  86 }