supernova 0.3.11 → 0.3.12

data/VERSION

@@ -1 +1 @@
-0.3.11
+0.3.12

data/lib/supernova/solr_indexer.rb

@@ -146,24 +146,26 @@ class Supernova::SolrIndexer
     db.send(db.respond_to?(:query) ? :query : :select_all, query)
   end
   
-  def index_query(query)
+  def index_query(query, &block)
     rows = query_db(query)
     if self.max_rows_to_direct_index < rows.count
-      index_with_json_file(rows)
+      index_with_json_file(rows, &block)
     else
-      index_directly(rows)
+      index_directly(rows, &block)
     end
   end
   
-  def index_directly(rows)
+  def index_directly(rows, &block)
     rows.each do |row|
-      Supernova::Solr.connection.add(row)
+      row = yield(row) if block_given?
+      row = Supernova::Solr.connection.add(row)
     end
     Supernova::Solr.connection.commit if rows.any?
   end
   
-  def index_with_json_file(rows)
+  def index_with_json_file(rows, &block)
     rows.each do |row|
+      row = yield(row) if block_given?
       write_to_file(row)
     end
     finish

data/spec/integration/solr_spec.rb

@@ -41,6 +41,11 @@ describe "Solr" do
       has :user_id, :type => :integer
       has :popularity, :type => :integer
       
+      def row_to_solr(row)
+        row["indexed_at_dt"] = Time.now.utc.iso8601
+        row
+      end
+      
       clazz Offer
     end
     
@@ -49,6 +54,7 @@ describe "Solr" do
       offer2 = Offer.create!(:user_id => 2, :popularity => 20)
       indexer = OfferIndex.new(:db => ActiveRecord::Base.connection)
       indexer.index!
+      OfferIndex.search_scope.to_a.first.instance_variable_get("@solr_doc")["indexed_at_dt"].should_not be_nil
       OfferIndex.search_scope.to_a.total_entries.should == 2
       OfferIndex.search_scope.order("user_id desc").to_a.should == [offer2, offer1]
       indexer.instance_variable_get("@index_file_path").should be_nil
@@ -61,6 +67,7 @@ describe "Solr" do
       indexer.index!
       indexer.instance_variable_get("@index_file_path").should_not be_nil
       OfferIndex.search_scope.to_a.total_entries.should == 2
+      OfferIndex.search_scope.to_a.first.instance_variable_get("@solr_doc")["indexed_at_dt"].should_not be_nil
       OfferIndex.search_scope.order("user_id desc").to_a.should == [offer2, offer1]
       File.should_not be_exists(indexer.instance_variable_get("@index_file_path"))
     end
@@ -71,6 +78,7 @@ describe "Solr" do
       indexer = OfferIndex.new(:db => ActiveRecord::Base.connection, :max_rows_to_direct_index => 0, :local_solr => true)
       indexer.index!
       indexer.instance_variable_get("@index_file_path").should_not be_nil
+      OfferIndex.search_scope.to_a.first.instance_variable_get("@solr_doc")["indexed_at_dt"].should_not be_nil
       OfferIndex.search_scope.to_a.total_entries.should == 2
       OfferIndex.search_scope.order("user_id desc").to_a.should == [offer2, offer1]
       File.should_not be_exists(indexer.instance_variable_get("@index_file_path"))

data/supernova.gemspec

@@ -5,7 +5,7 @@
 
 Gem::Specification.new do |s|
   s.name = %q{supernova}
-  s.version = "0.3.11"
+  s.version = "0.3.12"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Tobias Schwab"]

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: supernova
 version: !ruby/object:Gem::Version 
-  hash: 5
+  hash: 11
   prerelease: 
   segments: 
   - 0
   - 3
-  - 11
-  version: 0.3.11
+  - 12
+  version: 0.3.12
 platform: ruby
 authors: 
 - Tobias Schwab