Postgresql 批量插入命令COPY使用

  • 背景       

在很多场景下,我们经常会遇到将某个Excel或Csv文件中的数据,插入到Postgresql。对于这个需求,我们常规的处理办法就是将文件中的数据,按照文件表头名称转换成集合对象然后插入到数据库,当然这对于数据体量不大的文件而言,很显眼没有任何问题,但是如果数据体量一旦上来,将面临如下问题:

将文件中的数据转换为集合对象,就有可能造成内存溢出;

就算内存没有溢出,那么这么大的数据,分批插入效率也能难满足需求;

针对上述问题,Postgresql提供了批量插入神器,官方文档对copy给出了如下解释,认为COPY命令针对加载大量行进行了优化;它不如INSERT灵活,但对于大数据负载的开销要小得多。也就是COPY命令比较适合大量行数据的快速插入,但是灵活度上没有Insert方便。

  • COPY命令的官方API   
导入命令如下:
       Copy table_name[(column_name…)] from 'filename' [ [WITH] (option [, ...]) ] [where condition]
导出命令如下
      COPY { table_name [ ( column_name [, ...] ) ] | ( query ) } TO  'filename'  [ [ WITH ] ( option [, ...] ) ]
其中table_name为数据库表名称,我们可以根据文件名称和文件列表头来创建一个合适的表数据。 column_name为列名称,也就是我们可指定将数据填充到指定的某些列,如果不指定,则默认将数据列从左至右依次填充到每一个列中。 FILE 文件名称,即就是文件的绝对路径。 WITH 后面的option有如下选项: FORMAT format_name 文件格式名称,主要有csv、txt、二进制文件 FREEZE [ boolean ] 不常用 DELIMITER 'delimiter_character' 分隔符,比如CSV默认分隔符逗号,txt默认是制表符,二进制文件中禁止使用此选项 NULL 'null_string' 指定表示空值的字符串。在文本格式中默认为\N,在CSV格式中默认为不加引号的空字符串 HEADER [ boolean ] 指定文件包含一个头行,其中包含文件中每个列的名称,一般用在csv文件中,值可以是true、on、1; QUOTE 'quote_character' 指定在引用数据值时使用的引号字符,默认是双引号。 ESCAPE 'escape_character' FORCE_QUOTE { ( column_name [, ...] ) | * } FORCE_NOT_NULL ( column_name [, ...] ) FORCE_NULL ( column_name [, ...] ) ENCODING 'encoding_name' 指定对文本的编码格式,默认是客户端编码格式,可通过show client_encoding查看客户端编码; WHERE condition 任何不满足此条件的行都不会被插入到表中。如果将实际行值替换为任何变量引用时该行返回true,则该行满足条件。

比如要将一个test.csv文件导入到数据库,首先根据CSV的表头信息创建数据库表test,字段包括gid、zipcode、telephone、name、address创建成功后,

使用copy导入数据:

 

Copy test from 'D:/test.csv' with csv header DELIMITER ','

上面这个命令是csv的header字段和数据表的字段一一对应的操作,如果需要再导入的时候指定导入的字段,则可以按照如下方式指定对应的字段

Copy test(zipcode,telephone,name,address) from 'D:/test.csv' with csv header DELIMITER ','  

注意: 小编在使用导入命令时,发现windows操作系统下,文件路径中如果有中文,无法导入,系统会报could not open file  for reading: Invalid argument这个错误。

上面是将csv数据导入到数据库,此时如果我们需要将数据库中的数据,导出到csv文件,即可使用如下的方式:

Copy test to 'D:/test_new.csv' csv header DELIMITER ',' ENCODING 'UTF8'

对于导出数据中有中文的,特别说明下,如果我们导出的csv中的中文,试用excel查看是乱码的,但是试用其它文本编辑器(比如nodepad++)是好的,这时候只需要在文本编辑器中,将文件编码格式改为UTF-8-BOM就行,这应该是微软自己对文件编码格式的判断问题。

  • 对于上面官方给出的说明文档,有几点疑问:

官方文档的这个说明不知道是我理解错了,还是本身有问题,我开始按照它的说明文档进项操作,比如我按照文档如果加上上面的这个format就会报语法错误,去掉则没有问题:

Copy test to 'D:/test_new.csv' format csv header DELIMITER ',' ENCODING 'UTF8'

再比如,官方文档指出header后面可以跟true,on,1表示开启,但实际加上就会报错,比如如果你写成下面的方式就会报语法错误:

Copy test to 'D:/test_new.csv' csv header true DELIMITER ',' ENCODING 'UTF8'

而其它的key,比如DELIMITER,ENCODING就必须输入前面的key,所以不知道是自己哪里没理解还是本身这个就有问题,麻烦了解的朋友给说明下。

 

posted @ 2023-08-26 21:40  开放GIS  阅读(1564)  评论(0编辑  收藏  举报