datajoint/connection.py: fix datajoint#896

Author: Chris Turner

datajoint/connection.py

@@ -305,6 +305,10 @@ def query(self, query, args=(), *, as_dict=False, suppress_warnings=True, reconn
                 raise
             warnings.warn("MySQL server has gone away. Reconnecting to the server.")
             connect_host_hook(self)
+            if self.is_connected:
+                logger.info("Connected {user}@{host}:{port}".format(**self.conn_info))
+                self.connection_id = self.query('SELECT connection_id()').fetchone()[0]
+
             if self._in_transaction:
                 self.cancel_transaction()
                 raise errors.LostConnectionError("Connection was lost during a transaction.")