datajoint/connection.py: fix #896

datajoint · Mar 26, 2021 · f41f5cc · f41f5cc
1 parent 909749a
commit f41f5cc
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/datajoint/connection.py b/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.")