jsuchal-activerecord-fast-import 0.1.1 → 0.1.2

data/VERSION

@@ -1 +1 @@
-0.1.1
+0.1.2

data/activerecord-fast-import.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = %q{activerecord-fast-import}
-  s.version = "0.1.1"
+  s.version = "0.1.2"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Jan Suchal"]
-  s.date = %q{2009-08-19}
+  s.date = %q{2009-08-21}
   s.description = %q{Native MySQL additions to ActiveRecord, like LOAD DATA INFILE, ENABLE/DISABLE KEYS, TRUNCATE TABLE.}
   s.email = %q{johno@jsmf.net}
   s.extra_rdoc_files = [

data/lib/activerecord-fast-import.rb

@@ -1,53 +1,54 @@
-module ActiveRecord #:nodoc:
-  class Base
-    # Deletes all rows in table very fast, but without calling +destroy+ method
-    # nor any hooks.
-    def self.truncate_table
-      connection.execute("TRUNCATE TABLE #{quoted_table_name}")
-    end
-
-    # Disables key updates for model table
-    def self.disable_keys
-      connection.execute("ALTER TABLE #{quoted_table_name} DISABLE KEYS")
-    end
-
-    # Enables key updates for model table
-    def self.enable_keys
-      connection.execute("ALTER TABLE #{quoted_table_name} ENABLE KEYS")
-    end
-
-    # Loads data from file using MySQL native LOAD DATA INFILE query, disabling
-    # key updates for even faster import speed
-    #
-    # ==== Parameters
-    # * +file+ the file to load
-    # * +options+ (see <tt>load_data_infile</tt>)
-    def self.fast_import(file, options = {})
-      disable_keys
-      load_data_infile(file, options)
-      enable_keys
-    end
-
-    # Loads data from file using MySQL native LOAD DATA INFILE query
-    #
-    # ==== Parameters
-    # * +file+ the file to import
-    # * +options+
-    def self.load_data_infile(file, options = {})
-      sql = "LOAD DATA LOCAL INFILE '#{file}' INTO TABLE #{quoted_table_name} "
-      sql << "FIELDS TERMINATED BY '#{options[:fields_terminated_by]}' " if options[:fields_terminated_by]
-      sql << "LINES TERMINATED BY '#{options[:lines_terminated_by]}' " if options[:lines_terminated_by]
-      sql << "IGNORE #{options[:ignore_lines]} LINES " if options[:ignore_lines]
-      sql << "(" + options[:columns].join(', ') + ") " if options[:columns]
-      if options[:mapping]
-        mappings = []
-        options[:mapping].each_pair do |column, mapping|
-          mappings << "#{column} = #{mapping}"
-        end
-        sql << "SET #{mappings.join(', ')} " if mappings.size > 0
-      end
-      sql << ";"
-      connection.execute(sql)
-    end
-  end
+module ActiveRecord #:nodoc:
+  class Base
+    # Deletes all rows in table very fast, but without calling +destroy+ method
+    # nor any hooks.
+    def self.truncate_table
+      connection.execute("TRUNCATE TABLE #{quoted_table_name}")
+    end
+
+    # Disables key updates for model table
+    def self.disable_keys
+      connection.execute("ALTER TABLE #{quoted_table_name} DISABLE KEYS")
+    end
+
+    # Enables key updates for model table
+    def self.enable_keys
+      connection.execute("ALTER TABLE #{quoted_table_name} ENABLE KEYS")
+    end
+
+    # Loads data from file using MySQL native LOAD DATA INFILE query, disabling
+    # key updates for even faster import speed
+    #
+    # ==== Parameters
+    # * +file+ file(s) to import
+    # * +options+ (see <tt>load_data_infile</tt>)
+    def self.fast_import(files, options = {})
+      files = [files] unless files.is_a? Array
+      enable_keys
+      files.each {|file| load_data_infile(file, options)}
+      disable_keys
+    end
+
+    # Loads data from file using MySQL native LOAD DATA INFILE query
+    #
+    # ==== Parameters
+    # * +file+ the file to import
+    # * +options+ 
+    def self.load_data_infile(file, options = {})
+      sql = "LOAD DATA LOCAL INFILE '#{file}' INTO TABLE #{quoted_table_name} "
+      sql << "FIELDS TERMINATED BY '#{options[:fields_terminated_by]}' " if options[:fields_terminated_by]
+      sql << "LINES TERMINATED BY '#{options[:lines_terminated_by]}' " if options[:lines_terminated_by]
+      sql << "IGNORE #{options[:ignore_lines]} LINES " if options[:ignore_lines]
+      sql << "(" + options[:columns].join(', ') + ") " if options[:columns]
+      if options[:mapping]
+        mappings = []
+        options[:mapping].each_pair do |column, mapping|
+          mappings << "#{column} = #{mapping}"
+        end
+        sql << "SET #{mappings.join(', ')} " if mappings.size > 0
+      end
+      sql << ";"
+      connection.execute(sql)
+    end
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: jsuchal-activerecord-fast-import
 version: !ruby/object:Gem::Version 
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors: 
 - Jan Suchal
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-08-19 00:00:00 -07:00
+date: 2009-08-21 00:00:00 -07:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency