cassandra-关机时的驱动程序问题

import signal
from cassandra.cluster import Cluster

def graceful_shutdown(signum, frame):
    cluster.shutdown()
    print("Cassandra driver gracefully shut down.")

signal.signal(signal.SIGTERM, graceful_shutdown)

cluster = Cluster(['127.0.0.1'])
session = cluster.connect('mykeyspace')

from cassandra.cluster import Cluster
from cassandra.query import SimpleStatement

cluster = Cluster(['127.0.0.1'])
session = cluster.connect('mykeyspace')

query = SimpleStatement("SELECT * FROM mytable", fetch_size=10)
for row in query:
    print(row)

import logging

logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

try:
    for row in query:
        logger.info(row)
except Exception as e:
    logger.error(f"Error occurred: {e}")