Sha256: 9e9e8d18d783e9122803f0f50567b8057c825261478a107a759e767ef347b9dc

Contents?: true

Size: 1.3 KB

Versions: 11

Compression:

Stored size: 1.3 KB

Contents

# encoding: utf-8
#
require 'rsolr'
module Indexers
  # This is an indexer in its own right.
  #
  # TODO Perhaps merge with the existing indexer.
  #
  class Solr

    attr_reader :type, :fields, :solr

    # Takes a Configuration::Type.
    #
    def initialize type
      @type   = type
      @fields = type.solr_fields.map(&:name).map(&:to_sym)
      @solr   = RSolr.connect
    end

    # TODO Rewrite such that it works in batches.
    #
    def index
      puts "#{Time.now}: Indexing solr for #{type.name}:#{fields.join(', ')}"
      statement = "SELECT indexed_id, #{fields.join(',')} FROM #{type.snapshot_table_name}"
      
      # TODO Rewrite.
      #
      DB.connect
      results   = DB.connection.execute statement
      
      return unless results # TODO check
      
      type_name = @type.name.to_s
      
      solr.delete_by_query "type:#{type_name}"
      solr.commit
      
      documents = []
      
      results.each do |indexed_id, *values|
        values.each &:downcase!
        documents << hashed(values).merge(:id => indexed_id, :type => type_name)
      end

      solr.add documents
      solr.commit
      solr.optimize
    end

    def hashed values
      result = {}
      fields.zip(values).each do |field, value|
        result[field] = value
      end
      result
    end

  end
end

Version data entries

11 entries across 11 versions & 1 rubygems

Version Path
picky-0.2.3 lib/picky/indexers/solr.rb
picky-0.2.2 lib/picky/indexers/solr.rb
picky-0.2.1 lib/picky/indexers/solr.rb
picky-0.2.0 lib/picky/indexers/solr.rb
picky-0.1.0 lib/picky/indexers/solr.rb
picky-0.0.9 lib/picky/indexers/solr.rb
picky-0.0.8 lib/picky/indexers/solr.rb
picky-0.0.7 lib/picky/indexers/solr.rb
picky-0.0.6 lib/picky/indexers/solr.rb
picky-0.0.5 lib/picky/indexers/solr.rb
picky-0.0.4 lib/picky/indexers/solr.rb