Move registration to exchanges after agent start.

tools.descartes.prisma.agent.wildfly/src/main/java/tools/descartes/prisma/agent/wildfly/service/ModelExtractionService.java

@@ -214,23 +214,26 @@ public class ModelExtractionService extends AgentController implements Service<M
 			}
 		});
 
-		ApplicationScope applicationScope = (ApplicationScope) getContainerScope();
-		addNotificationHandler(applicationScope.getRuntimeEnvironmentInputPort(), "#", new NotificationHandler() {
-			@Override
-			public void handleNotification(ModelNotification event) {
-				EObject instance = event.getInstance();
-				if (instance instanceof RuntimeEnvironment) {
-					RuntimeEnvironment env = (RuntimeEnvironment) instance;
-					if (env.getOfClass() == RuntimeEnvironmentClasses.OS_VM
-							&& env.getName().equalsIgnoreCase(getLocalHost().getHostName())) {
-						wildFlyContainer = ModelSkeletonUtil.adopt(ModelExtractionService.this, env);
-					}
-				}
-			}
-		});
+
 
 		try {
 			start().get();
+
+			ApplicationScope applicationScope = (ApplicationScope) getContainerScope();
+			addNotificationHandler(applicationScope.getRuntimeEnvironmentInputPort(), "#", new NotificationHandler() {
+				@Override
+				public void handleNotification(ModelNotification event) {
+					EObject instance = event.getInstance();
+					if (instance instanceof RuntimeEnvironment) {
+						RuntimeEnvironment env = (RuntimeEnvironment) instance;
+						if (env.getOfClass() == RuntimeEnvironmentClasses.OS_VM
+								&& env.getName().equalsIgnoreCase(getLocalHost().getHostName())) {
+							wildFlyContainer = ModelSkeletonUtil.adopt(ModelExtractionService.this, env);
+						}
+					}
+				}
+			});
+
 			schedule(new ModelUpdater(), 1, TimeUnit.MINUTES);
 		} catch (InterruptedException | ExecutionException e) {
 			log.error("Error stopping agent.", e);