ruby - 如何使用 Ruby 的 sqlite3 gem 将 CSV 文件导入 SQLite 数据库

Question

这是我的 .CSV 文件的示例:

column1,column2
data,moreData
foo,bar
cats,dogs

使用 Ruby 和 sqlite3 gem，有没有办法制作 SQLite 数据库并将 .CSV 数据导入其中？

Answer 1

使用 csv 库解析 CSV，然后像往常一样插入它。

require "sqlite3"
require 'csv'

db = SQLite3::Database.new ":memory:"

# Create a database
rows = db.execute <<-SQL
  create table users (
    name varchar(30),
    age int
  );
SQL

csv = <<CSV
name,age
ben,12
sally,39
CSV

CSV.parse(csv, headers: true) do |row|
  db.execute "insert into users values ( ?, ? )", row.fields # equivalent to: [row['name'], row['age']]
end

db.execute( "select * from users" ) # => [["ben", 12], ["sally", 39]]