active_model_archive 1.1.2 → 1.2.0

data/Rakefile

@@ -1,2 +1,12 @@
-require 'bundler'
-Bundler::GemHelper.install_tasks
+require 'bundler/setup'
+require 'rake'
+require 'rake/testtask'
+
+task default: :test
+
+Rake::TestTask.new(:test) do |t|
+  t.libs << 'lib'
+  t.libs << 'test'
+  t.pattern = 'test/unit/**/*_test.rb'
+  t.verbose = true
+end

data/active_model_archive.gemspec

@@ -2,7 +2,7 @@
 
 Gem::Specification.new do |s|
   s.name        = "active_model_archive"
-  s.version     = '1.1.2'
+  s.version     = '1.2.0'
   s.platform    = Gem::Platform::RUBY
   s.authors     = ["Grant Rodgers"]
   s.email       = ["grantr@gmail.com"]

data/lib/active_model_archive/railtie.rb

@@ -0,0 +1,7 @@
+module ActiveModelArchive
+  class Railtie < Rails::Railtie
+    rake_tasks do
+      load 'active_model_archive/tasks/archive.rake'
+    end
+  end
+end

data/lib/active_model_archive/tasks/archive.rake

@@ -0,0 +1,56 @@
+namespace :archive do
+  # == Example:
+  # rake archive:dump CLASS=Widgets FORMAT=tabs PER_FILE=10000 FILE=/mnt/widgets.tsv
+  desc "archive CLASS, using PER_FILE for number of records per FILE"
+  task dump: :environment do
+    filename = ENV['FILE'] || (raise "FILE required.")
+    model = (ENV['CLASS'] || (raise "CLASS required")).camelize.constantize
+
+    if ENV['PER_FILE']
+      per_file = ENV['PER_FILE'].to_i
+    else
+      per_file = nil
+    end
+
+    if ENV['FORMAT']
+      model.archiver_type = ENV['FORMAT']
+    end
+
+    logger.info "Dumping to #{filename}"
+    count = model.dump!(filename, per_file: per_file) do |object|
+      begin
+        logger.info object.id
+      rescue => e
+        logger.error "OBJECT #{object.id} caused dump to fail"
+        raise e
+      end
+    end
+    logger.info "#{count} records dumped."
+  end
+
+  # == Example:
+  # rake archive:restore CLASS=Widgets FORMAT=tabs FILE=/mnt/widgets.tsv
+  desc "restore a dump from FILE"
+  task restore: :environment do
+    filename = ENV['FILE'] || (raise "FILE required.")
+    model = (ENV['CLASS'] || (raise "CLASS required")).camelize.constantize
+
+    no_publish = (ENV['PUBLISH'] != 'true')
+    model.disable_publishing! if no_publish && model.respond_to?(:disable_publishing!)
+
+    start = Time.now
+
+    File.open(filename) do |file|
+      logger.info "Restoring from #{file.path}. indexing=#{!no_index}, publishing=#{!no_publish}"
+      count = model.each_instance(file) do |record|
+        logger.info "Saving #{record.id}"
+        record.save(validate: false)
+        unless record.persisted?
+          puts "Error saving record: #{record.inspect}"
+        end
+      end
+
+      puts "#{count} records restored in #{Time.now - start}."
+    end
+  end
+end

data/lib/active_model_archive.rb

@@ -44,3 +44,5 @@ module ActiveModel
     end
   end
 end
+
+require 'active_model_archive/railtie'

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: active_model_archive
 version: !ruby/object:Gem::Version
-  version: 1.1.2
+  version: 1.2.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-05-25 00:00:00.000000000 Z
+date: 2012-06-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -62,7 +62,9 @@ files:
 - lib/active_model_archive/attributes.rb
 - lib/active_model_archive/dump.rb
 - lib/active_model_archive/file_manager.rb
+- lib/active_model_archive/railtie.rb
 - lib/active_model_archive/restore.rb
+- lib/active_model_archive/tasks/archive.rake
 - test/archiver/csv_test.rb
 - test/archiver/tabs_test.rb
 - test/attributes_test.rb