Class: ActiveRecord::Base

Inherits:
Object
  • Object
show all
Defined in:
lib/postgres-copy/active_record.rb

Class Method Summary collapse

Class Method Details

.pg_copy_from(path_or_io, options = {}) ⇒ Object 



23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
 
# File 'lib/postgres-copy/active_record.rb', line 23

def self.pg_copy_from path_or_io, options = {}
  options = {:delimiter => "\t"}.merge(options)
  io = path_or_io.instance_of?(String) ? File.open(path_or_io, 'r') : path_or_io
  # The first line should be always the HEADER.
  line = io.gets
  columns_list = options[:columns] || line.strip.split(options[:delimiter])
  columns_list = columns_list.map{|c| options[:map][c.to_s] } if options[:map]
  connection.execute %{COPY #{quoted_table_name} ("#{columns_list.join('","')}") FROM STDIN WITH DELIMITER '#{options[:delimiter]}' CSV}
  while line = io.gets do
    next if line.strip.size == 0
    if block_given?
      row = line.strip.split(options[:delimiter])
      yield(row)
      line = row.join(options[:delimiter]) + "\n"
    end
    connection.raw_connection.put_copy_data line
  end
  connection.raw_connection.put_copy_end
end

.pg_copy_to(path = nil) ⇒ Object 



3
4
5
6
7
8
9
10
11
12
13
14
 
# File 'lib/postgres-copy/active_record.rb', line 3

def self.pg_copy_to path = nil
  if path
    raise "You have to choose between exporting to a file or receiving the lines inside a block" if block_given?
    connection.execute "COPY (#{self.scoped.to_sql}) TO #{sanitize(path)} WITH DELIMITER '\t' CSV HEADER"
  else
    connection.execute "COPY (#{self.scoped.to_sql}) TO STDOUT WITH DELIMITER '\t' CSV HEADER"
    while line = connection.raw_connection.get_copy_data do
      yield(line) if block_given?
    end
  end
  return self
end

.pg_copy_to_string ⇒ Object 



16
17
18
19
20
21
 
# File 'lib/postgres-copy/active_record.rb', line 16

def self.pg_copy_to_string
  # It would be cool to work like an Enumerable
  data = ''
  self.pg_copy_to{|l| data += l }
  data
end