海量数据迁移促销

import psycopg2

def migrate_data(source_conn_str, dest_conn_str, table_name):
    # 连接到源数据库
    source_conn = psycopg2.connect(source_conn_str)
    source_cursor = source_conn.cursor()
    
    # 连接到目标数据库
    dest_conn = psycopg2.connect(dest_conn_str)
    dest_cursor = dest_conn.cursor()
    
    try:
        # 从源表中选择数据
        source_cursor.execute(f"SELECT * FROM {table_name}")
        rows = source_cursor.fetchall()
        
        # 将数据插入目标表
        for row in rows:
            dest_cursor.execute(f"INSERT INTO {table_name} VALUES %s", (row,))
        
        # 提交事务
        dest_conn.commit()
        print("数据迁移成功！")
    except Exception as e:
        print(f"数据迁移失败: {e}")
        dest_conn.rollback()
    finally:
        source_cursor.close()
        source_conn.close()
        dest_cursor.close()
        dest_conn.close()

# 示例调用
source_conn_str = "dbname=source_db user=user password=password host=localhost port=5432"
dest_conn_str = "dbname=dest_db user=user password=password host=localhost port=5432"
migrate_data(source_conn_str, dest_conn_str, "example_table")