Contents

class ModelSet
  class SolrQuery < Query
    include Conditioned

    MAX_SOLR_RESULTS = 1000

    def anchor!(query)
      add_conditions!( ids_clause(query.ids) )
    end    

    def size
      fetch_results if @size.nil?
      @size
    end

    def count
      fetch_results if @count.nil?
      @count
    end

    def ids
      fetch_results if @ids.nil?
      @ids
    end

  private

    def fetch_results
      query = "#{conditions.to_s};#{@sort_order.to_s}"
      
      solr_params = []
      solr_params << "q=#{ ERB::Util::url_encode(query) }"
      solr_params << "wt=ruby"
      solr_params << "fl=pk_i"
      
      if limit
        solr_params << "rows=#{limit}"
        solr_params << "start=#{offset}"
      else
        solr_params << "rows=#{MAX_SOLR_RESULTS}"
      end
      
      solr_params = solr_params.join('&')
      before_query(solr_params)
      
      # Catch any errors when calling solr so we can log the params.
      begin
        resp = eval ActsAsSolr::Post.execute(solr_params)
      rescue Exception => e
        on_exception(e, solr_params)
      end

      after_query(solr_params)
      
      @count = resp['response']['numFound']
      @ids   = resp['response']['docs'].collect {|doc| doc['pk_i'].to_i}.to_ordered_set
      @size  = @ids.size
    end

    def ids_clause(ids, field = nil)
      return 'pk_i:(false)' if ids.empty?
      field ||= 'pk_i'
      "#{field}:(#{ids.join(' OR ')})"
    end

    def sanitize_condition(condition)
      condition
    end
  end
end

Version data entries

10 entries across 10 versions & 2 rubygems

Version	Path
ninjudd-model_set-0.10.0	lib/model_set/solr_query.rb
ninjudd-model_set-0.10.1	lib/model_set/solr_query.rb
ninjudd-model_set-0.10.2	lib/model_set/solr_query.rb
ninjudd-model_set-0.10.3	lib/model_set/solr_query.rb
ninjudd-model_set-0.10.4	lib/model_set/solr_query.rb
ninjudd-model_set-0.10.5	lib/model_set/solr_query.rb
ninjudd-model_set-0.9.2	lib/model_set/solr_query.rb
model_set-0.11.1	lib/model_set/solr_query.rb
model_set-0.11.0	lib/model_set/solr_query.rb
model_set-0.10.6	lib/model_set/solr_query.rb