implementing onIOError method for handling IOException errors

Author: Kristijan Sedlak

README.markdown

@@ -47,9 +47,9 @@ Writing your own WebSocket Client
 
 The `net.tootallnate.websocket.WebSocketClient` abstract class can connect to
 valid WebSocket servers. The constructor expects a valid `ws://` URI to
-connect to. Important events `onOpen`, `onClose`, and `onMessage` get fired
-throughout the life of the WebSocketClient, and must be implemented in **your**
-subclass.
+connect to. Important events `onOpen`, `onClose`, `onMessage` and onIOError get 
+fired throughout the life of the WebSocketClient, and must be implemented in 
+**your** subclass.
 
 License
 -------

src/net/tootallnate/websocket/WebSocketClient.java

@@ -150,7 +150,7 @@ private boolean tryToConnect(InetSocketAddress remote) {
       client.register(selector, SelectionKey.OP_CONNECT);
 
     } catch (IOException ex) {
-      ex.printStackTrace();
+    	onIOError(ex);
       return false;
     }
 
@@ -180,7 +180,7 @@ public void run() {
           }
         }
       } catch (IOException ex) {
-        ex.printStackTrace();
+    	  onIOError(ex);
       } catch (NoSuchAlgorithmException ex) {
         ex.printStackTrace();
       }
@@ -348,6 +348,15 @@ public void onClose(WebSocket conn) {
     onClose();
   }
 
+  /**
+   * Triggered on any IOException error. This method should be overridden for custom 
+   * implementation of error handling (e.g. when network is not available). 
+   * @param ex
+   */
+   public void onIOError(IOException ex) {
+ 	  ex.printStackTrace();
+   }
+
   // ABTRACT METHODS /////////////////////////////////////////////////////////
   public abstract void onMessage(String message);
   public abstract void onOpen();

src/net/tootallnate/websocket/WebSocketServer.java

@@ -196,7 +196,7 @@ public void run() {
       selector = Selector.open();
       server.register(selector, server.validOps());
     } catch (IOException ex) {
-      ex.printStackTrace();
+    	onIOError(ex);
       return;
     }
 
@@ -250,7 +250,7 @@ public void run() {
           }
         }
       } catch (IOException ex) {
-        ex.printStackTrace();
+    	  onIOError(ex);
       } catch (RuntimeException ex) {
         ex.printStackTrace();
       } catch (NoSuchAlgorithmException ex) {
@@ -415,6 +415,15 @@ public void onClose(WebSocket conn) {
     }
   }
 
+  /**
+   * Triggered on any IOException error. This method should be overridden for custom 
+   * implementation of error handling (e.g. when network is not available). 
+   * @param ex
+   */
+   public void onIOError(IOException ex) {
+ 	  ex.printStackTrace();
+   }
+
   private byte[] getPart(String key) {
     long keyNumber = Long.parseLong(key.replaceAll("[^0-9]",""));
     long keySpace = key.split("\u0020").length - 1;