Merge pull request #38 from elw00d/master

Introduce overridable setupConnection() method

Author: osheroff

src/main/java/com/github/shyiko/mysql/binlog/BinaryLogClient.java

@@ -554,14 +554,7 @@ public void connect() throws IOException, IllegalStateException {
                     }
                     binlogPosition = 4;
                 }
-                ChecksumType checksumType = fetchBinlogChecksum();
-                if (checksumType != ChecksumType.NONE) {
-                    confirmSupportOfChecksum(checksumType);
-                }
-                setMasterServerId();
-                if (heartbeatInterval > 0) {
-                    enableHeartbeat();
-                }
+                setupConnection();
                 gtid = null;
                 tx = false;
                 requestBinaryLogStream();
@@ -614,6 +607,20 @@ public void connect() throws IOException, IllegalStateException {
         }
     }
 
+    /**
+     * Apply additional options for connection before requesting binlog stream.
+     */
+    protected void setupConnection() throws IOException {
+        ChecksumType checksumType = fetchBinlogChecksum();
+        if (checksumType != ChecksumType.NONE) {
+            confirmSupportOfChecksum(checksumType);
+        }
+        setMasterServerId();
+        if (heartbeatInterval > 0) {
+            enableHeartbeat();
+        }
+    }
+
     private PacketChannel openChannel() throws IOException {
         Socket socket = socketFactory != null ? socketFactory.createSocket() : new Socket();
         socket.connect(new InetSocketAddress(hostname, port), (int) connectTimeout);