commit b09fdf34153c7bc8a15777da311e7b54d51d25cb
author dvaddire <dvaddire@partner.ciena.com> Tue Jul 04 11:42:53 2017 +0530
committer Simon Hunt <simon@onlab.us> Wed Jul 12 04:25:43 2017 +0000
tree 83977700f458aef2d22dcb7a8eeeefaf043aac85
parent 2e97be044913decdaa93413b8f0b18641418ce46

ONOS-6785
ServiceNotFoundException in ONOS shutdown

Change-Id: I793a6fc34a8b7e9a3b7ff0010c9adfaa3e7527c5

web/gui/src/main/java/org/onosproject/ui/impl/UiWebSocket.java

diff --git a/web/gui/src/main/java/org/onosproject/ui/impl/UiWebSocket.java b/web/gui/src/main/java/org/onosproject/ui/impl/UiWebSocket.java
index cdd84ce..ded8806 100644
--- a/web/gui/src/main/java/org/onosproject/ui/impl/UiWebSocket.java
+++ b/web/gui/src/main/java/org/onosproject/ui/impl/UiWebSocket.java
@@ -214,9 +214,12 @@
 
     @Override
     public synchronized void onClose(int closeCode, String message) {
-        tokenService().revokeToken(sessionToken);
-        log.info("Session token revoked");
-
+        try {
+            tokenService().revokeToken(sessionToken);
+            log.info("Session token revoked");
+        } catch (ServiceNotFoundException e) {
+            log.error("Unable to reference UiTokenService");
+        }
         sessionToken = null;
 
         topoSession.destroy();
@@ -410,11 +413,7 @@
     }
 
     private UiTokenService tokenService() {
-        UiTokenService service = directory.get(UiTokenService.class);
-        if (service == null) {
-            log.error("Unable to reference UiTokenService");
-        }
-        return service;
+        return directory.get(UiTokenService.class);
     }
 
     // sends the collated localization bundle data up to the client.