iktdev/MediaProcessing
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Database connect if not connected

Browse Source
This commit is contained in:
bskjon 2024-10-08 20:05:03 +02:00
parent 4f9de1220d
commit 15e719fe8e
1 changed files with 33 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
apps/pyMetadata/app.py
View File

@ -95,24 +95,41 @@ class EventsPullerThread(threading.Thread):
logger.error("Error inserting into database: %s", err) logger.error("Error inserting into database: %s", err)
return False return False
def __connect_to_datasource(self):
try:
self.connection = mysql.connector.connect(
host=events_server_address,
port=events_server_port,
database=events_server_database_name,
user=events_server_username,
password=events_server_password
)
if self.connection.is_connected():
logging.info(f"Successfully connected to {events_server_database_name} at {events_server_address}:{events_server_port}")
except Exception as e:
logging.error(f"Error while connecting to database: {e}")
self.connection = None
def run(self) -> None: def run(self) -> None:
logger.info(f"Using {events_server_address}:{events_server_port} on table: {events_server_database_name}") logger.info(f"Using {events_server_address}:{events_server_port} on table: {events_server_database_name}")
while not self.shutdown.is_set(): while not self.shutdown.is_set():
producedMessage: bool = False producedMessage: bool = False
try: while not self.shutdown.is_set():
connection = mysql.connector.connect( if self.connection is None or not self.connection.is_connected():
host=events_server_address, logging.info("Attempting to reconnect to the database...")
port=events_server_port, self.__connect_to_datasource()
database=events_server_database_name,
user=events_server_username, if self.connection is None: # Connection failed, wait and retry
password=events_server_password time.sleep(5) # Wait 5 seconds before retrying
) continue # Go back to the start of the connection loop
except: else:
logging.error(f"Unable to connect to {events_server_address}:{events_server_port}. Either the server or database: {events_server_database_name} is not present yet!") # If connection is successful, exit the connection loop and proceed
break
try: try:
rows = self.getEventsAvailable(connection=connection) rows = self.getEventsAvailable(connection=self.connection)
for row in rows: for row in rows:
if (row is not None): if (row is not None):
try: try:
@ -139,7 +156,7 @@ Producing message
============================================================================\n""" ============================================================================\n"""
logger.info(producedMessage) logger.info(producedMessage)
producedEvent = self.storeProducedEvent(connection=connection, event=producedEvent) producedEvent = self.storeProducedEvent(connection=self.connection, event=producedEvent)
except Exception as e: except Exception as e:
"""Produce failure here""" """Produce failure here"""
@ -156,14 +173,14 @@ Producing message
data=None, data=None,
eventType="EventMediaMetadataSearchPerformed" eventType="EventMediaMetadataSearchPerformed"
) )
self.storeProducedEvent(connection=connection, event=producedEvent) self.storeProducedEvent(connection=self.connection, event=producedEvent)
except mysql.connector.Error as err: except mysql.connector.Error as err:
logger.error("Database error: %s", err) logger.error("Database error: %s", err)
finally: finally:
if connection: if self.connection:
connection.close() self.connection.close()
connection = None self.connection = None
# Introduce a small sleep to reduce CPU usage # Introduce a small sleep to reduce CPU usage
time.sleep(2) time.sleep(2)
if (self.shutdown.is_set()): if (self.shutdown.is_set()):