Class: ActiveFedora::SolrService

Inherits:
Object
  • Object
show all
Defined in:
lib/active_fedora/solr_service.rb

Constant Summary collapse

MAX_ROWS =
10_000

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(options = {}) ⇒ SolrService

Returns a new instance of SolrService.



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

def initialize(options = {})
  @options = { read_timeout: 120, open_timeout: 120, url: 'http://localhost:8080/solr' }.merge(options)
end

Instance Attribute Details

#connObject



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

def conn
  @conn ||= RSolr.connect @options
end

#optionsObject (readonly)

Returns the value of attribute options.



5
6
7
# File 'lib/active_fedora/solr_service.rb', line 5

def options
  @options
end

Class Method Details

.add(doc, params = {}) ⇒ Object

Parameters:

  • doc (Hash)

    the document to index, or an array of docs

  • params (Hash) (defaults to: {})

    :commit => commits immediately :softCommit => commit to memory, but don’t flush to disk



86
87
88
# File 'lib/active_fedora/solr_service.rb', line 86

def add(doc, params = {})
  SolrService.instance.conn.add(doc, params: params)
end

.commitObject



90
91
92
# File 'lib/active_fedora/solr_service.rb', line 90

def commit
  SolrService.instance.conn.commit
end

.count(query, args = {}) ⇒ Integer

Get the count of records that match the query

Parameters:

  • query (String)

    a solr query

  • args (Hash) (defaults to: {})

    arguments to pass through to ‘args’ param of SolrService.query (note that :rows will be overwritten to 0)

Returns:

  • (Integer)

    number of records matching



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

def count(query, args = {})
  args = args.merge(rows: 0)
  SolrService.get(query, args)['response']['numFound'].to_i
end

.delete(id) ⇒ Object



69
70
71
# File 'lib/active_fedora/solr_service.rb', line 69

def delete(id)
  SolrService.instance.conn.delete_by_id(id, params: { 'softCommit' => true })
end

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



42
43
44
45
# File 'lib/active_fedora/solr_service.rb', line 42

def get(query, args = {})
  args = args.merge(q: query, qt: 'standard')
  SolrService.instance.conn.get(select_path, params: args)
end

.instanceObject



32
33
34
35
36
37
38
39
40
# File 'lib/active_fedora/solr_service.rb', line 32

def instance
  # Register Solr

  unless ActiveFedora::RuntimeRegistry.solr_service
    register(ActiveFedora.solr_config)
  end

  ActiveFedora::RuntimeRegistry.solr_service
end

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



47
48
49
50
# File 'lib/active_fedora/solr_service.rb', line 47

def post(query, args = {})
  args = args.merge(q: query, qt: 'standard')
  SolrService.instance.conn.post(select_path, data: args)
end

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



52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
# File 'lib/active_fedora/solr_service.rb', line 52

def query(query, args = {})
  Base.logger.warn "Calling ActiveFedora::SolrService.get without passing an explicit value for ':rows' is not recommended. You will end up with Solr's default (usually set to 10)\nCalled by #{caller[0]}" unless args.key?(:rows)
  method = args.delete(:method) || :get

  result = case method
           when :get
             get(query, args)
           when :post
             post(query, args)
           else
             raise "Unsupported HTTP method for querying SolrService (#{method.inspect})"
           end
  result['response']['docs'].map do |doc|
    ActiveFedora::SolrHit.new(doc)
  end
end

.register(options = {}) ⇒ Object

Parameters:

  • options (Hash) (defaults to: {})


20
21
22
# File 'lib/active_fedora/solr_service.rb', line 20

def register(options = {})
  ActiveFedora::RuntimeRegistry.solr_service = new(options)
end

.reset!Object



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

def reset!
  ActiveFedora::RuntimeRegistry.solr_service = nil
end

.select_pathObject



28
29
30
# File 'lib/active_fedora/solr_service.rb', line 28

def select_path
  ActiveFedora.solr_config.fetch(:select_path, 'select')
end