Add 2006 to Mysql expected error codes

danielduduta · danielduduta · commit 0929a9e15adf · 2014-05-20T08:12:39.000Z
diff --git a/pymysqlreplication/binlogstream.py b/pymysqlreplication/binlogstream.py
@@ -11,6 +11,10 @@
 from .constants.BINLOG import TABLE_MAP_EVENT, ROTATE_EVENT
 from .event import NotImplementedEvent
 
+
+MYSQL_EXPECTED_ERROR_CODES = [2013, 2006] #2013 Connection Lost
+                                          #2006 MySQL server has gone away
+
 class BinLogStreamReader(object):
     """Connect to replication stream and read event
     """
@@ -136,8 +140,7 @@ def fetchone(self):
                     pkt = self._stream_connection._read_packet()
             except pymysql.OperationalError as error:
                 code, message = error.args
-                # 2013: Connection Lost
-                if code == 2013:
+                if code in MYSQL_EXPECTED_ERROR_CODES:
                     self.__connected_stream = False
                     continue
 
@@ -196,8 +199,7 @@ def __get_table_information(self, schema, table):
                 return cur.fetchall()
             except pymysql.OperationalError as error:
                 code, message = error.args
-                # 2013: Connection Lost
-                if code == 2013:
+                if code in MYSQL_EXPECTED_ERROR_CODES:
                     self.__connected_ctl = False
                     continue
                 else:
