sunspot_mongo_mapper 0.1.0 → 0.1.1

data/lib/sunspot/mongo_mapper.rb

@@ -27,5 +27,25 @@ module Sunspot
         @clazz.all(:id.in => ids)
       end
     end
+
+    module ClassMethods
+      def find_in_batches(options = {}, &block)
+        puts "finding in batches: #{options.inspect}"
+        batch_size = options.delete(:batch_size) || 1000
+        start      = options.delete(:start) || 0
+        options.delete(:include)
+        records    = []
+
+        query(options).skip(start).order(:id.asc).each do |record|
+          records << record
+          if records.size == batch_size
+            yield records
+            records = []
+          end
+        end
+
+        yield records unless records.blank?
+      end
+    end
   end
 end

data/lib/sunspot/mongo_mapper/railtie.rb

@@ -0,0 +1,11 @@
+require 'rails'
+
+module Sunspot
+  module MongoMapper
+    class Railtie < ::Rails::Railtie
+      rake_tasks do
+        load "sunspot/mongo_mapper/tasks.rb"
+      end
+    end
+  end
+end

data/lib/sunspot/mongo_mapper/tasks.rb

@@ -0,0 +1,24 @@
+Rake.application['sunspot:reindex'].clear
+
+namespace :sunspot do
+  desc "Reindex all solr models that are located in your application's models directory."
+  task :reindex, [:batch_size, :models] => :environment do |t, args|
+    reindex_options = {:batch_commit => false}
+    case args[:batch_size]
+    when 'false'
+      reindex_options[:batch_size] = nil
+    when /^\d+$/
+      reindex_options[:batch_size] = args[:batch_size].to_i if args[:batch_size].to_i > 0
+    end
+    unless args[:models]
+      all_files = Dir.glob(Rails.root.join('app', 'models', '*.rb'))
+      all_models = all_files.map { |path| File.basename(path, '.rb').camelize.constantize }
+      sunspot_models = all_models.select { |m| m.respond_to?(:searchable?) && m.searchable? }
+    else
+      sunspot_models = args[:models].split('+').map{|m| m.constantize}
+    end
+    sunspot_models.each do |model|
+      model.solr_reindex reindex_options
+    end
+  end
+end

data/lib/sunspot_mongo_mapper.rb

@@ -1,3 +1,4 @@
 require 'sunspot/mongo_mapper'
+require 'sunspot/mongo_mapper/railtie'
 
 MongoMapper::Document.plugin Sunspot::MongoMapper

data/spec/spec_helper.rb

@@ -0,0 +1,11 @@
+require 'bundler'
+Bundler.require :default, :test
+require 'sunspot_mongo_mapper'
+
+MongoMapper.database = 'sunspot_mongo_mapper'
+
+RSpec.configure do |config|
+  config.before do
+    MongoMapper.database.collections.each(&:remove)
+  end
+end

data/spec/sunspot_mongo_mapper_spec.rb

@@ -0,0 +1,34 @@
+require 'spec_helper'
+
+describe Sunspot::MongoMapper do
+  class Post
+    include MongoMapper::Document
+  end
+
+  describe '.find_in_batches' do
+    before do
+      3.times {|i| Post.create!(:title => "Post #{i}") }
+    end
+
+    it 'should yield array of records in order' do
+      results = []
+
+      Post.find_in_batches(:batch_size => 2) do |records|
+        results << records
+      end
+
+      results[0].length.should == 2
+      results[1].length.should == 1
+    end
+
+    it 'should pass conditions' do
+      results = []
+      Post.find_in_batches(:title => 'Post 1') do |records|
+        results.concat records
+      end
+
+      results.size.should == 1
+      results.first.title.should == 'Post 1'
+    end
+  end
+end

data/sunspot_mongo_mapper.gemspec

@@ -2,7 +2,7 @@
 
 Gem::Specification.new do |s|
   s.name        = 'sunspot_mongo_mapper'
-  s.version     = '0.1.0'
+  s.version     = '0.1.1'
   s.authors     = ['Brandon Keepers']
   s.email       = ['brandon@opensoul.org']
   s.homepage    = "https://github.com/bkeepers/sunspot_mongo_mapper"
@@ -17,6 +17,7 @@ Gem::Specification.new do |s|
   s.add_runtime_dependency 'mongo_mapper'
   s.add_runtime_dependency 'sunspot',       '~> 1.1'
   s.add_runtime_dependency 'sunspot_rails', '~> 1.1'
+  s.add_runtime_dependency 'rails',         '~> 3.0'
   s.add_development_dependency 'rspec',     '~> 2.0'
   s.add_development_dependency 'rake'
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: sunspot_mongo_mapper
 version: !ruby/object:Gem::Version 
-  hash: 27
+  hash: 25
   prerelease: 
   segments: 
   - 0
   - 1
-  - 0
-  version: 0.1.0
+  - 1
+  version: 0.1.1
 platform: ruby
 authors: 
 - Brandon Keepers
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-11-17 00:00:00 -05:00
+date: 2011-11-18 00:00:00 -05:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -63,9 +63,24 @@ dependencies:
   type: :runtime
   version_requirements: *id003
 - !ruby/object:Gem::Dependency 
-  name: rspec
+  name: rails
   prerelease: false
   requirement: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 7
+        segments: 
+        - 3
+        - 0
+        version: "3.0"
+  type: :runtime
+  version_requirements: *id004
+- !ruby/object:Gem::Dependency 
+  name: rspec
+  prerelease: false
+  requirement: &id005 !ruby/object:Gem::Requirement 
     none: false
     requirements: 
     - - ~>
@@ -76,11 +91,11 @@ dependencies:
         - 0
         version: "2.0"
   type: :development
-  version_requirements: *id004
+  version_requirements: *id005
 - !ruby/object:Gem::Dependency 
   name: rake
   prerelease: false
-  requirement: &id005 !ruby/object:Gem::Requirement 
+  requirement: &id006 !ruby/object:Gem::Requirement 
     none: false
     requirements: 
     - - ">="
@@ -90,7 +105,7 @@ dependencies:
         - 0
         version: "0"
   type: :development
-  version_requirements: *id005
+  version_requirements: *id006
 description: A Sunspot wrapper for MongoMapper.
 email: 
 - brandon@opensoul.org
@@ -107,7 +122,11 @@ files:
 - README.md
 - Rakefile
 - lib/sunspot/mongo_mapper.rb
+- lib/sunspot/mongo_mapper/railtie.rb
+- lib/sunspot/mongo_mapper/tasks.rb
 - lib/sunspot_mongo_mapper.rb
+- spec/spec_helper.rb
+- spec/sunspot_mongo_mapper_spec.rb
 - sunspot_mongo_mapper.gemspec
 has_rdoc: true
 homepage: https://github.com/bkeepers/sunspot_mongo_mapper
@@ -143,5 +162,6 @@ rubygems_version: 1.6.1
 signing_key: 
 specification_version: 3
 summary: A Sunspot wrapper for MongoMapper.
-test_files: []
-
+test_files: 
+- spec/spec_helper.rb
+- spec/sunspot_mongo_mapper_spec.rb