implementing onIOError method for handling IOException errors

Kristijan Sedlak · Kristijan Sedlak · commit daabf822ae85 · 2011-06-29T11:26:17.000+02:00
diff --git a/README.markdown b/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
 -------
diff --git a/src/net/tootallnate/websocket/WebSocketClient.java b/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();
diff --git a/src/net/tootallnate/websocket/WebSocketServer.java b/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;

Original file line number	Diff line number	Diff line change
`@@ -150,7 +150,7 @@ private boolean tryToConnect(InetSocketAddress remote) {`
`150`	`150`	`client.register(selector, SelectionKey.OP_CONNECT);`
`151`	`151`
`152`	`152`	`} catch (IOException ex) {`
`153`		`- ex.printStackTrace();`
	`153`	`+ onIOError(ex);`
`154`	`154`	`return false;`
`155`	`155`	`}`
`156`	`156`
`@@ -180,7 +180,7 @@ public void run() {`
`180`	`180`	`}`
`181`	`181`	`}`
`182`	`182`	`} catch (IOException ex) {`
`183`		`- ex.printStackTrace();`
	`183`	`+ onIOError(ex);`
`184`	`184`	`} catch (NoSuchAlgorithmException ex) {`
`185`	`185`	`ex.printStackTrace();`
`186`	`186`	`}`
`@@ -348,6 +348,15 @@ public void onClose(WebSocket conn) {`
`348`	`348`	`onClose();`
`349`	`349`	`}`
`350`	`350`
	`351`	`+ /**`
	`352`	`+ * Triggered on any IOException error. This method should be overridden for custom`
	`353`	`+ * implementation of error handling (e.g. when network is not available).`
	`354`	`+ * @param ex`
	`355`	`+ */`
	`356`	`+ public void onIOError(IOException ex) {`
	`357`	`+ ex.printStackTrace();`
	`358`	`+ }`
	`359`	`+`
`351`	`360`	`// ABTRACT METHODS /////////////////////////////////////////////////////////`
`352`	`361`	`public abstract void onMessage(String message);`
`353`	`362`	`public abstract void onOpen();`