From 16fec4a1a088c4e60743be0a0db74256cfc90adc Mon Sep 17 00:00:00 2001
From: Leijurv <leijurv@gmail.com>
Date: Fri, 23 Nov 2018 16:09:59 -0800
Subject: [PATCH] buffered connection creation helper

---
 src/comms/java/cabaletta/comms/BufferedConnection.java  | 8 ++++++++
 src/main/java/baritone/behavior/ControllerBehavior.java | 6 +-----
 2 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/src/comms/java/cabaletta/comms/BufferedConnection.java b/src/comms/java/cabaletta/comms/BufferedConnection.java
index b5ff35c9b..6fb079984 100644
--- a/src/comms/java/cabaletta/comms/BufferedConnection.java
+++ b/src/comms/java/cabaletta/comms/BufferedConnection.java
@@ -83,4 +83,12 @@ public class BufferedConnection implements IConnection {
         }
         return msgs;
     }
+
+    public static BufferedConnection makeBuffered(IConnection conn) {
+        if (conn instanceof BufferedConnection) {
+            return (BufferedConnection) conn;
+        } else {
+            return new BufferedConnection(conn);
+        }
+    }
 }
diff --git a/src/main/java/baritone/behavior/ControllerBehavior.java b/src/main/java/baritone/behavior/ControllerBehavior.java
index bc697f8b6..1c18bf079 100644
--- a/src/main/java/baritone/behavior/ControllerBehavior.java
+++ b/src/main/java/baritone/behavior/ControllerBehavior.java
@@ -106,11 +106,7 @@ public class ControllerBehavior extends Behavior implements IMessageListener {
 
     public void connectTo(IConnection conn) {
         disconnect();
-        if (conn instanceof BufferedConnection) {
-            this.conn = (BufferedConnection) conn;
-        } else {
-            this.conn = new BufferedConnection(conn);
-        }
+        this.conn = BufferedConnection.makeBuffered(conn);
     }
 
     public void disconnect() {