summaryrefslogtreecommitdiffstats
path: root/PolicyEngineAPI/src/main
diff options
context:
space:
mode:
authorpa834y <pa834y@att.com>2018-04-24 20:25:22 -0400
committerpa834y <pa834y@att.com>2018-04-24 20:25:55 -0400
commitb63b7672e6ba731c53198c69309c0d8e9ee7f138 (patch)
tree495b463c99b918803a4d609a25025ddddd8a7716 /PolicyEngineAPI/src/main
parent30b44d3e752d44f8f870594a05464c9e1c5158b6 (diff)
Add websocket client code.
Change-Id: I2d3939c698a0df6bec068063f08566226f65bfd3 Issue-ID: POLICY-756 Signed-off-by: pa834y <pa834y@att.com>
Diffstat (limited to 'PolicyEngineAPI/src/main')
-rw-r--r--PolicyEngineAPI/src/main/java/org/onap/policy/std/AutoClientEnd.java436
-rw-r--r--PolicyEngineAPI/src/main/java/org/onap/policy/std/ManualClientEnd.java214
-rw-r--r--PolicyEngineAPI/src/main/java/org/onap/policy/std/StdPolicyEngine.java2
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();