Class: GdsApi::Rummager

Inherits:
Base
  • Object
show all
Defined in:
lib/gds_api/rummager.rb

Defined Under Namespace

Classes: UnknownAPIVersion, V1, V2

Constant Summary collapse

DEFAULT_API_VERSION = 
'V1'.freeze
API_VERSIONS = 
{
  'V1' => GdsApi::Rummager::V1,
  'V2' => GdsApi::Rummager::V2,
}.freeze

Instance Attribute Summary

Attributes inherited from Base

#options

Instance Method Summary collapse

Methods inherited from Base

#client, #create_client, #get_list, #url_for_slug

Constructor Details

#initialize(endpoint_url, options = {}) ⇒ Rummager

Returns a new instance of Rummager.



54
55
56
57
58
59
60
61
 
# File 'lib/gds_api/rummager.rb', line 54

def initialize(endpoint_url, options = {})
  super
  # The API version provides a simple wrapper around this base class so that we
  # can still access the shared methods present in this class.
  version = options.fetch(:api_version, DEFAULT_API_VERSION)
  api_class = API_VERSIONS[version] || raise(UnknownAPIVersion)
  @api = api_class.new(self)
end

Instance Method Details

#add_document(*args) ⇒ GdsApi::Response

Add a document to the search index.

Parameters:

  • type (String) —

    The rummager/elasticsearch document type.

  • id (String) —

    The rummager/elasticsearch id. Typically the same as the ‘link` field, but this is not strictly enforced.

  • document (Hash) —

    The document to add. Must match the rummager schema matchin the ‘type` parameter and contain a `link` field.

  • index_name (V2 only) —

    Name of the index to be deleted from on GOV.UK - we only allow deletion from metasearch

Returns:

  • (GdsApi::Response) —

    A status code of 202 indicates the document has been successfully queued.

See Also:



130
131
132
 
# File 'lib/gds_api/rummager.rb', line 130

def add_document(*args)
  @api.add_document(*args)
end

#advanced_search(args) ⇒ Object

Deprecated.

Only in use by Whitehall. Use the ‘#search` method.

Advanced search.

Raises:

  • (ArgumentError) 


114
115
116
117
118
 
# File 'lib/gds_api/rummager.rb', line 114

def advanced_search(args)
  raise ArgumentError.new("Args cannot be blank") if args.nil? || args.empty?
  request_path = "#{base_url}/advanced_search?#{Rack::Utils.build_nested_query(args)}"
  get_json(request_path)
end

#base_url ⇒ Object 



172
173
174
 
# File 'lib/gds_api/rummager.rb', line 172

def base_url
  endpoint
end

#batch_search(searches, additional_headers = {}) ⇒ Object

Perform a batch search.

# @see github.com/alphagov/rummager/blob/master/doc/search-api.md

Parameters:

  • searches (Array) —

    An array valid search queries. Maximum of 6. See Rummager documentation for options.



78
79
80
81
82
83
84
85
 
# File 'lib/gds_api/rummager.rb', line 78

def batch_search(searches, additional_headers = {})
  url_friendly_searches = searches.each_with_index.map do |search, index|
    { index => search }
  end
  searches_query = { search: url_friendly_searches }
  request_url = "#{base_url}/batch_search.json?#{Rack::Utils.build_nested_query(searches_query)}"
  get_json(request_url, additional_headers)
end

#delete_content(base_path) ⇒ Object

Delete a content-document from the index by base path.

Content documents are pages on GOV.UK that have a base path and are returned in searches. This excludes best bets, recommended-links, and contacts, which may be deleted with ‘delete_document`.

Parameters:

  • base_path —

    Base path of the page on GOV.UK.

See Also:



142
143
144
145
 
# File 'lib/gds_api/rummager.rb', line 142

def delete_content(base_path)
  request_url = "#{base_url}/content?link=#{base_path}"
  delete_json(request_url)
end

#delete_document(*args) ⇒ Object

Delete a non-content document from the search index.

For example, best bets, recommended links, or contacts.

Parameters:

  • type (String) —

    The rummager/elasticsearch document type.

  • id (String) —

    The rummager/elasticsearch id. Typically the same as the ‘link` field.

  • index_name (V2 only) —

    Name of the index to be deleted from on GOV.UK - we only allow deletion from metasearch



168
169
170
 
# File 'lib/gds_api/rummager.rb', line 168

def delete_document(*args)
  @api.delete_document(*args)
end

#documents_url ⇒ Object 



176
177
178
 
# File 'lib/gds_api/rummager.rb', line 176

def documents_url
  "#{base_url}/documents"
end

#get_content(base_path) ⇒ Object

Retrieve a content-document from the index.

Content documents are pages on GOV.UK that have a base path and are returned in searches. This excludes best bets, recommended-links, and contacts.

Parameters:

  • base_path (String) —

    Base path of the page on GOV.UK.

See Also:



155
156
157
158
 
# File 'lib/gds_api/rummager.rb', line 155

def get_content(base_path)
  request_url = "#{base_url}/content?link=#{base_path}"
  get_json(request_url)
end

#search(args, additional_headers = {}) ⇒ Object

Perform a search.

Parameters:

  • args (Hash) —

    A valid search query. See Rummager documentation for options.

See Also:



68
69
70
71
 
# File 'lib/gds_api/rummager.rb', line 68

def search(args, additional_headers = {})
  request_url = "#{base_url}/search.json?#{Rack::Utils.build_nested_query(args)}"
  get_json(request_url, additional_headers)
end

#search_enum(args, page_size: 100, additional_headers: {}) ⇒ Object

Perform a search, returning the results as an enumerator.

The enumerator abstracts away rummager’s pagination and fetches new pages when necessary.

Parameters:

  • args (Hash) —

    A valid search query. See Rummager documentation for options.

  • page_size (Integer) (defaults to: 100) —

    Number of results in each page.

See Also:



96
97
98
99
100
101
102
103
104
105
106
107
108
109
 
# File 'lib/gds_api/rummager.rb', line 96

def search_enum(args, page_size: 100, additional_headers: {})
  Enumerator.new do |yielder|
    (0..Float::INFINITY).step(page_size).each do |index|
      search_params = args.merge(start: index.to_i, count: page_size)
      results = search(search_params, additional_headers).to_h.fetch('results', [])
      results.each do |result|
        yielder << result
      end
      if results.count < page_size
        break
      end
    end
  end
end