mwmitchell-rsolr_ext 0.2.0 → 0.3.0

data/lib/rsolr_ext/params.rb

@@ -0,0 +1,23 @@
+# utility for querying building
+# RSolrExt::Params.escape('a string...')
+# q = RSolrExt::Params.create_fielded_queries(:name=>'a string...', :cat=>[:one, :two])
+# q == ["a string...", "cat:one", "cat:two"]
+class RSolrExt::Params
+  
+  def self.escape(value, quote=false)
+    quote ? %("#{value}") : value
+  end
+  
+  def self.create_fielded_queries(params, quote=nil)
+    params.collect do |k,v|
+    	v.is_a?(Array) ? v.collect{|vv|%(#{k}:#{escape(vv, quote)})} : "#{k}:#{escape(v, quote)}"
+    end.flatten
+  end
+
+  def self.calculate_start(page, rows)
+    page = page.to_s.to_i-1
+    page = page < 1 ? 0 : page
+    page * rows.to_i
+  end
+  
+end

data/lib/rsolr_ext/response/base.rb

@@ -0,0 +1,35 @@
+#
+# my_solr_hash.extend RSolrExt::Response::Base
+# my_solr_hash.header
+# my_solr_hash.ok?
+#
+module RSolrExt::Response::Base
+  
+  def header
+    self[:responseHeader]
+  end
+  
+  def params
+    header[:params]
+  end
+  
+  def status
+    header[:status].to_i
+  end
+  
+  def query_time
+    header[:QTime]
+  end
+  
+  def ok?
+    self.status == 0
+  end
+  
+  # converts to mash, then extends
+  def self.create(hash)
+    mash = hash.is_a?(Mash) ? hash : hash.to_mash
+    mash.extend self
+    mash
+  end
+  
+end # end Base

data/lib/rsolr_ext/response/luke.rb

@@ -0,0 +1,53 @@
+module RSolrExt::Response::Luke
+  
+  include Base
+  
+  def index
+    self[:index]
+  end
+  
+  def directory
+    index[:directory]
+  end
+  
+  def has_deletions
+    index[:hasDeletions]
+  end
+  
+  def current
+    index[:current]
+  end
+  
+  def max_doc
+    index[:max_doc]
+  end
+  
+  def num_docs
+    index[:numDocs]
+  end
+  
+  def version
+    index[:version]
+  end
+  
+  alias :has_deletions? :has_deletions
+  alias :optimized? :optimized
+  alias :current? :current
+
+  # Returns an array of fields from the index
+  # An optional rule can be used for "grepping" field names:
+  # field_list(/_facet$/)
+  def field_list(rule=nil)
+    self[:fields].select do |k,v|
+      rule ? k =~ rule : true
+    end.collect{|k,v|k}
+  end
+  
+  # converts to mash, then extends
+  def self.create(hash)
+    mash = hash.is_a?(Mash) ? hash : hash.to_mash
+    mash.extend self
+    mash
+  end
+  
+end# end Luke

data/lib/rsolr_ext/response/select.rb

@@ -0,0 +1,216 @@
+module RSolrExt::Response::Select
+  
+  # module for adding helper methods to each solr response[:docs] object
+  module DocExt
+
+    # Helper method to check if value/multi-values exist for a given key.
+    # The value can be a string, or a RegExp
+    # Example:
+    # doc.has?(:location_facet)
+    # doc.has?(:location_facet, 'Clemons')
+    # doc.has?(:id, 'h009', /^u/i)
+    def has?(k, *values)
+      return if self[k].nil?
+      return true if self.key?(k) and values.empty?
+      target = self[k]
+      if target.is_a?(Array)
+        values.each do |val|
+          return target.any?{|tv| val.is_a?(Regexp) ? (tv =~ val) : (tv==val)}
+        end
+      else
+        return values.any? {|val| val.is_a?(Regexp) ? (target =~ val) : (target == val)}
+      end
+    end
+
+    # helper
+    # key is the name of the field
+    # opts is a hash with the following valid keys:
+    #  - :sep - a string used for joining multivalued field values
+    #  - :default - a value to return when the key doesn't exist
+    # if :sep is nil and the field is a multivalued field, the array is returned
+    def get(key, opts={:sep=>', ', :default=>nil})
+      if self.key? key
+        val = self[key]
+        (val.is_a?(Array) and opts[:sep]) ? val.join(opts[:sep]) : val
+      else
+        opts[:default]
+      end
+    end
+
+  end
+
+  module Facets
+    
+    # represents a facet value; which is a field value and its hit count
+    class FacetValue
+      attr_reader :value,:hits
+      def initialize(value,hits)
+        @value,@hits=value,hits
+      end
+    end
+    
+    # represents a facet; which is a field and its values
+    class Facet
+      attr_reader :field
+      attr_accessor :values
+      def initialize(field)
+        @field=field
+        @values=[]
+      end
+    end
+
+    # @response.facets.each do |facet|
+    #   facet.field
+    # end
+    # "caches" the result in the @facets instance var
+    def facets
+      # memoize!
+      @facets ||= (
+        facet_fields.inject([]) do |acc,(facet_field_name,values_and_hits_list)|
+          acc << facet = Facet.new(facet_field_name)
+          # the values_and_hits_list is an array where a value is immediately followed by it's hit count
+          # so we shift off an item (the value)
+          while value = values_and_hits_list.shift
+            # and then shift off the next to get the hit value
+            facet.values << FacetValue.new(value, values_and_hits_list.shift)
+            # repeat until there are no more pairs in the values_and_hits_list array
+          end
+          acc
+        end
+      )
+    end
+
+    # pass in a facet field name and get back a Facet instance
+    def facet_by_field_name(name)
+      @facets_by_field_name ||= {}
+      @facets_by_field_name[name] ||= (
+        facets.detect{|facet|facet.field.to_s == name.to_s}
+      )
+    end
+    
+    def facet_counts
+      @facet_counts ||= self[:facet_counts] || {}
+    end
+
+    # Returns the hash of all the facet_fields (ie: {'instock_b' => ['true', 123, 'false', 20]}
+    def facet_fields
+      @facet_fields ||= facet_counts[:facet_fields] || {}
+    end
+
+    # Returns all of the facet queries
+    def facet_queries
+      @facet_queries ||= facet_counts[:facet_queries] || {}
+    end
+    
+  end # end Facets
+  
+  #
+  #
+  #
+  class FacetPaginator
+    
+    attr_reader :total, :items, :previous_offset, :next_offset
+    
+    def initialize(all_facet_values, offset, limit)
+      offset = offset.to_s.to_i
+      limit = limit.to_s.to_i
+      total = all_facet_values.size
+      @items = all_facet_values.slice(0, limit-1)
+      @has_next = total == limit
+      @has_previous = offset > 0
+      @next_offset = offset + (limit-1)
+      @previous_offset = offset - (limit-1)
+    end
+
+    def has_next?
+      @has_next
+    end
+
+    def has_previous?
+      @has_previous
+    end
+
+  end
+  
+  #
+  #
+  #
+  class Paginator
+    
+    attr_reader :start, :per_page, :total
+    
+    def initialize(start, per_page, total)
+      @start = start.to_s.to_i
+      @per_page = per_page.to_s.to_i
+      @total = total.to_s.to_i
+    end
+    
+    # Returns the current page calculated from 'rows' and 'start'
+    # WillPaginate hook
+    def current_page
+      return 1 if start < 1
+      @current_page ||= (start / per_page).ceil + 1
+    end
+
+    # Calcuates the total pages from 'numFound' and 'rows'
+    # WillPaginate hook
+    def total_pages
+      @total_pages ||= per_page > 0 ? (total / per_page.to_f).ceil : 1
+    end
+
+    # returns the previous page number or 1
+    # WillPaginate hook
+    def previous_page
+      @previous_page ||= (current_page > 1) ? current_page - 1 : 1
+    end
+    
+    # returns the next page number or the last
+    # WillPaginate hook
+    def next_page
+      @next_page ||= (current_page < total_pages) ? current_page + 1 : total_pages
+    end
+  end
+  
+  def paginator
+    @paginator ||= Paginator.new(start, rows, total)
+  end
+  
+  # The main select response class.
+  # Includes the top level Response::Base module
+  # Includes the Pagination module.
+  # Each solr hash doc is extended by the DocExt module.
+  
+  include RSolrExt::Response::Base
+  include Facets
+  
+  def response
+    self[:response]
+  end
+  
+  def num_found
+    response[:numFound]
+  end
+  
+  def start
+    response[:start]
+  end
+  
+  def rows
+    params[:rows]
+  end
+  
+  alias :total :num_found
+  alias :offset :start
+  
+  def docs
+    @docs ||= response[:docs].collect{ |d| d=d.to_mash; d.extend(DocExt); d }
+  end
+  
+  # converts to mash, then extends
+  def self.create(hash)
+    mash = hash.is_a?(Mash) ? hash : hash.to_mash
+    mash.extend self
+    mash
+  end
+  
+end # end Select

data/lib/rsolr_ext/response/update.rb

@@ -0,0 +1,13 @@
+# for update responses
+module RSolrExt::Response::Update
+  
+  include Base
+  
+  # converts to mash, then extends
+  def self.create(hash)
+    mash = hash.is_a?(Mash) ? hash : hash.to_mash
+    mash.extend self
+    mash
+  end
+  
+end

data/lib/rsolr_ext/response.rb

@@ -0,0 +1,8 @@
+module RSolrExt::Response
+  
+  autoload :Base, 'rsolr_ext/response/base'
+  autoload :Select, 'rsolr_ext/response/select'
+  autoload :Luke, 'rsolr_ext/response/luke'
+  autoload :Update, 'rsolr_ext/response/update'
+  
+end

data/lib/rsolr_ext.rb

@@ -13,7 +13,7 @@ end
 
 module RSolrExt
   
-  VERSION = '0.2.0'
+  VERSION = '0.2.1'
   
   autoload :Params, 'rsolr_ext/params'
   autoload :Response, 'rsolr_ext/response'

data/rsolr_ext.gemspec

@@ -1,7 +1,7 @@
 Gem::Specification.new do |s|
 	s.name = "rsolr_ext"
-	s.version = "0.2.0"
-	s.date = "2009-03-12"
+	s.version = "0.3.0"
+	s.date = "2009-03-13"
 	s.summary = "An extension lib for RSolr"
 	s.email = "goodieboy@gmail.com"
 	s.homepage = "http://github.com/mwmitchell/rsolr_ext"
@@ -10,12 +10,12 @@ Gem::Specification.new do |s|
 	s.authors = ["Matt Mitchell"]
 	s.files = [
 	"lib/rsolr_ext.rb",
-	"lib/rsolr_ext/params",
-	"lib/rsolr_ext/response",
-	"lib/rsolr_ext/response/base",
-	"lib/rsolr_ext/response/luke",
-	"lib/rsolr_ext/response/select",
-	"lib/rsolr_ext/response/update",
+	"lib/rsolr_ext/params.rb",
+	"lib/rsolr_ext/response.rb",
+	"lib/rsolr_ext/response/base.rb",
+	"lib/rsolr_ext/response/luke.rb",
+	"lib/rsolr_ext/response/select.rb",
+	"lib/rsolr_ext/response/update.rb",
 	"LICENSE",
 	"README.rdoc",
 	"rsolr_ext.gemspec"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: mwmitchell-rsolr_ext
 version: !ruby/object:Gem::Version 
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors: 
 - Matt Mitchell
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-03-12 00:00:00 -07:00
+date: 2009-03-13 00:00:00 -07:00
 default_executable: 
 dependencies: []
 
@@ -24,12 +24,12 @@ extra_rdoc_files:
 - README.rdoc
 files: 
 - lib/rsolr_ext.rb
-- lib/rsolr_ext/params
-- lib/rsolr_ext/response
-- lib/rsolr_ext/response/base
-- lib/rsolr_ext/response/luke
-- lib/rsolr_ext/response/select
-- lib/rsolr_ext/response/update
+- lib/rsolr_ext/params.rb
+- lib/rsolr_ext/response.rb
+- lib/rsolr_ext/response/base.rb
+- lib/rsolr_ext/response/luke.rb
+- lib/rsolr_ext/response/select.rb
+- lib/rsolr_ext/response/update.rb
 - LICENSE
 - README.rdoc
 - rsolr_ext.gemspec