Add websocket client code.

Change-Id: I2d3939c698a0df6bec068063f08566226f65bfd3
Issue-ID: POLICY-756
Signed-off-by: pa834y <pa834y@att.com>

3 files changed, 299 insertions, 353 deletions

diff --git a/PolicyEngineAPI/src/main/java/org/onap/policy/std/AutoClientEnd.java b/PolicyEngineAPI/src/main/java/org/onap/policy/std/AutoClientEnd.java
index 3f97e19f7..9f9dc37e4 100644
--- a/PolicyEngineAPI/src/main/java/org/onap/policy/std/AutoClientEnd.java
+++ b/PolicyEngineAPI/src/main/java/org/onap/policy/std/AutoClientEnd.java
@@ -20,16 +20,8 @@
 
 package org.onap.policy.std;
 
-import java.io.IOException;
 import java.net.URI;
-
 import javax.websocket.ClientEndpoint;
-import javax.websocket.OnClose;
-import javax.websocket.OnError;
-import javax.websocket.OnMessage;
-import javax.websocket.OnOpen;
-import javax.websocket.Session;
-
 import org.java_websocket.client.WebSocketClient;
 import org.java_websocket.handshake.ServerHandshake;
 import org.onap.policy.api.NotificationHandler;
@@ -38,231 +30,211 @@ import org.onap.policy.api.NotificationType;
 import org.onap.policy.api.PDPNotification;
 import org.onap.policy.common.logging.flexlogger.FlexLogger;
 import org.onap.policy.common.logging.flexlogger.Logger;
-import org.onap.policy.xacml.api.XACMLErrorConstants; 
+import org.onap.policy.xacml.api.XACMLErrorConstants;
 
 @ClientEndpoint
 public class AutoClientEnd extends WebSocketClient {
-	private static StdPDPNotification notification = null;
-	private static StdPDPNotification oldNotification = null;
-	private static AutoClientEnd client = null;
-	private static NotificationScheme scheme = null;
-	private static NotificationHandler handler = null;
-	private static String url = null;
-	private static Session session = null;
-	private static boolean status = false; 
-	private static boolean stop = false;
-	private static boolean message = false;
-	private static boolean error = false;
-	private static Logger logger = FlexLogger.getLogger(AutoClientEnd.class.getName());
-	
-	private AutoClientEnd(URI serverUri) {
-		super(serverUri);
-	}
-
-	@Override
-	public void onClose(int arg0, String arg1, boolean arg2) {
-		// Not implemented
-	}
-
-	@Override
-	public void onError(Exception arg0) {
-		// Not implemented
-	}
-
-	@Override
-	public void onMessage(String arg0) {
-		// Not implemented
-	}
-
-	@Override
-	public void onOpen(ServerHandshake arg0) {
-		// Not implemented
-	}
-
-	public static void setAuto(NotificationScheme scheme,
-			NotificationHandler handler) {
-		AutoClientEnd.scheme = scheme;
-		AutoClientEnd.handler = handler;
-	}
-
-	public static void setScheme(NotificationScheme scheme) {
-		AutoClientEnd.scheme = scheme;
-	}
-	
-	public static boolean getStatus(){
-		return AutoClientEnd.status;
-	}
-
-	public static String getURL() {
-		return AutoClientEnd.url;
-	}
-
-	public static void start(String url) {
-		AutoClientEnd.url = url;
-		
-		if (scheme == null || handler == null ||
-			! (scheme.equals(NotificationScheme.AUTO_ALL_NOTIFICATIONS) ||
-					scheme.equals(NotificationScheme.AUTO_NOTIFICATIONS) ) ||
-			AutoClientEnd.client != null) {
-			return;
-		}
-
-		if (url.contains("https")) {
-			url = url.replaceAll("https", "wss");
-		}
-		else {
-			url = url.replaceAll("http", "ws");
-		}
-		
-		
-		// Stop and Start needs to be done.
-		try {
-			logger.info("Starting Auto Notification with the PDP server : " + url);
-			client = new AutoClientEnd(new URI(url	+ "notifications"));
-			status = true;
-			if(error){
-				// The URL's will be in Sync according to design Spec. 
-				ManualClientEnd.start(AutoClientEnd.url);
-				StdPDPNotification notification = NotificationStore.getDeltaNotification((StdPDPNotification)ManualClientEnd.result(NotificationScheme.MANUAL_ALL_NOTIFICATIONS));
-				if(notification.getNotificationType()!=null&&oldNotification!=notification){
-				    oldNotification= notification;
-				    AutoClientEnd.notification = notification;
-				    callHandler();
-				}
-				error = false;
-			}
-			//
-		} catch (Exception e) {
-			logger.error(XACMLErrorConstants.ERROR_SYSTEM_ERROR + e);
-			client = null;
-			status = false;
-			changeURL();
-		}
-	}
-	
-	private static void changeURL(){
-		// Change the PDP if it is not Up. 
-		StdPolicyEngine.rotatePDPList();
-		start(StdPolicyEngine.getPDPURL());
-	}
-
-	public static void stop() {
-		if (client == null) {
-			return;
-		}
-		client.close();
-		if(session!=null){
-			try {
-				stop = true;
-				logger.info("\n Closing Auto Notification WebSocket Connection.. ");
-				session.close();
-				session = null;
-			} catch (IOException e) {
-				logger.error("Error closing websocket connection", e);
-			}
-		}
-		client = null;
-		status = false;
-		stop = false;
-	}
-
-	private static void callHandler() {
-		if (handler == null || scheme == null) {
-			return;
-		}
-		if (scheme.equals(NotificationScheme.AUTO_ALL_NOTIFICATIONS)) {
-			boolean removed = false;
-			boolean updated = false;
-			if (notification.getRemovedPolicies() != null && !notification.getRemovedPolicies().isEmpty()) {
-				removed = true;
-			}
-			if (notification.getLoadedPolicies() != null && !notification.getLoadedPolicies().isEmpty()) {
-				updated = true;
-			}
-			if (removed && updated) {
-				notification.setNotificationType(NotificationType.BOTH);
-			} else if (removed) {
-				notification.setNotificationType(NotificationType.REMOVE);
-			} else if (updated) {
-				notification.setNotificationType(NotificationType.UPDATE);
-			}
-			try{
-				handler.notificationReceived(notification);
-			}catch (Exception e){
-				logger.error("Error in Clients Handler Object : ", e);
-			}
-		} else if (scheme.equals(NotificationScheme.AUTO_NOTIFICATIONS)) {
-			PDPNotification newNotification = MatchStore.checkMatch(notification);
-			if (newNotification.getNotificationType() != null) {
-				try{
-					handler.notificationReceived(newNotification);
-				}catch (Exception e){
-					logger.error("Error in Clients Handler Object : ", e);
-				}
-			}
-		}
-	}
-
-	// WebSockets Code..
-	@OnOpen
-	public static void onOpen(Session session){
-		logger.debug("Auto Notification Session Started... " + session.getId());
-		if(AutoClientEnd.session == null){
-			AutoClientEnd.session = session;
-		}
-	}
-
-	@OnError
-	public static void onError(Session session, Throwable e) {
-		// trying to Restart by self.
-		logger.error(XACMLErrorConstants.ERROR_PROCESS_FLOW + "Session Error.. "+ session.getId() + "\n Error is : " + e );
-		stop();
-		if (url != null) {
-			client = null;
-			status = false;
-			error= true;
-			start(url);
-		}
-	}
-
-	@OnClose
-	public static void onClose(Session session) {
-		logger.info("Session ended with "+ session.getId());
-		if(!stop && !message){
-			// This Block of code is executed if there is any Network Failure or if the Notification is Down. 
-			logger.error(XACMLErrorConstants.ERROR_SYSTEM_ERROR + "Disconnected from Notification Server");
-			client = null;
-			status = false;
-			AutoClientEnd.session=null;
-			// Try to connect Back to available PDP.
-			error = true;
-			start(url);
-		}
-		AutoClientEnd.message=false;
-	}
-
-	@OnMessage
-	public static void onMessage(String message, Session session) throws IOException {
-		AutoClientEnd.message = true;
-		logger.debug("Auto Notification Recieved Message " + message + " Session info is : " + session.getId());
-		try {
-			notification = NotificationUnMarshal.notificationJSON(message);
-		} catch (Exception e) {
-			logger.error("PE500 " + e);
-		}
-		if(AutoClientEnd.session == session){
-			try{
-				NotificationStore.recordNotification(notification);
-			}catch(Exception e){
-				logger.error(e);
-			}
-			if(oldNotification!=notification){
-				oldNotification= notification;
-				callHandler();
-			}
-		}else{
-			session.close();
-		}
-		AutoClientEnd.message = false;
-	}
+    private static StdPDPNotification notification = null;
+    private static StdPDPNotification oldNotification = null;
+    private static AutoClientEnd client = null;
+    private static NotificationScheme scheme = null;
+    private static NotificationHandler handler = null;
+    private static String url = null;
+    private static boolean status = false;
+    private static boolean stop = false;
+    private static boolean message = false;
+    private static boolean error = false;
+    private static Logger logger = FlexLogger.getLogger(AutoClientEnd.class.getName());
+
+    private AutoClientEnd(URI serverUri) {
+        super(serverUri);
+    }
+
+    @Override
+    public void onMessage(String msg) {
+        logger.info("Received Auto Notification from : " + getURI() + ", Notification: " + msg);
+        AutoClientEnd.message = true;
+        try {
+            AutoClientEnd.notification = NotificationUnMarshal.notificationJSON(msg);
+        } catch (Exception e) {
+            logger.error("PE500 " + e);
+        }
+        try {
+            NotificationStore.recordNotification(notification);
+        } catch (Exception e) {
+            logger.error(e);
+        }
+        if (AutoClientEnd.oldNotification != AutoClientEnd.notification) {
+            AutoClientEnd.oldNotification = AutoClientEnd.notification;
+            callHandler();
+        }
+
+        AutoClientEnd.message = false;
+    }
+
+    @Override
+    public void onClose(int code, String reason, boolean remote) {
+        logger.info("AutoClientEnd disconnected from: " + getURI() + "; Code: " + code + ", reason :  " + reason);
+        if (!AutoClientEnd.stop && !AutoClientEnd.message) {
+            // This Block of code is executed if there is any Network Failure or
+            // if the Notification is Down.
+            logger.error(XACMLErrorConstants.ERROR_SYSTEM_ERROR + "Disconnected from Notification Server");
+            AutoClientEnd.client = null;
+            AutoClientEnd.status = false;
+            // Try to connect Back to available PDP.
+            AutoClientEnd.error = true;
+            start(url);
+        }
+        AutoClientEnd.message = false;
+    }
+
+    @Override
+    public void onError(Exception ex) {
+        logger.error("XACMLErrorConstants.ERROR_PROCESS_FLOW + Error connecting to: " + getURI()
+                + ", Exception occured ...\n" + ex);
+        // trying to Restart by self.
+        stop();
+        if (AutoClientEnd.url != null) {
+            AutoClientEnd.client = null;
+            AutoClientEnd.status = false;
+            AutoClientEnd.error = true;
+            AutoClientEnd.start(AutoClientEnd.url);
+        }
+    }
+
+    @Override
+    public void onOpen(ServerHandshake arg0) {
+        logger.info("Auto Notification Session Started... " + getURI());
+    }
+
+    /**
+     * Sets the auto.
+     *
+     * @param scheme the scheme
+     * @param handler the handler
+     */
+    public static void setAuto(NotificationScheme scheme, NotificationHandler handler) {
+        logger.info("Auto Notification setAuto, scheme: " + scheme);
+        AutoClientEnd.scheme = scheme;
+        AutoClientEnd.handler = handler;
+    }
+
+    public static void setScheme(NotificationScheme scheme) {
+        AutoClientEnd.scheme = scheme;
+    }
+
+    public static boolean getStatus() {
+        return AutoClientEnd.status;
+    }
+
+    public static String getUrl() {
+        return AutoClientEnd.url;
+    }
+
+    /**
+     * Start.
+     *
+     * @param url the url
+     */
+    public static void start(String url) {
+        AutoClientEnd.url = url;
+
+        if (scheme == null || handler == null || !(scheme.equals(NotificationScheme.AUTO_ALL_NOTIFICATIONS)
+                || scheme.equals(NotificationScheme.AUTO_NOTIFICATIONS)) || AutoClientEnd.client != null) {
+            return;
+        }
+
+        if (url.contains("https")) {
+            url = url.replaceAll("https", "wss");
+        } else {
+            url = url.replaceAll("http", "ws");
+        }
+
+        // Stop and Start needs to be done.
+        try {
+            logger.info("Starting Auto Notification with the PDP server : " + url);
+            client = new AutoClientEnd(new URI(url + "notifications"));
+            client.connect();
+            status = true;
+            if (error) {
+                // will not trigger. leave it in to later add checks
+                // The URL's will be in Sync according to design Spec.
+                ManualClientEnd.start(AutoClientEnd.url);
+                StdPDPNotification notification = NotificationStore.getDeltaNotification(
+                        (StdPDPNotification) ManualClientEnd.result(NotificationScheme.MANUAL_ALL_NOTIFICATIONS));
+                if (notification.getNotificationType() != null && oldNotification != notification) {
+                    oldNotification = notification;
+                    AutoClientEnd.notification = notification;
+                    callHandler();
+                }
+                error = false;
+            }
+        } catch (Exception e) {
+            logger.error(XACMLErrorConstants.ERROR_SYSTEM_ERROR + e);
+            client = null;
+            status = false;
+            changeUrl();
+        }
+    }
+
+    private static void changeUrl() {
+        // Change the PDP if it is not Up.
+        StdPolicyEngine.rotatePDPList();
+        start(StdPolicyEngine.getPDPURL());
+    }
+
+    /**
+     * Stop the websocket connection.
+     */
+    public static void stop() {
+        if (client == null) {
+            return;
+        }
+        logger.info("\n Closing Auto Notification WebSocket Connection.. ");
+        stop = true;
+        try {
+            client.closeBlocking();
+        } catch (InterruptedException e) {
+            logger.info("\n Error Closing Auto Notification WebSocket Connection.. InterruptedException");
+        }
+        logger.info("\n Closed the Auto Notification WebSocket Connection.. ");
+        client = null;
+        status = false;
+        stop = false;
+    }
+
+    private static void callHandler() {
+        if (handler == null || scheme == null) {
+            return;
+        }
+        if (scheme.equals(NotificationScheme.AUTO_ALL_NOTIFICATIONS)) {
+            boolean removed = false;
+            boolean updated = false;
+            if (notification.getRemovedPolicies() != null && !notification.getRemovedPolicies().isEmpty()) {
+                removed = true;
+                notification.setNotificationType(NotificationType.REMOVE);
+            }
+            if (notification.getLoadedPolicies() != null && !notification.getLoadedPolicies().isEmpty()) {
+                updated = true;
+                notification.setNotificationType(NotificationType.UPDATE);
+            }
+            if (removed && updated) {
+                notification.setNotificationType(NotificationType.BOTH);
+            }
+            try {
+                handler.notificationReceived(notification);
+            } catch (Exception e) {
+                logger.error("Error in Clients Handler Object : ", e);
+            }
+        } else if (scheme.equals(NotificationScheme.AUTO_NOTIFICATIONS)) {
+            PDPNotification newNotification = MatchStore.checkMatch(notification);
+            if (newNotification.getNotificationType() != null) {
+                try {
+                    handler.notificationReceived(newNotification);
+                } catch (Exception e) {
+                    logger.error("Error in Clients Handler Object : ", e);
+                }
+            }
+        }
+    }
 }
diff --git a/PolicyEngineAPI/src/main/java/org/onap/policy/std/ManualClientEnd.java b/PolicyEngineAPI/src/main/java/org/onap/policy/std/ManualClientEnd.java
index 991bdca9c..2fe6dc006 100644
--- a/PolicyEngineAPI/src/main/java/org/onap/policy/std/ManualClientEnd.java
+++ b/PolicyEngineAPI/src/main/java/org/onap/policy/std/ManualClientEnd.java
@@ -20,144 +20,118 @@
 
 package org.onap.policy.std;
 
-import java.io.IOException;
 import java.net.URI;
 import java.util.concurrent.CountDownLatch;
-
 import javax.websocket.ClientEndpoint;
-import javax.websocket.OnClose;
-import javax.websocket.OnError;
-import javax.websocket.OnMessage;
-import javax.websocket.OnOpen;
-import javax.websocket.Session;
-
 import org.java_websocket.client.WebSocketClient;
 import org.java_websocket.handshake.ServerHandshake;
 import org.onap.policy.api.NotificationScheme;
 import org.onap.policy.api.NotificationType;
 import org.onap.policy.api.PDPNotification;
+import org.onap.policy.common.logging.flexlogger.FlexLogger;
+import org.onap.policy.common.logging.flexlogger.Logger;
 import org.onap.policy.std.StdPDPNotification;
-
 import org.onap.policy.xacml.api.XACMLErrorConstants;
 
-import org.onap.policy.common.logging.flexlogger.*; 
-
 @ClientEndpoint
 public class ManualClientEnd extends WebSocketClient {
-	private static CountDownLatch latch;
-	private static StdPDPNotification notification = null;
-	private static String resultJson = null;
-	private static Logger logger = FlexLogger.getLogger(ManualClientEnd.class.getName());
-	private static ManualClientEnd client;
-	
-	public ManualClientEnd(URI serverUri) {
-		super(serverUri);
-	}
+    private static CountDownLatch latch;
+    private static StdPDPNotification notification = null;
+    private static String resultJson = null;
+    private static Logger logger = FlexLogger.getLogger(ManualClientEnd.class.getName());
+    private static ManualClientEnd client;
+
+    public ManualClientEnd(URI serverUri) {
+        super(serverUri);
+    }
+
+    @Override
+    public void onClose(int code, String reason, boolean remote) {
+        logger.info("ManualClientEnd disconnected from: " + getURI() + "; Code: " + code + ", reason :  " + reason);
+        latch.countDown();
+    }
 
-	@Override
-	public void onClose(int arg0, String arg1, boolean arg2) {
-		// Not implemented
-	}
+    @Override
+    public void onError(Exception ex) {
+        logger.error("XACMLErrorConstants.ERROR_PROCESS_FLOW + ManualClientEnd - Error connecting to: " + getURI()
+                + ", Exception occured ...\n" + ex);
+        latch.countDown();
+    }
 
-	@Override
-	public void onError(Exception arg0) {
-		// Not implemented
-	}
+    @Override
+    public void onMessage(String message) {
+        logger.info("Manual Notification Recieved Message from : " + getURI() + ", Notification: " + message);
+        ManualClientEnd.resultJson = message;
+        try {
+            ManualClientEnd.notification = NotificationUnMarshal.notificationJSON(message);
+            latch.countDown();
+        } catch (Exception e) {
+            logger.error(XACMLErrorConstants.ERROR_DATA_ISSUE + e);
+            latch.countDown();
+        }
+    }
 
-	@Override
-	public void onMessage(String arg0) {
-		// Not implemented
-	}
+    @Override
+    public void onOpen(ServerHandshake arg0) {
+        logger.info("Manual Notification Session Started... " + getURI());
+        send("Manual");
+    }
 
-	@Override
-	public void onOpen(ServerHandshake arg0) {
-		// Not implemented
-	}
+    /**
+     * Start.
+     *
+     * @param url the url
+     */
+    public static void start(String url) {
+        latch = new CountDownLatch(1);
 
-	public static void start(String url) {
-		latch = new CountDownLatch(1);
+        if (url.contains("https")) {
+            url = url.replaceAll("https", "wss");
+        } else {
+            url = url.replaceAll("http", "ws");
+        }
 
-		if (url.contains("https")) {
-			url = url.replaceAll("https", "wss");
-		}
-		else {
-			url = url.replaceAll("http", "ws");
-		}
-		
-		try {
-			client = new ManualClientEnd(new URI(url+"notifications"));
-			latch.await();
-		} catch (Exception e) {
-			logger.error(XACMLErrorConstants.ERROR_SYSTEM_ERROR + e);
-		}
-	}
+        try {
+            client = new ManualClientEnd(new URI(url + "notifications"));
+            client.connect();
+            latch.await();
+            client.close();
+        } catch (Exception e) {
+            logger.error(XACMLErrorConstants.ERROR_SYSTEM_ERROR + e);
+        }
+    }
 
-	public static PDPNotification result(NotificationScheme scheme) {
-		if (resultJson == null || notification == null) {
-			logger.debug("No Result" );
-			return null;
-		} else {
-			if(scheme.equals(NotificationScheme.MANUAL_ALL_NOTIFICATIONS)) {
-				boolean removed = false;
-				boolean updated = false; 
-				if(notification.getRemovedPolicies()!=null && !notification.getRemovedPolicies().isEmpty()){
-					removed = true;
-				}
-				if(notification.getLoadedPolicies()!=null && !notification.getLoadedPolicies().isEmpty()){
-					updated = true;
-				}
-				if(removed && updated) {
-					notification.setNotificationType(NotificationType.BOTH);
-				}else if(removed){
-					notification.setNotificationType(NotificationType.REMOVE);
-				}else if(updated){
-					notification.setNotificationType(NotificationType.UPDATE);
-				}
-				return notification;
-			}else if(scheme.equals(NotificationScheme.MANUAL_NOTIFICATIONS)) {
-				return MatchStore.checkMatch(notification);
-			}else {
-				return null;
-			}
-		}
-	}
-	
-	// WebSockets Code.. 
-	@OnOpen
-	public void onOpen(Session session) throws IOException {
-		logger.info("Session Started with : " + session.getId());
-		session.getBasicRemote().sendText("Manual");
-	}
-	
-	@OnError
-	public void onError(Session session, Throwable e) {
-		logger.error(XACMLErrorConstants.ERROR_PROCESS_FLOW + "Error in: "+ session.getId());
-		latch.countDown();
-	}
-	
-	@OnClose
-	public void onClose(Session session) {
-		logger.info("Session ended with "+ session.getId());
-		latch.countDown();
-		client.close();
-	}
-	
-	@OnMessage
-	public static void onMessage(String message, Session session){
-		logger.debug(" Manual Notification Recieved Message : " + message +" Session info is : "+ session.getId());
-		resultJson = message;
-		try {
-			notification = NotificationUnMarshal.notificationJSON(message);
-		} catch (Exception e) {
-			logger.error(XACMLErrorConstants.ERROR_DATA_ISSUE + e);
-			latch.countDown();
-		}
-		try {
-			session.close();
-		} catch (IOException e) {
-			logger.error(XACMLErrorConstants.ERROR_PROCESS_FLOW + e);
-			latch.countDown();
-		} // For Manual Client..
-		latch.countDown();
-	}
+    /**
+     * Result.
+     *
+     * @param scheme the scheme
+     * @return the PDP notification
+     */
+    public static PDPNotification result(NotificationScheme scheme) {
+        if (resultJson == null || notification == null) {
+            logger.info("ManualClientENd - No Result available");
+            return null;
+        } else {
+            if (scheme.equals(NotificationScheme.MANUAL_ALL_NOTIFICATIONS)) {
+                boolean removed = false;
+                boolean updated = false;
+                if (notification.getRemovedPolicies() != null && !notification.getRemovedPolicies().isEmpty()) {
+                    removed = true;
+                    notification.setNotificationType(NotificationType.REMOVE);
+                }
+                if (notification.getLoadedPolicies() != null && !notification.getLoadedPolicies().isEmpty()) {
+                    updated = true;
+                    notification.setNotificationType(NotificationType.UPDATE);
+                }
+                if (removed && updated) {
+                    notification.setNotificationType(NotificationType.BOTH);
+                }
+                return notification;
+            } else if (scheme.equals(NotificationScheme.MANUAL_NOTIFICATIONS)) {
+                return MatchStore.checkMatch(notification);
+            } else {
+                return null;
+            }
+        }
+    }
 }
diff --git a/PolicyEngineAPI/src/main/java/org/onap/policy/std/StdPolicyEngine.java b/PolicyEngineAPI/src/main/java/org/onap/policy/std/StdPolicyEngine.java
index 2349c2e2f..f09b5779c 100644
--- a/PolicyEngineAPI/src/main/java/org/onap/policy/std/StdPolicyEngine.java
+++ b/PolicyEngineAPI/src/main/java/org/onap/policy/std/StdPolicyEngine.java
@@ -1103,7 +1103,7 @@ public class StdPolicyEngine {
                 this.dmaapThread = true;
             } else {
                 if (pdps.get(0) != null) {
-                    if (AutoClientEnd.getURL() == null) {
+                    if (AutoClientEnd.getUrl() == null) {
                         AutoClientEnd.start(pdps.get(0));
                     } else {
                         AutoClientEnd.stop();