Added proper handling of SocketException's

1 files changed, 3 insertions, 7 deletions

diff --git a/src/main/java/com/lumaserv/bgp/BGPSession.java b/src/main/java/com/lumaserv/bgp/BGPSession.java
index 669010b..3ffadd8 100644
--- a/src/main/java/com/lumaserv/bgp/BGPSession.java
+++ b/src/main/java/com/lumaserv/bgp/BGPSession.java
@@ -52,15 +52,11 @@ public class BGPSession implements Runnable {
     public void run() {
         try {
             while (!closed) {
-                try {
-                    handle(BGPPacket.read(inputStream));
-                } catch (SocketException ignored) {
-                    closed = true;
-                    configuration.getListener().onClose(this);
-                }
+                handle(BGPPacket.read(inputStream));
             }
         } catch (IOException ex) {
-            ex.printStackTrace();
+            closed = true;
+            configuration.getListener().onClose(this);
         }
     }