3

我想为用户提供在我的 Ruby on Rails 3.2 应用程序中上传文件的选项，并将数据放入数据库中。我想使用该COPY FROM命令，因为它比插入 ruby​​ 对象更快。

如果我做

User.connection.execute("COPY users (name, taxon_id, created_at, updated_at) FROM 'a.txt'")

我明白了

ActiveRecord::StatementInvalid: PG::Error: ERROR:  must be superuser to COPY to or from a file
HINT:  Anyone can COPY to stdout or from stdin. psql's \copy command also works for anyone.

但是，\copy也不适用于数据库连接。如何在 Heroku 上使用 PostgreSQL 从 Rails 中的平面文件轻松上传数据？您可以获得超级用户权限吗？

4

2 回答 2

4

感谢@PhilipHallstrom 的链接，我COPY FROM STDIN这样使用：

rc = User.connection.raw_connection
rc.exec("COPY users (name, taxon_id, updated_at, created_at) FROM STDIN")
begin
  until rc.put_copy_data( data )
    $stderr.puts "  waiting for connection to be writable..."
    sleep 0.1
  end
rescue Errno => err
  @errmsg = @errmsg + "%s while reading copy data: %s" % [ err.class.name, err.message ]
  error = true
else
  rc.put_copy_end
  while res = rc.get_result
    if (res.result_status != 1)
      error = true
      @errmsg = @errmsg + "Result of COPY is: %s" % [ res.res_status(res.result_status) ]
    end
  end
end
于 2013-01-25T20:27:44.193 回答
1

您可以使用 gem https://github.com/diogob/postgres-copy 假设data是一个 IO 对象。

User.copy_from data, columns: [:name, :taxon_id, :updated_at, :created_at]

或者直接使用上传文件的路径：

User.copy_from "/tmp/uploaded_file.csv", columns: [:name, :taxon_id, :updated_at, :created_at]
于 2014-06-18T02:17:58.567 回答