Bobby/stock_system
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: Disable session checking and optimize ClickHouse client config

Browse Source
This commit is contained in:
Bobby (aider) 2025-02-08 20:27:33 -08:00
parent 8d7ecb4f39
commit 4dcb9c969e
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/db/db_connection.py
View File

@ -14,7 +14,7 @@ logger = logging.getLogger(__name__)
def create_client(): def create_client():
""" """
Create a ClickHouse client with retry logic but without session management Create a ClickHouse client without session management
""" """
clickhouse_password = os.getenv("CLICKHOUSE_PASSWORD") clickhouse_password = os.getenv("CLICKHOUSE_PASSWORD")
if not clickhouse_password: if not clickhouse_password:
@ -34,13 +34,18 @@ def create_client():
password=clickhouse_password, password=clickhouse_password,
secure=True, secure=True,
connect_timeout=10, connect_timeout=10,
send_receive_timeout=300 send_receive_timeout=300,
query_limit=0, # No query limit
compress=True, # Enable compression
settings={
'session_check': 0 # Disable session checking
}
) )
# Test the connection with a simple query # Test the connection with a simple query
try: try:
client.query('SELECT 1') client.query('SELECT 1')
logger.info(f"Successfully established connection") logger.debug(f"Successfully established connection") # Changed to debug level
return client return client
except Exception as e: except Exception as e:
logger.error(f"Connection test failed: {str(e)}") logger.error(f"Connection test failed: {str(e)}")