From 68cd0071a65ab8cba2a7f357d52c77b42a074416 Mon Sep 17 00:00:00 2001
From: nathan <nathan@dorkbox.com>
Date: Wed, 28 Mar 2018 20:18:34 +0200
Subject: [PATCH] Cleaned up local channel registration

---
 .../registration/local/RegistrationLocalHandler.java   |  2 --
 .../local/RegistrationLocalHandlerClient.java          | 10 ++--------
 .../local/RegistrationLocalHandlerServer.java          |  9 +--------
 3 files changed, 3 insertions(+), 18 deletions(-)

diff --git a/src/dorkbox/network/connection/registration/local/RegistrationLocalHandler.java b/src/dorkbox/network/connection/registration/local/RegistrationLocalHandler.java
index 4a550861..c4dad9af 100644
--- a/src/dorkbox/network/connection/registration/local/RegistrationLocalHandler.java
+++ b/src/dorkbox/network/connection/registration/local/RegistrationLocalHandler.java
@@ -43,8 +43,6 @@ class RegistrationLocalHandler extends RegistrationHandler {
                .set(metaChannel);
 
         logger.trace("New LOCAL connection.");
-
-        registrationWrapper.connection0(metaChannel, null);
     }
 
     @Override
diff --git a/src/dorkbox/network/connection/registration/local/RegistrationLocalHandlerClient.java b/src/dorkbox/network/connection/registration/local/RegistrationLocalHandlerClient.java
index d45eba9c..ee29c9f8 100644
--- a/src/dorkbox/network/connection/registration/local/RegistrationLocalHandlerClient.java
+++ b/src/dorkbox/network/connection/registration/local/RegistrationLocalHandlerClient.java
@@ -32,14 +32,6 @@ class RegistrationLocalHandlerClient extends RegistrationLocalHandler {
         super(name, registrationWrapper);
     }
 
-    /**
-     * STEP 1: Channel is first created
-     */
-//  Calls the super class to init the local channel
-//  @Override
-//  protected void initChannel(Channel channel) {
-//  }
-
     /**
      * STEP 2: Channel is now active. Start the registration process
      */
@@ -75,11 +67,13 @@ class RegistrationLocalHandlerClient extends RegistrationLocalHandler {
             // Event though a local channel is XOR with everything else, we still have to make the client clean up it's state.
             registrationWrapper.startNextProtocolRegistration();
 
+            registrationWrapper.connection0(metaChannel, null);
             ConnectionImpl connection = metaChannel.connection;
 
             // have to setup connection handler
             pipeline.addLast(CONNECTION_HANDLER, connection);
 
+
             registrationWrapper.connectionConnected0(connection);
         }
         else {
diff --git a/src/dorkbox/network/connection/registration/local/RegistrationLocalHandlerServer.java b/src/dorkbox/network/connection/registration/local/RegistrationLocalHandlerServer.java
index 1826732b..7cd75131 100644
--- a/src/dorkbox/network/connection/registration/local/RegistrationLocalHandlerServer.java
+++ b/src/dorkbox/network/connection/registration/local/RegistrationLocalHandlerServer.java
@@ -31,14 +31,6 @@ class RegistrationLocalHandlerServer extends RegistrationLocalHandler {
         super(name, registrationWrapper);
     }
 
-    /**
-     * STEP 1: Channel is first created
-     */
-//  Calls the super class to init the local channel
-//  @Override
-//  protected void initChannel(Channel channel) {
-//  }
-
     /**
      * STEP 2: Channel is now active. Start the registration process (Client starts the process)
      */
@@ -73,6 +65,7 @@ class RegistrationLocalHandlerServer extends RegistrationLocalHandler {
         MetaChannel metaChannel = channel.attr(META_CHANNEL)
                                          .getAndSet(null);
         if (metaChannel != null) {
+            registrationWrapper.connection0(metaChannel, null);
             ConnectionImpl connection = metaChannel.connection;
 
             if (connection != null) {