海量数据迁移双十二优惠活动

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}")
        data = source_cursor.fetchall()
        
        # 将数据插入目标表
        dest_cursor.executemany(f"INSERT INTO {table_name} VALUES (%s, %s, ...)", data)
        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"
table_name = "example_table"

migrate_data(source_conn_str, dest_conn_str, table_name)