1 files changed, 227 insertions, 0 deletions
diff --git a/feature-test-transaction/src/main/java/org/onap/policy/drools/testtransaction/TestTransaction.java b/feature-test-transaction/src/main/java/org/onap/policy/drools/testtransaction/TestTransaction.java
new file mode 100644
index 00000000..b167898a
--- /dev/null
+++ b/feature-test-transaction/src/main/java/org/onap/policy/drools/testtransaction/TestTransaction.java
@@ -0,0 +1,227 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * feature-test-transaction
+ * ================================================================================
+ * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
+ * ================================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ * 
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ * 
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END=========================================================
+ */
+package org.onap.policy.drools.testtransaction;
+
+import java.util.EventObject;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+
+
+import org.onap.policy.drools.controller.DroolsController;
+import org.onap.policy.drools.controller.internal.MavenDroolsController;
+import org.onap.policy.drools.system.PolicyController;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+/**
+ * TestTransaction interface 
+ *
+ */
+public interface TestTransaction {
+
+	public static final String TT_FPC = "TT.FPC";
+	public static final String TT_COUNTER = "$ttc";
+	public static final String TT_UUID = "43868e59-d1f3-43c2-bd6f-86f89a61eea5";
+	public static long DEFAULT_TT_TASK_SLEEP = 20000;
+	
+	public static final TestTransaction manager = new TTImpl();
+
+	public void register(PolicyController controller);
+	public void unregister(PolicyController controller);
+}
+
+/**
+ * Implementation of TestTransaction interface.
+ * Controls the registering/unregistering of 
+ * PolicyController objects and the management
+ * of their related TTControllerTask threads. 
+ */
+class TTImpl implements TestTransaction {
+	
+	final protected Map<String, TTControllerTask> controllers = new HashMap<>();
+
+	@Override
+	public synchronized void register(PolicyController controller) {
+		if (controllers.containsValue(controller)) {
+			TTControllerTask controllerTask = controllers.get(controller.getName());
+			if (controllerTask.isAlive())
+				return;
+			
+			// continue : unregister, register operation
+		}
+		
+		TTControllerTask controllerTask = new TTControllerTask(controller);
+		controllers.put(controller.getName(), controllerTask);
+	}
+
+	@Override
+	public synchronized void unregister(PolicyController controller) {
+		if (!controllers.containsValue(controller))
+			return;
+		
+		TTControllerTask controllerTask = controllers.get(controller.getName());
+		controllerTask.stop();
+		
+		controllers.remove(controller.getName());
+	}
+}
+
+/**
+ * TTControllerTask implements the Runnabale interface
+ * Carries out the injection of an event into a drools 
+ * session and subsequent query of a counter to ensure
+ * that forward progress is occuring. 
+ * 
+ */
+class TTControllerTask implements Runnable {
+    // get an instance of logger 
+	private static final Logger logger = LoggerFactory.getLogger(TTControllerTask.class); 
+   
+	protected final PolicyController controller;
+
+	protected volatile boolean alive = true;
+	protected final Thread thread = new Thread(this);
+
+	public TTControllerTask(PolicyController controller) {
+		this.controller = controller;
+		thread.setName("tt-controller-task-" + controller.getName());
+		thread.start();
+	}
+	
+	public PolicyController getController() {
+		return controller;
+	}
+
+	public synchronized boolean isAlive() {
+		return alive;
+	}
+
+	public synchronized void stop() {
+		this.alive = false;
+		thread.interrupt();
+		try {
+			thread.join(1000);
+		} catch (InterruptedException e) {
+			logger.error("TestTransaction thread threw", e);
+			thread.interrupt();
+		}
+	}	
+	
+	public Thread getThread() {
+		return thread;
+	}
+
+	@Override
+	public void run() {
+		try {	
+			List<String> sessions = 
+					controller.getDrools().getSessionNames();
+			
+			if (!(controller.getDrools().isBrained())) {
+				alive = false;
+				logger.error(this + ": unknown drools controller");
+				return;
+			}
+			
+			DroolsController drools = controller.getDrools();
+			
+			HashMap<String,Long> fpcs = new HashMap<>();
+			for (String session: sessions) {
+				fpcs.put(session, -1L);
+			}
+			
+			while (controller.isAlive() && 
+				   !controller.isLocked() &&
+				   drools.isBrained() &&
+				   alive) {
+				
+				for (String session : sessions) {
+					List<Object> facts = controller.getDrools().factQuery(session,
+							TestTransaction.TT_FPC,
+							TestTransaction.TT_COUNTER,
+							false);
+					if (facts == null || facts.size() != 1) {
+						/* 
+						 * unexpected something wrong here, can't expect to recover 
+						 * note this exception is caught right below at the exit of run()
+						 */
+						logger.error("Controller: {}, with rules artifact: (group) {}, (artifact) {}, (version) {} - FPC query failed after EventObject insertion! ",
+										controller.getName(),
+										controller.getDrools().getGroupId(),
+										controller.getDrools().getArtifactId(), 
+										controller.getDrools().getVersion());
+						break;
+					}
+					logger.debug("Facts: {}", facts);
+					
+					long fpc = (Long) facts.get(0);
+					if (fpc != fpcs.get(session))
+						logger.info("Controller: {} , session {}  - Forward progress successful: {} -> {}",
+										controller.getName(),
+										session,
+										fpcs.get(session), 
+										fpc);
+					else 
+						logger.error("Controller: {}, session {} - Forward progress failure: {}",
+								controller.getName(),
+								session,
+								fpc);
+					
+					fpcs.put(session, fpc);									
+					drools.getContainer().insert(session, new EventObject(TestTransaction.TT_UUID));
+				}
+				
+				if (!alive)
+					return;
+				
+				if (!Thread.currentThread().isInterrupted())
+					Thread.sleep(TestTransaction.DEFAULT_TT_TASK_SLEEP);
+			}
+		} catch (InterruptedException e) {
+			logger.info("{}: stopping ...", this, e);
+		} 
+		catch (IllegalArgumentException e) {
+			logger.error("{}: controller {} has not been enabled for testing: ", this, controller.getName(), e.getMessage());
+		} catch (Exception e) {
+			logger.error("Controller: {} is not testable - TestTransaction caught exception: {} ",
+					controller.getName(),
+					e.getMessage());
+			logger.error("TestTransaction thread threw", e);
+		} finally {
+			logger.info("Exiting: {}", this);
+			alive = false;
+		}
+	}
+
+	@Override
+	public String toString() {
+		StringBuilder builder = new StringBuilder();
+		builder.append("TTControllerTask [controller=");
+		builder.append(controller);
+		builder.append(", alive=");
+		builder.append(alive);
+		builder.append(", thread=");
+		builder.append(thread.getName());
+		builder.append("]");
+		return builder.toString();
+	}
+	
+}