From a8a906647be009ac2dfaa3f82127bd8ccff8e539 Mon Sep 17 00:00:00 2001
From: Pablo Martin <pablo.martin@lolamarket.com>
Date: Thu, 21 Jul 2022 18:31:46 +0200
Subject: [PATCH] Clean up connection once the session is finished.

---
 query_performance_gauge.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/query_performance_gauge.py b/query_performance_gauge.py
index 288ed43..702e8d0 100644
--- a/query_performance_gauge.py
+++ b/query_performance_gauge.py
@@ -5,7 +5,7 @@ from typing import Union
 import mysql.connector.connection
 import trino.dbapi
 
-from connections import get_connection
+from connections import get_connection, clean_up_connection
 
 
 def run_measuring_session(config: dict) -> None:
@@ -31,6 +31,8 @@ def run_measuring_session(config: dict) -> None:
 
     print("Finished the measuring session.")
 
+    clean_up_connection(config["connection_details"])
+
 
 class TestableQuery:
     """