Refresh Context

Error Handling

Author: gauravat16

src/main/java/com/gaurav/restify/bootstrap/ConfigurationListener.java

@@ -5,6 +5,8 @@
 import com.gaurav.restify.constants.ConfigurationConstants;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.context.ApplicationContext;
 import org.springframework.context.event.ContextRefreshedEvent;
 import org.springframework.context.event.EventListener;
 import org.springframework.stereotype.Component;
@@ -18,6 +20,10 @@ public class ConfigurationListener {
 
     Logger logger = LoggerFactory.getLogger(ConfigurationUtil.class);
 
+    @Autowired
+    private static ApplicationContext appContext;
+
+
 
     @EventListener(ContextRefreshedEvent.class)
     private void instantiateConfiguration() {
@@ -34,7 +40,7 @@ private void createDirectories() {
             Files.createDirectories(ConfigurationConstants.REST_CONFIG_PATH);
         } catch (IOException e) {
             logger.error("Could not create directories", e);
-            System.exit(0);
+            System.exit(2);
 
         }
     }

src/main/java/com/gaurav/restify/bootstrap/ShutDownListener.java

@@ -14,7 +14,7 @@ public class ShutDownListener {
 
     @EventListener(ContextClosedEvent.class)
     private void applicationStopped() {
-        logger.warn("Application forced to close!");
+        logger.debug("Application forced to close!");
     }
 
 }

src/main/java/com/gaurav/restify/configuration/RestConfigurationManager.java

@@ -12,6 +12,7 @@
 import java.util.HashMap;
 
 import static com.gaurav.restify.constants.ConfigurationConstants.REST_CONFIG_FILE;
+import static com.gaurav.restify.constants.ErrorCodes.ERROR_TERMINATE;
 
 @Component
 public class RestConfigurationManager {
@@ -22,6 +23,7 @@ public class RestConfigurationManager {
     private static HashMap<String, RestJob> restJobHashMap = null;
     private static final Logger logger = LoggerFactory.getLogger(RestConfigurationManager.class);
 
+
     private RestConfigurationManager() {
 
     }
@@ -38,13 +40,13 @@ public static RestConfigurationManager getInstance() {
 
             } catch (JAXBException e) {
 
-                logger.error( "Could not parse xml file - " + REST_CONFIG_FILE, e);
-                System.exit(0);
+                logger.error("Could not parse xml file - " + REST_CONFIG_FILE, e);
+                System.exit(ERROR_TERMINATE);
+
 
             } catch (FileNotFoundException e) {
                 logger.error("xml file not found- " + REST_CONFIG_FILE, e);
-                System.exit(0);
-
+                System.exit(ERROR_TERMINATE);
 
             }
 
@@ -56,7 +58,7 @@ public static RestConfigurationManager getInstance() {
 
 
     private static void instantiateRestJobMap() {
-        logger.info( "Going to instantiate rest job map");
+        logger.info("Going to instantiate rest job map");
         if (null != restConfiguration) {
             restJobHashMap = new HashMap<>();
             restConfiguration.getJobs().forEach(job -> restJobHashMap.put(job.getCommand(), job));
@@ -71,4 +73,15 @@ public RestJob getRestJob(String commandName) {
     }
 
 
+    public void refreshConfiguration(){
+        logger.info("Going to refresh rest job map");
+
+        restConfigurationManager = null;
+
+        RestConfigurationManager.getInstance();
+
+
+    }
+
+
 }

src/main/java/com/gaurav/restify/configuration/configurationBeans/RestJob.java

@@ -2,6 +2,7 @@
 
 import javax.xml.bind.annotation.XmlRootElement;
 import javax.xml.bind.annotation.XmlType;
+import java.util.Arrays;
 
 @XmlRootElement(name = "RestJob")
 
@@ -53,4 +54,15 @@ public long getWaitTime() {
     public void setWaitTime(long waitTime) {
         this.waitTime = waitTime;
     }
+
+    @Override
+    public String toString() {
+        return "RestJob{" +
+                "path='" + path + '\'' +
+                ", commandType='" + commandType + '\'' +
+                ", command='" + command + '\'' +
+                ", args=" + Arrays.toString(args) +
+                ", waitTime=" + waitTime +
+                '}';
+    }
 }

src/main/java/com/gaurav/restify/constants/ErrorCodes.java

@@ -0,0 +1,6 @@
+package com.gaurav.restify.constants;
+
+public interface ErrorCodes {
+
+    int ERROR_TERMINATE = -1;
+}

src/main/java/com/gaurav/restify/controllers/ProcessController.java

@@ -5,17 +5,16 @@
 import com.gaurav.restify.beans.Response;
 import com.gaurav.restify.configuration.RestConfigurationManager;
 import com.gaurav.restify.configuration.configurationBeans.RestJob;
+import com.gaurav.restify.constants.ErrorCodes;
 import com.gaurav.restify.constants.ExecutorConstants;
 import com.gaurav.restify.services.ExecutorService;
+import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.PathVariable;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;
 
-import java.io.BufferedReader;
 import java.io.IOException;
-import java.io.InputStreamReader;
-import java.util.logging.Logger;
 
 @RestController
 public class ProcessController {
@@ -26,14 +25,21 @@ public class ProcessController {
     @Autowired
     private RestConfigurationManager restConfigurationManager;
 
+    private static final org.slf4j.Logger logger = LoggerFactory.getLogger(ProcessController.class);
+
 
     @RequestMapping(value = "/execute/{scriptName}")
     public Response executeScripts(@PathVariable String scriptName) {
 
+        logger.info("Current Rest Call for Script ::: " + scriptName);
+
         RestJob restJob = restConfigurationManager.getRestJob(scriptName);
 
+        logger.info("Rest Job Found " + restJob);
 
-        ExecutorTaskOutput taskOutput = null;
+
+        ExecutorTaskOutput taskOutput;
+        Response response;
         try {
 
 
@@ -44,16 +50,24 @@ public Response executeScripts(@PathVariable String scriptName) {
 
 
             taskOutput = executorService.executeTask(executorTask);
+            response = new Response(String.valueOf(taskOutput.getExitCode()));
+            response.setOutput(taskOutput.getOutput());
 
 
         } catch (IOException ioEx) {
 
-            ioEx.printStackTrace();
-        }
+            logger.error("File Not Found", ioEx);
+            response = new Response(String.valueOf(ErrorCodes.ERROR_TERMINATE));
+            response.setOutput(ioEx.toString());
 
+        } catch (Exception e) {
+
+            logger.error("Script execution failed! ", e);
+            response = new Response(String.valueOf(ErrorCodes.ERROR_TERMINATE));
+            response.setOutput(e.toString());
+
+        }
 
-        Response response = new Response(String.valueOf(taskOutput.getExitCode()));
-        response.setOutput(taskOutput.getOutput());
 
         return response;
     }

src/main/java/com/gaurav/restify/controllers/RefreshConfiguration.java

@@ -0,0 +1,26 @@
+package com.gaurav.restify.controllers;
+
+import com.gaurav.restify.beans.Response;
+import com.gaurav.restify.configuration.RestConfigurationManager;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+@RestController
+public class RefreshConfiguration {
+
+    @Autowired
+    RestConfigurationManager restConfigurationManager;
+
+    @RequestMapping(value = "/refresh")
+    public Response refreshConfiguration(){
+
+        restConfigurationManager.refreshConfiguration();
+
+        Response response = new Response(String.valueOf(0));
+        response.setOutput("Configuration Refreshed");
+
+        return response;
+
+    }
+}