63 lines
1.9 KiB
Ruby
63 lines
1.9 KiB
Ruby
module Polecat
|
|
# interface for searching an index
|
|
#
|
|
# Build on top of an Polecat::IndexReader, this class let's you search through
|
|
# all documents stored in an index.
|
|
class IndexSearcher
|
|
attr_reader :reader
|
|
attr_reader :default_field
|
|
|
|
# creates a new Polecat::IndexSearcher
|
|
#
|
|
# Create a new Polecat::IndexSearcher to search documents. Either a path
|
|
# to a directory or a Polecat::IndexReader has to be given, to make this
|
|
# searcher work.
|
|
# @example
|
|
# # the following has the same meaning
|
|
# IndexSearcher.new 'index_dir'
|
|
# IndexSearcher.new(IndexReader.new 'index_dir')
|
|
def initialize options
|
|
if options.has_key? :path
|
|
@reader = Polecat::IndexReader.new(options[:path])
|
|
elsif options.has_key? :reader
|
|
@reader = options[:reader]
|
|
raise ArgumentError, 'no reader' unless @reader.kind_of?(Polecat::IndexReader)
|
|
end
|
|
|
|
if options.has_key? :default_field
|
|
@default_field = options[:default_field]
|
|
end
|
|
end
|
|
|
|
# returns the path of the index directory
|
|
# @return [String] path of the index directory
|
|
def path
|
|
@reader.path
|
|
end
|
|
|
|
# searches through all documents
|
|
#
|
|
# Run the query against the @default_field@ of every stored document to get
|
|
# a list of all matching documents.
|
|
# @param [String] query a String which get's matched against the documents
|
|
# @return [Array] a list of all matching documents
|
|
def search query
|
|
@content = @reader.read if @content.nil?
|
|
@content.select do |doc|
|
|
#doc.attributes.fetch(@default_field).fetch(:value) == query
|
|
rs = []
|
|
query.terms.each do |term|
|
|
if term.compare(doc.send(term.field))
|
|
rs << true
|
|
end
|
|
end
|
|
if query.relation == :and
|
|
rs.count == query.terms.count
|
|
else
|
|
!rs.empty?
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|