activerecord-fast-import 0.1.3 → 0.1.4

data/VERSION

@@ -1 +1 @@
-0.1.3
+0.1.4

data/activerecord-fast-import.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = %q{activerecord-fast-import}
-  s.version = "0.1.3"
+  s.version = "0.1.4"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Jan Suchal"]
-  s.date = %q{2009-09-09}
+  s.date = %q{2009-10-08}
   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,3 +1,5 @@
+require 'activerecord'
+
 module ActiveRecord #:nodoc:
   class Base
     # Deletes all rows in table very fast, but without calling +destroy+ method
@@ -16,6 +18,13 @@ module ActiveRecord #:nodoc:
       connection.execute("ALTER TABLE #{quoted_table_name} ENABLE KEYS")
     end
 
+    # Disables keys, yields block, enables keys.
+    def self.with_keys_disabled
+      disable_keys
+      yield
+      enable_keys
+    end
+
     # Loads data from file using MySQL native LOAD DATA INFILE query, disabling
     # key updates for even faster import speed
     #
@@ -24,9 +33,9 @@ module ActiveRecord #:nodoc:
     # * +options+ (see <tt>load_data_infile</tt>)
     def self.fast_import(files, options = {})
       files = [files] unless files.is_a? Array
-      disable_keys
-      files.each {|file| load_data_infile(file, options)}
-      enable_keys
+      with_keys_disabled do
+        files.each {|file| load_data_infile(file, options)}
+      end
     end
 
     # Loads data from file using MySQL native LOAD DATA INFILE query

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: activerecord-fast-import
 version: !ruby/object:Gem::Version 
-  version: 0.1.3
+  version: 0.1.4
 platform: ruby
 authors: 
 - Jan Suchal
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-09-09 00:00:00 +02:00
+date: 2009-10-08 00:00:00 +02:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency