0

0

SQL COPY FROM STDIN 的流式导入与事务大小控制优化

舞姬之光

舞姬之光

发布时间:2026-02-26 19:01:02

|

340人浏览过

|

来源于php中文网

原创

postgresql流式copy导入需手动分批、显式事务控制(每批begin/commit),推荐5000–50000行/批,配合禁用索引、wal调优等提升性能。

sql copy from stdin 的流式导入与事务大小控制优化

SQL 的 COPY FROM STDIN 是 PostgreSQL 中最快的数据导入方式之一,但默认行为是“全量提交”或依赖外部事务控制,容易导致内存溢出、锁表时间过长、失败后回滚代价高。要真正实现流式导入并合理控制事务大小,关键在于:**手动分批、显式事务管理、避免单一大事务,同时保持 COPY 的高效性**。

流式导入 ≠ 一次性读完再 COPY

所谓“流式”,是指不把全部数据加载进内存,而是边读边写、分段提交。PostgreSQL 的 COPY FROM STDIN 本身不提供内置分批能力,它只负责接收标准输入流中的数据行——所以流式控制必须由客户端代码(如 Python、Go、psql 脚本)实现:

  • 从文件、管道或网络流中按行或按块(如每 10,000 行)读取数据
  • 对每个批次单独发起一次 COPY FROM STDIN(在同一个数据库连接内)
  • 每个批次前后用 BEGIN / COMMIT 包裹,形成独立小事务
  • 避免使用 psql -c "COPY ... FROM STDIN" 直接传大文件,那仍是单事务

事务大小需权衡吞吐与稳定性

批次太小(如每次 100 行):事务开销占比高,整体导入变慢;批次太大(如 100 万行):单次失败需重试大量数据,内存和 WAL 压力陡增。推荐起始值为 5,000–50,000 行/批,具体根据以下因素调整:

造次
造次

Liblib打造的AI原创IP视频创作社区

下载
  • 行宽:宽表(如含 JSONB、TEXT 字段)应减小批次,防止单批内存超限
  • 服务器配置work_memmax_wal_size 影响单事务 WAL 生成与排序性能
  • 业务容忍度:若要求“至多丢失 N 行”,N 就是最大批次大小
  • 可配合 pg_stat_progress_copy 视图监控实时进度,便于动态调优

Python 示例:带事务控制的流式 COPY

使用 psycopg2 时,不要用 cursor.copy_from() 一次性传整个文件对象。正确做法是分块读取 + 手动构造 COPY 流:

with open('data.csv') as f:
    batch = []
    for line in f:
        batch.append(line)
        if len(batch) >= 10000:
            cursor.execute("BEGIN")
            with io.StringIO(''.join(batch)) as buf:
                cursor.copy_from(buf, 'my_table', sep=',', columns=cols)
            cursor.execute("COMMIT")
            batch.clear()
    # 处理余下数据
    if batch:
        cursor.execute("BEGIN")
        with io.StringIO(''.join(batch)) as buf:
            cursor.copy_from(buf, 'my_table', sep=',', columns=cols)
        cursor.execute("COMMIT")

注意:务必关闭自动提交(conn.autocommit = False),否则每个 copy_from 会隐式提交,失去事务控制能力。

其他实用优化点

  • 禁用索引与约束:导入前 ALTER TABLE ... DISABLE TRIGGER ALL,导入后再重建,提速明显
  • 调整 WAL 级别:临时设为 replicaminimal(仅限非复制环境),减少日志量
  • 使用 UNLOGGED 表:若允许崩溃丢失数据,建表时加 UNLOGGED,COPY 速度可提升 2–3 倍
  • 并行导入慎用:多连接并发 COPY 同一表会争抢锁,建议单连接分批 + 异步 I/O 提升吞吐,而非盲目加进程

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
数据分析工具有哪些
数据分析工具有哪些

数据分析工具有Excel、SQL、Python、R、Tableau、Power BI、SAS、SPSS和MATLAB等。详细介绍:1、Excel,具有强大的计算和数据处理功能;2、SQL,可以进行数据查询、过滤、排序、聚合等操作;3、Python,拥有丰富的数据分析库;4、R,拥有丰富的统计分析库和图形库;5、Tableau,提供了直观易用的用户界面等等。

1047

2023.10.12

SQL中distinct的用法
SQL中distinct的用法

SQL中distinct的语法是“SELECT DISTINCT column1, column2,...,FROM table_name;”。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

339

2023.10.27

SQL中months_between使用方法
SQL中months_between使用方法

在SQL中,MONTHS_BETWEEN 是一个常见的函数,用于计算两个日期之间的月份差。想了解更多SQL的相关内容,可以阅读本专题下面的文章。

379

2024.02.23

SQL出现5120错误解决方法
SQL出现5120错误解决方法

SQL Server错误5120是由于没有足够的权限来访问或操作指定的数据库或文件引起的。想了解更多sql错误的相关内容,可以阅读本专题下面的文章。

1864

2024.03.06

sql procedure语法错误解决方法
sql procedure语法错误解决方法

sql procedure语法错误解决办法:1、仔细检查错误消息;2、检查语法规则;3、检查括号和引号;4、检查变量和参数;5、检查关键字和函数;6、逐步调试;7、参考文档和示例。想了解更多语法错误的相关内容,可以阅读本专题下面的文章。

378

2024.03.06

oracle数据库运行sql方法
oracle数据库运行sql方法

运行sql步骤包括:打开sql plus工具并连接到数据库。在提示符下输入sql语句。按enter键运行该语句。查看结果,错误消息或退出sql plus。想了解更多oracle数据库的相关内容,可以阅读本专题下面的文章。

1436

2024.04.07

sql中where的含义
sql中where的含义

sql中where子句用于从表中过滤数据,它基于指定条件选择特定的行。想了解更多where的相关内容,可以阅读本专题下面的文章。

585

2024.04.29

sql中删除表的语句是什么
sql中删除表的语句是什么

sql中用于删除表的语句是drop table。语法为drop table table_name;该语句将永久删除指定表的表和数据。想了解更多sql的相关内容,可以阅读本专题下面的文章。

437

2024.04.29

Golang 实际项目案例:从需求到上线
Golang 实际项目案例:从需求到上线

《Golang 实际项目案例:从需求到上线》以真实业务场景为主线,完整覆盖需求分析、架构设计、模块拆分、编码实现、性能优化与部署上线全过程,强调工程规范与实践决策,帮助开发者打通从技术实现到系统交付的关键路径,提升独立完成 Go 项目的综合能力。

1

2026.02.26

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号