如何在Ruby中将数据库结果保存为CSV

Question

我从MySQL返回查询结果，我需要将它们附加到一个csv文件中。 我可以检索数据并成功显示它，但是很难将其写入csv。 到目前为止，这是我的Ruby。

#! /usr/bin/ruby

require 'mysql'
require 'csv'

# variables
file_path = '/Users/pierce/tmp/'

# need mysql queries here / use mysql2 calls
test_db = Mysql.new("localhost", "root", "", "db")

series_1_results = test_db.query("SELECT 'Impressions' AS label_imp,FORMAT(sum(impressions),',') AS impressions FROM table ")

series_1_results.each_hash do |f|
  puts "#{f['label_imp']}"",\"""#{f['impressions']}""\""
end

最后一条语句给出正确的结果：印象数“ 34,017”。 当我删除它并在下面包含CSV.open语句时，我得到以下错误。 第87行开始：csv <<

CSV.open("#{file_path}"'test_file.csv', "wb") do |csv|
    series_1_results.each_hash do |s1|
      csv << ["#{s1['label_imp']}","#{s1['impressions']}"]
    end
end

test_csv.rb:87:in `[]': can't convert String into Integer (TypeError)

我正在运行Ruby 1.9.2，感谢您的任何建议。

Answer 1

使用mysql gem（mysql2是不同的），您只能使用each_hash遍历mysql结果集一次：在消耗each_hash一行之后，mysql不再将其返回给您，因为它认为您已经完成了它。 第一次调用each_hash会消耗所有行，因此第二次进行遍历时，将没有行，也没有输出。

如果删除输出的puts版本，则应该可以使用