diff --git a/deployment/helm/datamate/values.yaml b/deployment/helm/datamate/values.yaml index 55bb1b27..921ae3ad 100644 --- a/deployment/helm/datamate/values.yaml +++ b/deployment/helm/datamate/values.yaml @@ -163,18 +163,18 @@ runtime: - --port - "8081" env: &runtimeEnv - - name: MYSQL_HOST + - name: PG_HOST value: "datamate-database" - - name: MYSQL_PORT - value: "3306" - - name: MYSQL_USER - value: "root" - - name: MYSQL_PASSWORD + - name: PG_PORT + value: "5432" + - name: PG_USER + value: "postgres" + - name: PG_PASSWORD valueFrom: secretKeyRef: name: datamate-conf key: DB_PASSWORD - - name: MYSQL_DATABASE + - name: PG_DATABASE value: "datamate" volumes: - *datasetVolume diff --git a/runtime/python-executor/datamate/sql_manager/sql_manager.py b/runtime/python-executor/datamate/sql_manager/sql_manager.py index 14deae64..087264e5 100644 --- a/runtime/python-executor/datamate/sql_manager/sql_manager.py +++ b/runtime/python-executor/datamate/sql_manager/sql_manager.py @@ -23,7 +23,7 @@ def create_connect(max_retries=5, base_delay=1): drivername="postgresql+psycopg2", # 核心修改:使用 pg 驱动 username=os.getenv("PG_USER", "postgres"), # 建议修改环境变量名 password=os.getenv("PG_PASSWORD", "password"), - host=os.getenv("PG_HOST", "postgres"), + host=os.getenv("PG_HOST", "datamate-database"), port=int(os.getenv("PG_PORT", 5432)), # 修改默认端口为 5432 database=os.getenv("PG_DATABASE", "datamate"), )