我希望能够将CSV文件中的数据导入到已有的数据库表中。我不想保存CSV文件，只需要将其中的数据提取出来放到已有的表格中。我使用的是Ruby 1.9.2和Rails 3。
以下是我的表格：
create_table "mouldings", :force => true do |t|
t.string "suppliers_code"
t.datetime "created_at"
t.datetime "updated_at"
t.string "name"
t.integer "supplier_id"
t.decimal "length", :precision => 3, :scale => 2
t.decimal "cost", :precision => 4, :scale => 2
t.integer "width"
t.integer "depth"
end
你能给我一些代码，展示最佳实现方式吗？感谢。
require 'csv'
csv_text = File.read('...')
csv = CSV.parse(csv_text, :headers => true)
csv.each do |row|
Moulding.create!(row.to_hash)
end
以下是简化版本的yfeldblum答案，它更加简单易懂，同时也适用于大文件：
require 'csv'
CSV.foreach(filename, headers: true) do |row|
Moulding.create!(row.to_hash)
end
不需要使用with_indifferent_access或symbolize_keys，也不需要先将文件读入字符串。
它不会一次性将整个文件保留在内存中，而是逐行读取并为每一行创建一个Moulding对象。
smarter_csv宝石是专门为这种用例而创建的：从CSV文件读取数据并快速创建数据库条目。
require 'smarter_csv'
options = {}
SmarterCSV.process('input_file.csv', options) do |chunk|
chunk.each do |data_hash|
Moulding.create!( data_hash )
end
end
chunk_size一次读取N个csv行，然后在内部循环中使用Resque生成作业来创建新记录，而不是立即创建它们-这样你就可以将生成条目的负载分散到多个工作者中。./ 前缀。 - Tilo您可以尝试使用Upsert：
require 'upsert' # add this to your Gemfile
require 'csv'
u = Upsert.new Moulding.connection, Moulding.table_name
CSV.foreach(file, headers: true) do |row|
selector = { name: row['name'] } # this treats "name" as the primary key and prevents the creation of duplicates by name
setter = row.to_hash
u.row selector, setter
end
如果您希望这样做，您还可以考虑从表中删除自动增量主键，并将主键设置为name。或者，如果有一些属性的组合形成了一个主键，请使用该组合作为选择器。不需要索引，它只会使它更快。
最好将与数据库相关的过程放在 transaction 块中。下面的代码片段是将一组语言种子添加到 Language 模型中的完整过程。
require 'csv'
namespace :lan do
desc 'Seed initial languages data with language & code'
task init_data: :environment do
puts '>>> Initializing Languages Data Table'
ActiveRecord::Base.transaction do
csv_path = File.expand_path('languages.csv', File.dirname(__FILE__))
csv_str = File.read(csv_path)
csv = CSV.new(csv_str).to_a
csv.each do |lan_set|
lan_code = lan_set[0]
lan_str = lan_set[1]
Language.create!(language: lan_str, code: lan_code)
print '.'
end
end
puts ''
puts '>>> Languages Database Table Initialization Completed'
end
end
languages.csv 文件的一部分：aa,Afar
ab,Abkhazian
af,Afrikaans
ak,Akan
am,Amharic
ar,Arabic
as,Assamese
ay,Aymara
az,Azerbaijani
ba,Bashkir
...
desc "Imports a CSV file into an ActiveRecord table"
task :csv_model_import, [:filename, :model] => [:environment] do |task,args|
lines = File.new(args[:filename], "r:ISO-8859-1").readlines
header = lines.shift.strip
keys = header.split(',')
lines.each do |line|
values = line.strip.split(',')
attributes = Hash[keys.zip values]
Module.const_get(args[:model]).create(attributes)
end
end
rake csv_model_import[file.csv,模型名称]。我知道这是一个老问题，但它仍然在谷歌的前10个链接中。
逐行保存不是很有效，因为它会导致循环中的数据库调用，最好避免这种情况，特别是当你需要插入大量数据时。
使用批量插入更好（而且速度显著更快）。
INSERT INTO `mouldings` (suppliers_code, name, cost)
VALUES
('s1', 'supplier1', 1.111),
('s2', 'supplier2', '2.222')
Model.connection.execute(RAW SQL STRING)（不推荐），或者使用gemactiverecord-import（它于2010年8月11日首次发布）。在这种情况下，只需将数据放入数组rows中，并调用Model.import rows。
以下模块可在任何模型上进行扩展，它将根据 CSV 中定义的列标题导入数据。
注意：
在 models/concerns 文件夹中创建一个名为 "csv_importer.rb" 的文件。
module CsvImporter
extend ActiveSupport::Concern
require 'csv'
def convert_csv_to_book_attributes(csv_path)
csv_rows = CSV.open(csv_path).each.to_a.compact
columns = csv_rows[0].map(&:strip).map(&:to_sym)
csv_rows.shift
return columns, csv_rows
end
def import_by_csv(csv_path)
columns, attributes_array = convert_csv_to_book_attributes(csv_path)
message = ""
begin
self.import columns, attributes_array, validate: false
message = "Import Successful."
rescue => e
message = e.message
end
return message
end
end
将extend CsvImporter添加到您想要扩展此功能的任何模型中。
在您的控制器中，您可以拥有以下操作来利用此功能：
def import_file
model_name = params[:table_name].singularize.camelize.constantize
csv = params[:file].path
@message = model_name.import_by_csv(csv)
end
class Moulding < ActiveRecord::Base
acts_as_importable
end
然后，您现在可以使用：
Moulding.import!(file: File.open(PATH_TO_FILE))
请确保您的表头与表格列名匹配