1

我正在尝试按照 Jonaphin 在http://www.genesx.com/2012/06/import-csv-files-100x-faster-in-rails-3/上给出的示例从 csv 文件中导入数据

我正在尝试从 postgres 获取最后插入的索引,但由于某种原因,代码无法正常工作。

def process_data!(file, survey_id)
 headers = [
    "zip_code",
    "booking_date"
 ]
 ActiveRecord::Base.establish_connection

 created_at = Time.now.strftime("%Y-%m-%d %H:%M:%S")

 CSV.foreach(file, {headers: :first_row}) do |row|
  sql_keys = []
  sql_vals = []

  headers.each_with_index do |key, idx|
    val = row[idx]

    sql_keys << key
    sql_vals << ActiveRecord::Base.connection.quote(val)
  end

  sql = "
    INSERT INTO bookings (survey_id, #{sql_keys.join(', ')}, created_at, updated_at)
    VALUES (#{survey_id}, #{sql_vals.join(', ')}, '#{created_at}', '#{created_at}')
  "

  booking_id = ActiveRecord::Base.connection.insert(sql)

 end
end

booking_id现在应该有最后插入行的 id,但由于某种原因它是nil.

我尝试ActiveRecord::Base.connection.insert(sql)在控制台中运行并且它有效(即,它返回了我想要的 id)。

有人知道这里出了什么问题吗?我需要更改一些设置以允许 postgres 返回最后一个 id 吗?

4

1 回答 1

1
INSERT INTO bookings (survey_id, #{sql_keys.join(', ')}, created_at, updated_at)
VALUES (#{survey_id}, #{sql_vals.join(', ')}, '#{created_at}', '#{created_at}')
returning survey_id
于 2012-10-17T19:26:47.553 回答