From 6582caa5218435a860c6ed17a5d91e5c706bd984 Mon Sep 17 00:00:00 2001
From: nathan <nathan@dorkbox.com>
Date: Wed, 4 Apr 2018 15:02:17 +0200
Subject: [PATCH] Added better close listener for UDP

---
 .../network/connection}/DatagramCloseListener.java       | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)
 rename src/{io/netty/bootstrap => dorkbox/network/connection}/DatagramCloseListener.java (81%)

diff --git a/src/io/netty/bootstrap/DatagramCloseListener.java b/src/dorkbox/network/connection/DatagramCloseListener.java
similarity index 81%
rename from src/io/netty/bootstrap/DatagramCloseListener.java
rename to src/dorkbox/network/connection/DatagramCloseListener.java
index 060a2de3..92e80a7f 100644
--- a/src/io/netty/bootstrap/DatagramCloseListener.java
+++ b/src/dorkbox/network/connection/DatagramCloseListener.java
@@ -13,10 +13,9 @@
  *  See the License for the specific language governing permissions and
  *  limitations under the License.
  */
-package io.netty.bootstrap;
+package dorkbox.network.connection;
 
-import dorkbox.network.connection.Connection;
-import dorkbox.network.connection.Listener;
+import io.netty.bootstrap.DatagramCloseMessage;
 
 /**
  * Uses a hint to close remote UDP connections
@@ -26,6 +25,8 @@ class DatagramCloseListener implements Listener.OnMessageReceived<Connection, Da
     @Override
     public
     void received(final Connection connection, final DatagramCloseMessage message) {
-        connection.close();
+        if (connection instanceof ConnectionImpl) {
+            ((ConnectionImpl) connection).forceClose();
+        }
     }
 }