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():
"""
Create a ClickHouse client with retry logic but without session management
Create a ClickHouse client without session management
"""
clickhouse_password = os.getenv("CLICKHOUSE_PASSWORD")
if not clickhouse_password:
@ -34,13 +34,18 @@ def create_client():
password=clickhouse_password,
secure=True,
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
try:
client.query('SELECT 1')
logger.info(f"Successfully established connection")
logger.debug(f"Successfully established connection") # Changed to debug level
return client
except Exception as e:
logger.error(f"Connection test failed: {str(e)}")