oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix incorrect removal of last_query_response from database client

Browse Source
This commit is contained in:
Marek Wolan
2024-08-20 15:46:54 +00:00
parent bceedbf72d
commit 71ec71e400
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/primaite/simulator/system/applications/database_client.py
View File

@@ -75,6 +75,8 @@ class DatabaseClient(Application, identifier="DatabaseClient"):
server_password: Optional[str] = None server_password: Optional[str] = None
_query_success_tracker: Dict[str, bool] = {} _query_success_tracker: Dict[str, bool] = {}
"""Keep track of connections that were established or verified during this step. Used for rewards.""" """Keep track of connections that were established or verified during this step. Used for rewards."""
last_query_response: Optional[Dict] = None
"""Keep track of the latest query response. Used to determine rewards."""
_server_connection_id: Optional[str] = None _server_connection_id: Optional[str] = None
"""Connection ID to the Database Server.""" """Connection ID to the Database Server."""
client_connections: Dict[str, DatabaseClientConnection] = {} client_connections: Dict[str, DatabaseClientConnection] = {}
@@ -381,6 +383,9 @@ class DatabaseClient(Application, identifier="DatabaseClient"):
if not self.native_connection: if not self.native_connection:
return False return False
# reset last query response
self.last_query_response = None
uuid = str(uuid4()) uuid = str(uuid4())
self._query_success_tracker[uuid] = False self._query_success_tracker[uuid] = False
return self.native_connection.query(sql) return self.native_connection.query(sql)
@@ -404,6 +409,7 @@ class DatabaseClient(Application, identifier="DatabaseClient"):
connection_id=connection_id, connection_request_id=payload["connection_request_id"] connection_id=connection_id, connection_request_id=payload["connection_request_id"]
) )
elif payload["type"] == "sql": elif payload["type"] == "sql":
self.last_query_response = payload
query_id = payload.get("uuid") query_id = payload.get("uuid")
status_code = payload.get("status_code") status_code = payload.get("status_code")
self._query_success_tracker[query_id] = status_code == 200 self._query_success_tracker[query_id] = status_code == 200