Class: ActiveFedora::SolrService

Inherits:
Object
  • Object
show all
Includes:
Loggable, Solrizer::FieldNameMapper
Defined in:
lib/active_fedora/solr_service.rb

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(host, args) ⇒ SolrService

Returns a new instance of SolrService.



21
22
23
24
25
26
 
# File 'lib/active_fedora/solr_service.rb', line 21

def initialize(host, args)
  host = 'http://localhost:8080/solr' unless host
  args = {:read_timeout => 120, :open_timeout => 120}.merge(args.dup)
  args.merge!(:url=>host)
  @conn = RSolr.connect args
end

Instance Attribute Details

#conn ⇒ Object (readonly)

Returns the value of attribute conn.



10
11
12
 
# File 'lib/active_fedora/solr_service.rb', line 10

def conn
  @conn
end

Class Method Details

.add(doc) ⇒ Object 



96
97
98
 
# File 'lib/active_fedora/solr_service.rb', line 96

def self.add(doc)
  SolrService.instance.conn.add(doc)
end

.class_from_solr_document(hit) ⇒ Object 



52
53
54
55
56
57
 
# File 'lib/active_fedora/solr_service.rb', line 52

def self.class_from_solr_document(hit)
    model_value = nil
    hit[solr_name("has_model", :symbol)].each {|value| model_value ||= Model.from_class_uri(value)}
    logger.warn "Could not find a model for #{hit["id"]}, defaulting to ActiveFedora::Base" unless model_value
    model_value || ActiveFedora::Base
end

.commit ⇒ Object 



100
101
102
 
# File 'lib/active_fedora/solr_service.rb', line 100

def self.commit
  SolrService.instance.conn.commit
end

.construct_query_for_pids(pid_array) ⇒ Object

Construct a solr query for a list of pids This is used to get a solr response based on the list of pids in an object’s RELS-EXT relationhsips If the pid_array is empty, defaults to a query of “id:NEVER_USE_THIS_ID”, which will return an empty solr response

Parameters:

  • pid_array (Array) —

    the pids that you want included in the query



63
64
65
66
67
68
69
70
71
 
# File 'lib/active_fedora/solr_service.rb', line 63

def self.construct_query_for_pids(pid_array)
  query = ""
  pid_array.each_index do |i|
    query << "#{SOLR_DOCUMENT_ID}:#{escape_uri_for_query(pid_array[i])}"
    query << " OR " if i != pid_array.length-1
  end
  query = "id:NEVER_USE_THIS_ID" if query.empty? || query == "id:"
  return query
end

.construct_query_for_rel(predicate, target_uri) ⇒ Object 



77
78
79
 
# File 'lib/active_fedora/solr_service.rb', line 77

def self.construct_query_for_rel(predicate, target_uri)
  "#{solr_name(predicate, :symbol)}:#{escape_uri_for_query(target_uri)}"
end

.count(query) ⇒ Object

Get the count of records that match the query

Parameters:

  • query (String) —

    a solr query



92
93
94
 
# File 'lib/active_fedora/solr_service.rb', line 92

def self.count(query)
    SolrService.query(query, :raw=>true, :rows=>0)['response']['numFound'].to_i
end

.escape_uri_for_query(uri) ⇒ Object 



73
74
75
 
# File 'lib/active_fedora/solr_service.rb', line 73

def self.escape_uri_for_query(uri)
  return uri.gsub(/(:)/, '\\:').gsub(/(\/)/, '\\/')
end

.instance ⇒ Object

Raises:



28
29
30
31
32
33
34
35
36
37
 
# File 'lib/active_fedora/solr_service.rb', line 28

def self.instance
# Register Solr
    
  unless Thread.current[:solr_service]
    register(ActiveFedora.solr_config[:url])
  end

  raise SolrNotInitialized unless Thread.current[:solr_service]
  Thread.current[:solr_service]
end

.query(query, args = {}) ⇒ Object 



81
82
83
84
85
86
87
 
# File 'lib/active_fedora/solr_service.rb', line 81

def self.query(query, args={})
  raw = args.delete(:raw)
  args = args.merge(:q=>query, :qt=>'standard')
  result = SolrService.instance.conn.get('select', :params=>args)
  return result if raw
  result['response']['docs']
end

.register(host = nil, args = {}) ⇒ Object 



12
13
14
15
 
# File 'lib/active_fedora/solr_service.rb', line 12

def self.register(host=nil, args={})
  load_mappings
  Thread.current[:solr_service]=self.new(host, args)
end

.reify_solr_results(solr_result, opts = {}) ⇒ Object 



39
40
41
42
43
44
45
46
47
48
49
50
 
# File 'lib/active_fedora/solr_service.rb', line 39

def self.reify_solr_results(solr_result,opts={})
  results = []
  solr_result.each do |hit|
    classname = class_from_solr_document(hit)
    if opts[:load_from_solr]
      results << classname.load_instance_from_solr(hit[SOLR_DOCUMENT_ID])
    else
      results << classname.find(hit[SOLR_DOCUMENT_ID])
    end
  end
  return results
end

.reset! ⇒ Object 



17
18
19
 
# File 'lib/active_fedora/solr_service.rb', line 17

def self.reset!
  Thread.current[:solr_service] = nil
end