当前位置：首页 > article >正文

自动化执行 SQL 脚本解决方案

article 2025/1/9 11:35:49

自动化执行 SQL 脚本解决方案

手动登录多个库和数据源执行 SQL 脚本非常繁琐，以下是解决此问题的几种自动化方案。

目标：让 SQL 脚本自动化执行

方案 1：使用 Python 脚本连接并执行 SQL

Python 可以通过数据库连接库（如 pyodbc 或 pymysql）来自动化连接和执行 SQL 脚本。

Python 自动执行 SQL 脚本

import pyodbc
import time

# 数据库连接配置
DATABASES = [
    {"name": "Database1", "dsn": "DSN1", "username": "user1", "password": "pass1"},
    {"name": "Database2", "dsn": "DSN2", "username": "user2", "password": "pass2"},
    # 按需添加更多数据库
]

# SQL 脚本文件路径
SQL_FILE_PATH = "upgrade_script.sql"

def execute_sql_script(connection, sql_file_path):
    with open(sql_file_path, "r") as file:
        sql_script = file.read()
        cursor = connection.cursor()
        for statement in sql_script.split(";"):  # 按分号拆分 SQL 语句
            if statement.strip():
                print(f"Executing: {statement.strip()[:50]}...")
                cursor.execute(statement)
        connection.commit()

def main():
    for db in DATABASES:
        try:
            print(f"Connecting to {db['name']}...")
            connection = pyodbc.connect(
                f"DSN={db['dsn']};UID={db['username']};PWD={db['password']}"
            )
            execute_sql_script(connection, SQL_FILE_PATH)
            print(f"Finished executing script on {db['name']}.")
        except Exception as e:
            print(f"Error on {db['name']}: {e}")
        finally:
            if connection:
                connection.close()
    print("All databases updated successfully!")

if __name__ == "__main__":
    start_time = time.time()
    main()
    print(f"Execution completed in {time.time() - start_time:.2f} seconds.")

配置说明

数据库连接：
- DSN 是数据库的 ODBC 数据源名称。如果未配置 DSN，可以直接用 host 和 port 替代。
- 替换 username 和 password 为对应的凭据。
SQL 脚本路径：
- 将升级的 SQL 脚本保存为 upgrade_script.sql 文件。
数据库类型支持：
- 针对不同数据库（如 MySQL、PostgreSQL、SQL Server），使用合适的 Python 库：
  - MySQL: 使用 pymysql 或 mysql-connector-python。
  - PostgreSQL: 使用 psycopg2。
  - SQL Server: 使用 pyodbc。

方案 2：批量执行 SQL 脚本（Shell 或批处理）

如果 Python 不是首选，可以使用 Shell 或批处理文件实现。

示例：使用 `mysql` 命令行工具

#!/bin/bash

# 数据库配置
DATABASES=(
  "Database1 -uuser1 -ppass1"
  "Database2 -uuser2 -ppass2"
)

# SQL 脚本路径
SQL_FILE="upgrade_script.sql"

# 遍历所有数据库并执行脚本
for db in "${DATABASES[@]}"; do
  echo "Executing script on $db..."
  mysql $db < $SQL_FILE
done

echo "All databases updated successfully!"