fix: Properly shutdown control connection and scheduler in CassandraO…

…nlineStore
ExpediaGroup · Jan 22, 2025 · c887250 · c887250
1 parent 916591b
commit c887250
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/...python/feast/infra/online_stores/contrib/cassandra_online_store/cassandra_online_store.py b/...python/feast/infra/online_stores/contrib/cassandra_online_store/cassandra_online_store.py
@@ -455,8 +455,11 @@ def __del__(self):
                     print(
                         f"{self._cluster.client_id}: Del Connection ID: {connection.get_connections()}"
                     )
-                # self._cluster.control_connection.shutdown()
-                self._cluster.shutdown()
+                self._cluster.control_connection.shutdown()
+                if not self._cluster.scheduler.is_shutdown:
+                    print(f"{self._cluster.client_id}: Shutting down scheduler")
+                    self._cluster.scheduler.shutdown()
+                # self._cluster.shutdown()
 
             else:
                 print(f"{self._cluster.client_id}: Cluster is not active")