Sha256: 3a334c7f09726dddeac3785b63e2ed57ab14f720d18f0752c5b3a67aababe93c

Contents?: true

Size: 1.29 KB

Versions: 11

Compression:

Stored size: 1.29 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
      timed_exclaim "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.11.2 lib/picky/indexers/solr.rb
picky-0.11.1 lib/picky/indexers/solr.rb
picky-0.11.0 lib/picky/indexers/solr.rb
picky-0.10.5 lib/picky/indexers/solr.rb
picky-0.10.4 lib/picky/indexers/solr.rb
picky-0.10.2 lib/picky/indexers/solr.rb
picky-0.10.1 lib/picky/indexers/solr.rb
picky-0.10.0 lib/picky/indexers/solr.rb
picky-0.9.4 lib/picky/indexers/solr.rb
picky-0.9.3 lib/picky/indexers/solr.rb
picky-0.9.2 lib/picky/indexers/solr.rb