Ensure shutdown hook is called on docker stop

Change-Id: I7f2165c6c4b8cfeb3d62c4a286b62f4e42dde449
Signed-off-by: Blimkie, Steven (sb787e) <steven.blimkie@amdocs.com>

2 files changed, 161 insertions, 155 deletions

diff --git a/src/main/java/org/openecomp/modelloader/entity/model/ModelArtifactHandler.java b/src/main/java/org/openecomp/modelloader/entity/model/ModelArtifactHandler.java
index 6f9f64f..19b69cf 100644
--- a/src/main/java/org/openecomp/modelloader/entity/model/ModelArtifactHandler.java
+++ b/src/main/java/org/openecomp/modelloader/entity/model/ModelArtifactHandler.java
@@ -166,11 +166,14 @@ public class ModelArtifactHandler extends ArtifactHandler {
   private String getURL(ModelArtifact model) {
     String baseURL = config.getAaiBaseUrl().trim();
     String subURL = null;
+    String instance = null;
     if (model.getType().equals(ArtifactType.MODEL)) {
       subURL = config.getAaiModelUrl(model.getModelNamespaceVersion()).trim();
+      instance = model.getModelInvariantId();
     }
     else {
       subURL = config.getAaiNamedQueryUrl(model.getModelNamespaceVersion()).trim();
+      instance = model.getNameVersionId();
     }
 
     if ( (!baseURL.endsWith("/")) && (!subURL.startsWith("/")) ) {
@@ -185,18 +188,21 @@ public class ModelArtifactHandler extends ArtifactHandler {
       subURL = subURL + "/";
     }
 
-    String url = baseURL + subURL + model.getModelInvariantId();
+    String url = baseURL + subURL + instance;
     return url;
   }
 
   private String getModelVerURL(ModelArtifact model) {
     String baseURL = config.getAaiBaseUrl().trim();
     String subURL = null;
+    String instance = null;
     if (model.getType().equals(ArtifactType.MODEL)) {
       subURL = config.getAaiModelUrl(model.getModelNamespaceVersion()).trim() + model.getModelInvariantId() + AAI_MODEL_VER;
+      instance = model.getModelVerId();
     }
     else {
       subURL = config.getAaiNamedQueryUrl(model.getModelNamespaceVersion()).trim();
+      instance = model.getNameVersionId();
     }
 
     if ( (!baseURL.endsWith("/")) && (!subURL.startsWith("/")) ) {
@@ -211,8 +217,7 @@ public class ModelArtifactHandler extends ArtifactHandler {
       subURL = subURL + "/";
     }
 
-    String url = baseURL + subURL + model.getModelVerId();
-    System.out.println(url);
+    String url = baseURL + subURL + instance;
     return url;
   }
 
diff --git a/src/main/java/org/openecomp/modelloader/service/ModelLoaderService.java b/src/main/java/org/openecomp/modelloader/service/ModelLoaderService.java
index aeaa6f4..786f1a2 100644
--- a/src/main/java/org/openecomp/modelloader/service/ModelLoaderService.java
+++ b/src/main/java/org/openecomp/modelloader/service/ModelLoaderService.java
@@ -48,156 +48,157 @@ import javax.ws.rs.core.Response;
  * capabilities between AAI and an ASDC.
  */
 public class ModelLoaderService implements ModelLoaderInterface {
-
-  protected static final String FILESEP = (System.getProperty("file.separator") == null) ? "/"
-      : System.getProperty("file.separator");
-
-  protected static final String CONFIG_DIR = System.getProperty("CONFIG_HOME") + FILESEP;
-  protected static final String CONFIG_AUTH_LOCATION = CONFIG_DIR + "auth" + FILESEP;
-  protected static final String CONFIG_FILE = CONFIG_DIR + "model-loader.properties";
-
-  private IDistributionClient client;
-  private ModelLoaderConfig config;
-  private Timer timer = null;
-
-  static Logger logger = LoggerFactory.getInstance().getLogger(ModelLoaderService.class.getName());
-
-  /**
-   * Responsible for loading configuration files and calling initialization.
-   */
-  public ModelLoaderService() {
-    start();
-  }
-
-  protected void start() {
-    // Load model loader system configuration
-    logger.info(ModelLoaderMsgs.LOADING_CONFIGURATION);
-    Properties configProperties = new Properties();
-    try {
-      configProperties.load(new FileInputStream(CONFIG_FILE));
-    } catch (IOException e) {
-      String errorMsg = "Failed to load configuration: " + e.getMessage();
-      logger.error(ModelLoaderMsgs.ASDC_CONNECTION_ERROR, errorMsg);
-      shutdown();
-    }
-
-    config = new ModelLoaderConfig(configProperties, CONFIG_AUTH_LOCATION);
-    init();
-  }
-
-  @Override
-  public void finalize() {
-    preShutdownOperations();
-  }
-
-  /**
-   * Responsible for stopping the connection to the distribution client before
-   * the resource is destroyed.
-   */
-  protected void preShutdownOperations() {
-    logger.info(ModelLoaderMsgs.STOPPING_CLIENT);
-    if (client != null) {
-      client.stop();
-    }
-  }
-
-  /**
-   * Responsible for loading configuration files, initializing model
-   * distribution clients, and starting them.
-   */
-  protected void init() {
-    // Initialize distribution client
-    logger.debug(ModelLoaderMsgs.INITIALIZING, "Initializing distribution client...");
-    client = DistributionClientFactory.createDistributionClient();
-    EventCallback callback = new EventCallback(client, config);
-
-    IDistributionClientResult initResult = client.init(config, callback);
-
-    if (initResult.getDistributionActionResult() != DistributionActionResultEnum.SUCCESS) {
-      String errorMsg = "Failed to initialize distribution client: "
-          + initResult.getDistributionMessageResult();
-      logger.error(ModelLoaderMsgs.ASDC_CONNECTION_ERROR, errorMsg);
-
-      // Kick off a timer to retry the SDC connection
-      timer = new Timer();
-      TimerTask task = new SdcConnectionJob(client, config, callback, timer);
-      timer.schedule(task, new Date(), 60000);
-    }
-    else {
-      // Start distribution client
-      logger.debug(ModelLoaderMsgs.INITIALIZING, "Starting distribution client...");
-      IDistributionClientResult startResult = client.start();
-      if (startResult.getDistributionActionResult() != DistributionActionResultEnum.SUCCESS) {
-        String errorMsg = "Failed to start distribution client: "
-            + startResult.getDistributionMessageResult();
-        logger.error(ModelLoaderMsgs.ASDC_CONNECTION_ERROR, errorMsg);
-
-        // Kick off a timer to retry the SDC connection
-        timer = new Timer();
-        TimerTask task = new SdcConnectionJob(client, config, callback, timer);
-        timer.schedule(task, new Date(), 60000);
-      }
-      else {
-        logger.info(ModelLoaderMsgs.INITIALIZING, "Connection to SDC established");
-      }
-    }
-  }
-
-  /**
-   * Shut down the process.
-   */
-  private void shutdown() {
-    preShutdownOperations();
-
-    // TODO: Find a better way to shut down the model loader.
-    try {
-      // Give logs time to write to file
-      Thread.sleep(2000);
-    } catch (InterruptedException e) {
-      // Nothing we can do at this point
-    }
-
-    Runtime.getRuntime().halt(1);
-  }
-
-  /** (non-Javadoc)
-   * @see org.openecomp.modelloader.service.ModelLoaderInterface#loadModel(java.lang.String)
-   */
-  @Override
-  public Response loadModel(String modelid) {
-    Response response = Response.ok("{\"model_loaded\":\"" + modelid + "\"}").build();
-
-    return response;
-  }
-
-  /** (non-Javadoc)
-   * @see org.openecomp.modelloader.service.ModelLoaderInterface#saveModel(java.lang.String, java.lang.String)
-   */
-  @Override
-  public Response saveModel(String modelid, String modelname) {
-    Response response = Response.ok("{\"model_saved\":\"" + modelid + "-" + modelname + "\"}")
-        .build();
-
-    return response;
-  }
-
-  @Override
-  public Response ingestModel(String modelid, HttpServletRequest req, String payload)
-      throws IOException {
-    Response response;
-
-    if (config.getIngestSimulatorEnabled()) {
-      logger.info(ModelLoaderMsgs.DISTRIBUTION_EVENT, "Received test artifact");
-
-      ModelArtifactHandler handler = new ModelArtifactHandler(config);
-      handler.loadModelTest(payload.getBytes());
-
-      response = Response.ok().build();
-    } else {
-      logger.debug("Simulation interface disabled");
-      response = Response.serverError().build();
-    }
-
-    return response;
-  }
+	
+	protected static final String FILESEP = (System.getProperty("file.separator") == null) ? "/"
+            : System.getProperty("file.separator");
+
+	protected static final String CONFIG_DIR = System.getProperty("CONFIG_HOME") + FILESEP;
+	protected static final String CONFIG_AUTH_LOCATION = CONFIG_DIR + "auth" + FILESEP;
+	protected static final String CONFIG_FILE = CONFIG_DIR + "model-loader.properties";
+
+	private IDistributionClient client;
+	private ModelLoaderConfig config;
+	private Timer timer = null;
+
+	static Logger logger = LoggerFactory.getInstance().getLogger(ModelLoaderService.class.getName());
+
+	/**
+	 * Responsible for loading configuration files and calling initialization.
+	 */
+	public ModelLoaderService() {
+		start();
+	}
+
+	protected void start() {
+		// Load model loader system configuration
+		logger.info(ModelLoaderMsgs.LOADING_CONFIGURATION);
+		Properties configProperties = new Properties();
+		try {
+			configProperties.load(new FileInputStream(CONFIG_FILE));
+		} catch (IOException e) {
+			String errorMsg = "Failed to load configuration: " + e.getMessage();
+			logger.error(ModelLoaderMsgs.ASDC_CONNECTION_ERROR, errorMsg);
+			shutdown();
+		}
+
+		config = new ModelLoaderConfig(configProperties, CONFIG_AUTH_LOCATION);
+		init();
+		
+		Runtime.getRuntime().addShutdownHook(new Thread(new Runnable(){
+			public void run() {
+				preShutdownOperations();
+			}
+		}));
+	}
+	
+	/**
+	 * Responsible for stopping the connection to the distribution client before
+	 * the resource is destroyed.
+	 */
+	protected void preShutdownOperations() {
+		logger.info(ModelLoaderMsgs.STOPPING_CLIENT);
+		if (client != null) {
+			client.stop();
+		}
+	}
+
+	/**
+	 * Responsible for loading configuration files, initializing model
+	 * distribution clients, and starting them.
+	 */
+	protected void init() {
+		// Initialize distribution client
+		logger.debug(ModelLoaderMsgs.INITIALIZING, "Initializing distribution client...");
+		client = DistributionClientFactory.createDistributionClient();
+		EventCallback callback = new EventCallback(client, config);
+		
+		IDistributionClientResult initResult = client.init(config, callback);
+
+		if (initResult.getDistributionActionResult() != DistributionActionResultEnum.SUCCESS) {
+			String errorMsg = "Failed to initialize distribution client: "
+					+ initResult.getDistributionMessageResult();
+			logger.error(ModelLoaderMsgs.ASDC_CONNECTION_ERROR, errorMsg);
+			
+			// Kick off a timer to retry the SDC connection
+			timer = new Timer();
+			TimerTask task = new SdcConnectionJob(client, config, callback, timer);
+			timer.schedule(task, new Date(), 60000);
+		}
+		else {
+			// Start distribution client
+			logger.debug(ModelLoaderMsgs.INITIALIZING, "Starting distribution client...");
+			IDistributionClientResult startResult = client.start();
+			if (startResult.getDistributionActionResult() != DistributionActionResultEnum.SUCCESS) {
+				String errorMsg = "Failed to start distribution client: "
+						+ startResult.getDistributionMessageResult();
+				logger.error(ModelLoaderMsgs.ASDC_CONNECTION_ERROR, errorMsg);
+
+				// Kick off a timer to retry the SDC connection
+				timer = new Timer();
+				TimerTask task = new SdcConnectionJob(client, config, callback, timer);
+				timer.schedule(task, new Date(), 60000);
+			}
+			else {
+				logger.info(ModelLoaderMsgs.INITIALIZING, "Connection to SDC established");
+			}
+		}
+	}
+
+	/**
+	 * Shut down the process.
+	 */
+	private void shutdown() {
+		preShutdownOperations();
+
+		// TODO: Find a better way to shut down the model loader.
+		try {
+			// Give logs time to write to file
+			Thread.sleep(2000);
+		} catch (InterruptedException e) {
+			// Nothing we can do at this point
+		}
+
+		Runtime.getRuntime().halt(1);
+	}
+
+	/** (non-Javadoc)
+	 * @see org.openecomp.modelloader.service.ModelLoaderInterface#loadModel(java.lang.String)
+	 */
+	@Override
+	public Response loadModel(String modelid) {
+		Response response = Response.ok("{\"model_loaded\":\"" + modelid + "\"}").build();
+
+		return response;
+	}
+
+	/** (non-Javadoc)
+	 * @see org.openecomp.modelloader.service.ModelLoaderInterface#saveModel(java.lang.String, java.lang.String)
+	 */
+	@Override
+	public Response saveModel(String modelid, String modelname) {
+		Response response = Response.ok("{\"model_saved\":\"" + modelid + "-" + modelname + "\"}")
+				.build();
+
+		return response;
+	}
+
+	@Override
+	public Response ingestModel(String modelid, HttpServletRequest req, String payload)
+			throws IOException {
+		Response response;
+
+		if (config.getIngestSimulatorEnabled()) {
+			logger.info(ModelLoaderMsgs.DISTRIBUTION_EVENT, "Received test artifact");
+
+			ModelArtifactHandler handler = new ModelArtifactHandler(config);
+			handler.loadModelTest(payload.getBytes());
+
+			response = Response.ok().build();
+		} else {
+			logger.debug("Simulation interface disabled");
+			response = Response.serverError().build();
+		}
+
+		return response;
+	}
 }