Class: ElastomerClient::Client
- Inherits:
-
Object
- Object
- ElastomerClient::Client
- Includes:
- Notifications
- Defined in:
- lib/elastomer_client/notifications.rb,
lib/elastomer_client/client.rb,
lib/elastomer_client/client/bulk.rb,
lib/elastomer_client/client/docs.rb,
lib/elastomer_client/client/index.rb,
lib/elastomer_client/client/nodes.rb,
lib/elastomer_client/client/tasks.rb,
lib/elastomer_client/client/errors.rb,
lib/elastomer_client/client/cluster.rb,
lib/elastomer_client/client/reindex.rb,
lib/elastomer_client/client/scroller.rb,
lib/elastomer_client/client/snapshot.rb,
lib/elastomer_client/client/template.rb,
lib/elastomer_client/client/percolator.rb,
lib/elastomer_client/client/repository.rb,
lib/elastomer_client/client/multi_search.rb,
lib/elastomer_client/client/rest_api_spec.rb,
lib/elastomer_client/client/delete_by_query.rb,
lib/elastomer_client/client/multi_percolate.rb,
lib/elastomer_client/client/update_by_query.rb,
lib/elastomer_client/client/native_delete_by_query.rb
Overview
inject our instrument method into the Client class
Defined Under Namespace
Modules: RestApiSpec Classes: Bulk, Cluster, Docs, Error, Index, MultiPercolate, MultiSearch, NativeDeleteByQuery, Nodes, Percolator, Reindex, Repository, Scroller, Snapshot, Tasks, Template, UpdateByQuery
Constant Summary collapse
- IVAR_BLACK_LIST =
[:@basic_auth, :@token_auth]
- IVAR_NOISY_LIST =
[:@api_spec, :@cluster, :@connection]
- MAX_REQUEST_SIZE =
250 MB
2**20 * 250
- TimeoutError =
Wrapper classes for specific Faraday errors.
Class.new Error
- ConnectionFailed =
Class.new Error
- ResourceNotFound =
Class.new Error
- ParsingError =
Class.new Error
- SSLError =
Class.new Error
- ServerError =
Class.new Error
- RequestError =
Class.new Error
- RequestSizeError =
Class.new Error
- IndexNotFoundError =
Provide some nice errors for common Elasticsearch exceptions. These are all subclasses of the more general RequestError
Class.new RequestError
- QueryParsingError =
Class.new RequestError
- SearchContextMissing =
Class.new RequestError
- RejectedExecutionError =
Class.new RequestError
- DocumentAlreadyExistsError =
Class.new RequestError
- IncompatibleVersionException =
Exception for operations that are unsupported with the version of Elasticsearch being used.
Class.new Error
- IllegalArgument =
Exception for client-detected and server-raised invalid Elasticsearch request parameter.
Class.new Error
- DEFAULT_OPTS =
{ index: nil, type: nil, scroll: "5m", size: 50, }.freeze
- OpaqueIdError =
Error raised when a conflict is detected between the UUID sent in the ‘X-Opaque-Id’ request header and the one received in the response header.
Class.new Client::Error
Constants included from Notifications
Instance Attribute Summary collapse
-
#compress_body ⇒ Object
readonly
Returns the value of attribute compress_body.
-
#compression ⇒ Object
readonly
Returns the value of attribute compression.
-
#es_version ⇒ Object
readonly
Returns the value of attribute es_version.
-
#host ⇒ Object
readonly
Returns the value of attribute host.
-
#max_request_size ⇒ Object
readonly
Returns the value of attribute max_request_size.
-
#open_timeout ⇒ Object
readonly
Returns the value of attribute open_timeout.
-
#port ⇒ Object
readonly
Returns the value of attribute port.
-
#read_timeout ⇒ Object
readonly
Returns the value of attribute read_timeout.
-
#strict_params ⇒ Object
(also: #strict_params?)
readonly
Returns the value of attribute strict_params.
-
#url ⇒ Object
readonly
Returns the value of attribute url.
Instance Method Summary collapse
-
#add_sort_by_doc(query) ⇒ Object
Internal: Add sort by doc to query.
-
#api_spec ⇒ Object
Returns the ApiSpec for the specific version of Elasticsearch that this Client is connected to.
-
#assert_param_presence(param, name = "input value") ⇒ Object
Internal: Ensure that the parameter has a valid value.
-
#bulk(body = nil, params = nil) ⇒ Object
The ‘bulk` method can be used in two ways.
-
#bulk_stream_items(ops, params = {}) ⇒ Object
Stream bulk actions from an Enumerator and passes the response items to the given block.
-
#bulk_stream_responses(ops, params = {}) ⇒ Object
Stream bulk actions from an Enumerator.
-
#clear_scroll(scroll_ids) ⇒ Object
Delete one or more scroll IDs.
-
#cluster ⇒ Object
Returns a Cluster instance.
-
#connection ⇒ Object
Internal: Provides access to the Faraday::Connection used by this client for all requests to the server.
-
#continue_scroll(scroll_id, scroll = "5m") ⇒ Object
Continue scrolling a query.
-
#delete(path, params = {}) ⇒ Object
Internal: Sends an HTTP DELETE request to the server.
-
#delete_by_query(query, params = {}) ⇒ Object
Execute delete_by_query using the native _delete_by_query API if supported or the application-level implementation.
-
#docs(name = nil, type = nil) ⇒ Object
Provides access to document-level API commands.
-
#dup ⇒ Object
Returns a duplicate of this Client connection configured in the exact same fashion.
-
#expand_path(path, params) ⇒ Object
Internal: Apply path expansions to the ‘path` and append query parameters to the `path`.
-
#extract_body(params) ⇒ Object
Internal: Extract the :body from the params Hash and convert it to a JSON String format.
-
#get(path, params = {}) ⇒ Object
Internal: Sends an HTTP GET request to the server.
-
#handle_errors(response) ⇒ Object
Internal: Inspect the Faraday::Response and raise an error if the status is in the 5XX range or if the response body contains an ‘error’ field.
-
#head(path, params = {}) ⇒ Object
Internal: Sends an HTTP HEAD request to the server.
-
#index(name = nil) ⇒ Object
Provides access to index-level API commands.
-
#info ⇒ Object
Returns the information Hash from the attached Elasticsearch instance.
-
#initialize(host: "localhost", port: 9200, url: nil, read_timeout: 5, open_timeout: 2, opaque_id: false, adapter: Faraday.default_adapter, max_request_size: MAX_REQUEST_SIZE, strict_params: false, es_version: nil, compress_body: false, compression: Zlib::DEFAULT_COMPRESSION, basic_auth: nil, token_auth: nil, &block) ⇒ Client
constructor
Create a new client that can be used to make HTTP requests to the Elasticsearch server.
- #inspect ⇒ Object
-
#instrument(path, body, params) ⇒ Object
Internal: A noop method that simply yields to the block.
-
#is_ok?(item) ⇒ Boolean
Internal: Determine whether or not a response item has an HTTP status code in the range 200 to 299.
-
#multi_percolate(body = nil, params = nil) ⇒ Object
(also: #mpercolate)
Execute an array of percolate actions in bulk.
-
#multi_search(body = nil, params = nil) ⇒ Object
(also: #msearch)
Execute an array of searches in bulk.
-
#native_delete_by_query(query, parameters = {}) ⇒ Object
Delete documents based on a query using the Elasticsearch _delete_by_query API.
-
#nodes(node_id = nil) ⇒ Object
Provides access to node-level API commands.
-
#ping ⇒ Object
(also: #available?)
Returns true if the server is available; returns false otherwise.
-
#post(path, params = {}) ⇒ Object
Internal: Sends an HTTP POST request to the server.
-
#put(path, params = {}) ⇒ Object
Internal: Sends an HTTP PUT request to the server.
-
#reindex ⇒ Object
Returns a Reindex instance.
-
#repository(name = nil) ⇒ Object
Returns a Repository instance.
-
#request(method, path, params) ⇒ Object
Internal: Sends an HTTP request to the server.
-
#reset! ⇒ Object
Internal: Reset the client by removing the current Faraday::Connection.
-
#scan(query, opts = {}) ⇒ Object
Create a new Scroller instance for scrolling all results from a ‘query` via “scan” semantics by sorting by _doc.
-
#scroll(query, opts = {}) ⇒ Object
Create a new Scroller instance for scrolling all results from a ‘query`.
-
#semantic_version ⇒ Object
Returns a Semantic::Version for the attached Elasticsearch instance.
-
#snapshot(repository = nil, name = nil) ⇒ Object
Provides access to snapshot API commands.
-
#start_scroll(opts = {}) ⇒ Object
Begin scrolling a query.
-
#tasks ⇒ Object
Returns a Tasks instance for querying the cluster bound to this client for metadata about internal tasks in flight, and to submit administrative requests (like cancellation) concerning those tasks.
-
#template(name) ⇒ Object
Returns a Template instance.
-
#update_by_query(query, parameters = {}) ⇒ Object
Update documents based on a query using the Elasticsearch _update_by_query API.
-
#version ⇒ Object
Returns the version String of the attached Elasticsearch instance.
- #version_support ⇒ Object
-
#wrap_faraday_error(error, method, path) ⇒ Object
Internal: Returns a new ElastomerClient::Client error that wraps the given Faraday error.
Constructor Details
#initialize(host: "localhost", port: 9200, url: nil, read_timeout: 5, open_timeout: 2, opaque_id: false, adapter: Faraday.default_adapter, max_request_size: MAX_REQUEST_SIZE, strict_params: false, es_version: nil, compress_body: false, compression: Zlib::DEFAULT_COMPRESSION, basic_auth: nil, token_auth: nil, &block) ⇒ Client
Create a new client that can be used to make HTTP requests to the Elasticsearch server.
see lib/elastomer_client/client/errors.rb#L92-L94
Method params:
:host - the host as a String
:port - the port number of the server
:url - the URL as a String (overrides :host and :port)
:read_timeout - the timeout in seconds when reading from an HTTP connection
:open_timeout - the timeout in seconds when opening an HTTP connection
:adapter - the Faraday adapter to use (defaults to :excon)
:opaque_id - set to `true` to use the 'X-Opaque-Id' request header
:max_request_size - the maximum allowed request size in bytes (defaults to 250 MB)
:strict_params - set to `true` to raise exceptions when invalid request params are used
:es_version - set to the Elasticsearch version (example: "5.6.6") to avoid an HTTP request to get the version.
:compress_body - set to true to enable request body compression (default: false)
:compression - The compression level (0-9) when request body compression is enabled (default: Zlib::DEFAULT_COMPRESSION)
:basic_auth - a Hash containing basic authentication :username and :password values to use on connections
:token_auth - an authentication token as a String to use on connections (overrides :basic_auth)
42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 |
# File 'lib/elastomer_client/client.rb', line 42 def initialize(host: "localhost", port: 9200, url: nil, read_timeout: 5, open_timeout: 2, opaque_id: false, adapter: Faraday.default_adapter, max_request_size: MAX_REQUEST_SIZE, strict_params: false, es_version: nil, compress_body: false, compression: Zlib::DEFAULT_COMPRESSION, basic_auth: nil, token_auth: nil, &block) @url = url || "http://#{host}:#{port}" uri = Addressable::URI.parse @url @host = uri.host @port = uri.port @read_timeout = read_timeout @open_timeout = open_timeout @adapter = adapter @opaque_id = opaque_id @max_request_size = max_request_size @strict_params = strict_params @es_version = es_version @compress_body = compress_body @compression = compression @basic_auth = basic_auth @token_auth = token_auth @connection_block = block end |
Instance Attribute Details
#compress_body ⇒ Object (readonly)
Returns the value of attribute compress_body.
72 73 74 |
# File 'lib/elastomer_client/client.rb', line 72 def compress_body @compress_body end |
#compression ⇒ Object (readonly)
Returns the value of attribute compression.
73 74 75 |
# File 'lib/elastomer_client/client.rb', line 73 def compression @compression end |
#es_version ⇒ Object (readonly)
Returns the value of attribute es_version.
71 72 73 |
# File 'lib/elastomer_client/client.rb', line 71 def es_version @es_version end |
#host ⇒ Object (readonly)
Returns the value of attribute host.
67 68 69 |
# File 'lib/elastomer_client/client.rb', line 67 def host @host end |
#max_request_size ⇒ Object (readonly)
Returns the value of attribute max_request_size.
69 70 71 |
# File 'lib/elastomer_client/client.rb', line 69 def max_request_size @max_request_size end |
#open_timeout ⇒ Object (readonly)
Returns the value of attribute open_timeout.
68 69 70 |
# File 'lib/elastomer_client/client.rb', line 68 def open_timeout @open_timeout end |
#port ⇒ Object (readonly)
Returns the value of attribute port.
67 68 69 |
# File 'lib/elastomer_client/client.rb', line 67 def port @port end |
#read_timeout ⇒ Object (readonly)
Returns the value of attribute read_timeout.
68 69 70 |
# File 'lib/elastomer_client/client.rb', line 68 def read_timeout @read_timeout end |
#strict_params ⇒ Object (readonly) Also known as: strict_params?
Returns the value of attribute strict_params.
70 71 72 |
# File 'lib/elastomer_client/client.rb', line 70 def strict_params @strict_params end |
#url ⇒ Object (readonly)
Returns the value of attribute url.
67 68 69 |
# File 'lib/elastomer_client/client.rb', line 67 def url @url end |
Instance Method Details
#add_sort_by_doc(query) ⇒ Object
Internal: Add sort by doc to query.
Raises an exception if the query contains a sort already. Returns the query as a hash
126 127 128 129 130 131 132 133 134 135 136 137 138 |
# File 'lib/elastomer_client/client/scroller.rb', line 126 def add_sort_by_doc(query) if query.nil? query = {} elsif query.is_a? String query = MultiJson.load(query) end if query.has_key? :sort raise ArgumentError, "Query cannot contain a sort (found sort '#{query[:sort]}' in query: #{query})" end query.merge(sort: [:_doc]) end |
#api_spec ⇒ Object
Returns the ApiSpec for the specific version of Elasticsearch that this Client is connected to.
123 124 125 |
# File 'lib/elastomer_client/client.rb', line 123 def api_spec @api_spec ||= RestApiSpec.api_spec(version) end |
#assert_param_presence(param, name = "input value") ⇒ Object
Internal: Ensure that the parameter has a valid value. Strings, Symbols, Numerics, and Arrays of those things are valid. Things like ‘nil` and empty strings are right out. This method also performs a little formating on the parameter:
-
leading and trailing whitespace is removed
-
arrays are flattend
-
and then joined into a String
-
numerics are converted to their string equivalents
param - The param Object to validate name - Optional param name as a String (used in exception messages)
Returns the validated param as a String. Raises an ArgumentError if the param is not valid.
445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 |
# File 'lib/elastomer_client/client.rb', line 445 def assert_param_presence(param, name = "input value") case param when String, Symbol, Numeric param = param.to_s.strip raise ArgumentError, "#{name} cannot be blank: #{param.inspect}" if param =~ /\A\s*\z/ param when Array param.flatten.map { |item| assert_param_presence(item, name) }.join(",") when nil raise ArgumentError, "#{name} cannot be nil" else raise ArgumentError, "#{name} is invalid: #{param.inspect}" end end |
#bulk(body = nil, params = nil) ⇒ Object
The ‘bulk` method can be used in two ways. Without a block the method will perform an API call, and it requires a bulk request body and optional request parameters. If given a block, the method will use a Bulk instance to assemble the operations called in the block into a bulk request and dispatch it at the end of the block.
See www.elastic.co/guide/en/elasticsearch/reference/current/docs-bulk.html
body - Request body as a String (required if a block is not given) params - Optional request parameters as a Hash
:request_size - Optional maximum request size in bytes
:action_count - Optional maximum action size
block - Passed to a Bulk instance which assembles the operations
into one or more bulk requests.
Examples
bulk(request_body, :index => 'default-index')
bulk(:index => 'default-index') do |b|
b.index(document1)
b.index(document2, :_type => 'default-type')
b.delete(document3)
...
end
Returns the response body as a Hash
33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
# File 'lib/elastomer_client/client/bulk.rb', line 33 def bulk(body = nil, params = nil) if block_given? params, body = (body || {}), nil yield bulk_obj = Bulk.new(self, params) bulk_obj.call else raise "bulk request body cannot be nil" if body.nil? params ||= {} updated_params = params.merge(body:, action: "bulk", rest_api: "bulk") updated_params.delete(:type) if version_support.es_version_8_plus? response = self.post "{/index}{/type}/_bulk", updated_params response.body end end |
#bulk_stream_items(ops, params = {}) ⇒ Object
Stream bulk actions from an Enumerator and passes the response items to the given block.
Examples
ops = [
[:index, document1, {:_type => "foo", :_id => 1}],
[:create, document2],
[:delete, {:_type => "bar", :_id => 42}]
]
bulk_stream_items(ops, :index => 'default-index') do |item|
puts item
end
# return value:
# {
# "took" => 256,
# "errors" => false,
# "success" => 3,
# "failure" => 0
# }
# sample response item for ES5:
# {
# "delete": {
# "_index": "foo",
# "_type": "bar",
# "_id": "42",
# "_version": 3,
# "status": 200,
# "found": true
# }
# }
# sample response item for ES8:
# {
# "delete": {
# "_index": "foo",
# "_id": "42",
# "_version": 3,
# "status": 200,
# "result": "deleted"
# }
# }
Returns a Hash of stats about items from the responses.
134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 |
# File 'lib/elastomer_client/client/bulk.rb', line 134 def bulk_stream_items(ops, params = {}) stats = { "took" => 0, "errors" => false, "success" => 0, "failure" => 0 } bulk_stream_responses(ops, params).each do |response| stats["took"] += response["took"] stats["errors"] |= response["errors"] response["items"].each do |item| if is_ok?(item) stats["success"] += 1 else stats["failure"] += 1 end yield item end end stats end |
#bulk_stream_responses(ops, params = {}) ⇒ Object
Stream bulk actions from an Enumerator.
Examples
ops = [
[:index, document1, {:_type => "foo", :_id => 1}],
[:create, document2],
[:delete, {:_type => "bar", :_id => 42}]
]
bulk_stream_responses(ops, :index => 'default-index').each do |response|
puts response
end
Returns an Enumerator of responses.
64 65 66 67 68 69 70 71 72 73 74 75 76 |
# File 'lib/elastomer_client/client/bulk.rb', line 64 def bulk_stream_responses(ops, params = {}) bulk_obj = Bulk.new(self, params) Enumerator.new do |yielder| ops.each do |action, *args| response = bulk_obj.send(action, *args) yielder.yield response unless response.nil? end response = bulk_obj.call yielder.yield response unless response.nil? end end |
#clear_scroll(scroll_ids) ⇒ Object
Delete one or more scroll IDs. see www.elastic.co/guide/en/elasticsearch/reference/current/search-request-scroll.html#_clear_scroll_api
scroll_id - One or more scroll IDs
Returns the response body as a Hash.
117 118 119 120 |
# File 'lib/elastomer_client/client/scroller.rb', line 117 def clear_scroll(scroll_ids) response = delete "/_search/scroll", body: {scroll_id: Array(scroll_ids)}, action: "search.clear_scroll", rest_api: "clear_scroll" response.body end |
#cluster ⇒ Object
Returns a Cluster instance.
7 8 9 |
# File 'lib/elastomer_client/client/cluster.rb', line 7 def cluster @cluster ||= Cluster.new self end |
#connection ⇒ Object
Internal: Provides access to the Faraday::Connection used by this client for all requests to the server.
Returns a Faraday::Connection
131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 |
# File 'lib/elastomer_client/client.rb', line 131 def connection @connection ||= Faraday.new(url) do |conn| conn.response(:parse_json) # Request compressed responses from ES and decompress them conn.use(:gzip) conn.request(:encode_json) conn.request(:limit_size, max_request_size:) if max_request_size conn.request(:elastomer_compress, compression:) if compress_body conn.[:timeout] = read_timeout conn.[:open_timeout] = open_timeout if token_auth? conn.token_auth(@token_auth) elsif basic_auth? conn.basic_auth(@basic_auth[:username], @basic_auth[:password]) end @connection_block&.call(conn) conn.request(:opaque_id) if @opaque_id if @adapter.is_a?(Array) conn.adapter(*@adapter) else conn.adapter(@adapter) end end end |
#continue_scroll(scroll_id, scroll = "5m") ⇒ Object
Continue scrolling a query. See www.elastic.co/guide/en/elasticsearch/reference/current/search-request-scroll.html
scroll_id - The current scroll ID as a String scroll - The keep alive time of the scrolling request (5 minutes by default)
Examples
scroll_id = client.start_scroll(body: '{"query":{"match_all":{}}}', index: 'test')['_scroll_id']
h = client.continue_scroll scroll_id # scroll to get the next set of results
scroll_id = h['_scroll_id'] # and store the scroll_id to use later
h = client.continue_scroll scroll_id # scroll again to get the next set of results
scroll_id = h['_scroll_id'] # and store the scroll_id to use later
# repeat until the results are empty
Returns the response body as a Hash.
100 101 102 103 104 105 106 107 108 109 |
# File 'lib/elastomer_client/client/scroller.rb', line 100 def continue_scroll(scroll_id, scroll = "5m") response = get "/_search/scroll", body: {scroll_id:}, scroll:, action: "search.scroll", rest_api: "scroll" response.body rescue RequestError => err if err.error && err.error["caused_by"]["type"] == "search_context_missing_exception" raise SearchContextMissing, "No search context found for scroll ID #{scroll_id.inspect}" else raise err end end |
#delete(path, params = {}) ⇒ Object
Internal: Sends an HTTP DELETE request to the server.
path - The path as a String params - Parameters Hash
Returns a Faraday::Response Raises an ElastomerClient::Client::Error on 4XX and 5XX responses
220 221 222 |
# File 'lib/elastomer_client/client.rb', line 220 def delete(path, params = {}) request :delete, path, params end |
#delete_by_query(query, params = {}) ⇒ Object
Execute delete_by_query using the native _delete_by_query API if supported or the application-level implementation.
Warning: These implementations have different parameters and return types. If you want to use one or the other consistently, use ElastomerClient::Client#native_delete_by_query or ElastomerClient::Client#app_delete_by_query directly.
11 12 13 |
# File 'lib/elastomer_client/client/delete_by_query.rb', line 11 def delete_by_query(query, params = {}) send(:native_delete_by_query, query, params) end |
#docs(name = nil, type = nil) ⇒ Object
Provides access to document-level API commands. Indexing documents and searching documents are both handled by this module.
name - The name of the index as a String (optional) type - The document type as a String (optional)
See www.elastic.co/guide/en/elasticsearch/reference/current/docs.html
Returns a Docs instance.
15 16 17 |
# File 'lib/elastomer_client/client/docs.rb', line 15 def docs(name = nil, type = nil) Docs.new self, name, type end |
#dup ⇒ Object
Returns a duplicate of this Client connection configured in the exact same fashion.
78 79 80 81 82 83 84 85 86 87 88 |
# File 'lib/elastomer_client/client.rb', line 78 def dup self.class.new \ url:, read_timeout:, open_timeout:, adapter: @adapter, opaque_id: @opaque_id, max_request_size:, basic_auth: @basic_auth, token_auth: @token_auth end |
#expand_path(path, params) ⇒ Object
Internal: Apply path expansions to the ‘path` and append query parameters to the `path`. We are using an Addressable::Template to replace ’expansion‘ fields found in the path with the values extracted from the `params` Hash. Any remaining elements in the `params` hash are treated as query parameters and appended to the end of the path.
path - The path as a String params - Parameters Hash
Examples
expand_path('/foo{/bar}', {bar: 'hello', q: 'what', p: 2})
#=> '/foo/hello?q=what&p=2'
expand_path('/foo{/bar}{/baz}', {baz: 'no bar'}
#=> '/foo/no%20bar'
Returns an Addressable::Uri
351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 |
# File 'lib/elastomer_client/client.rb', line 351 def (path, params) template = Addressable::Template.new path expansions = {} query_values = params.dup query_values.delete :action query_values.delete :context rest_api = query_values.delete :rest_api template.keys.map(&:to_sym).each do |key| value = query_values.delete key value = assert_param_presence(value, key) unless path =~ /{\/#{key}}/ && value.nil? expansions[key] = value end if rest_api query_values = if strict_params? api_spec.validate_params!(api: rest_api, params: query_values) else api_spec.select_params(api: rest_api, from: query_values) end end uri = template.(expansions) query_values.transform_keys!(&:to_s) uri.query_values = (uri.query_values || {}).merge(query_values) unless query_values.empty? uri.to_s end |
#extract_body(params) ⇒ Object
Internal: Extract the :body from the params Hash and convert it to a JSON String format. If the params Hash does not contain a :body then no action is taken and ‘nil` is returned.
If a :body is present and is a String then it is assumed to a JSON String and returned “as is”.
If a :body is present and is an Array then we join the values together with newlines and append a trailing newline. This is a special case for dealing with ES ‘bulk` imports and `multi_search` methods.
Otherwise we convert the :body to a JSON string and return.
params - Parameters Hash
Returns the request body as a String or ‘nil` if no :body is present
314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 |
# File 'lib/elastomer_client/client.rb', line 314 def extract_body(params) body = params.delete :body return if body.nil? body = case body when String body when Array body << nil unless body.last.nil? body.join "\n" else MultiJson.dump body end # Prevent excon from changing the encoding (see https://github.com/github/elastomer-client/issues/138) body.freeze end |
#get(path, params = {}) ⇒ Object
Internal: Sends an HTTP GET request to the server.
path - The path as a String params - Parameters Hash
Returns a Faraday::Response Raises an ElastomerClient::Client::Error on 4XX and 5XX responses
187 188 189 |
# File 'lib/elastomer_client/client.rb', line 187 def get(path, params = {}) request :get, path, params end |
#handle_errors(response) ⇒ Object
Internal: Inspect the Faraday::Response and raise an error if the status is in the 5XX range or if the response body contains an ‘error’ field. In the latter case, the value of the ‘error’ field becomes our exception message. In the absence of an ‘error’ field the response body is used as the exception message.
The raised exception will contain the response object.
response - The Faraday::Response object.
Returns the response. Raises an ElastomerClient::Client::Error on 500 responses or responses containing and ‘error’ field.
408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 |
# File 'lib/elastomer_client/client.rb', line 408 def handle_errors(response) raise ServerError, response if response.status >= 500 if response.body.is_a?(Hash) && (error = response.body["error"]) root_cause = Array(error["root_cause"]).first || error case root_cause["type"] when "index_not_found_exception"; raise IndexNotFoundError, response when "illegal_argument_exception"; raise IllegalArgument, response when "es_rejected_execution_exception"; raise RejectedExecutionError, response # Elasticsearch 2.x.x root_cause type for document already existing when "document_already_exists_exception"; raise DocumentAlreadyExistsError, response # Elasticsearch 5.x.x root_cause type for document already existing when "version_conflict_engine_exception"; raise DocumentAlreadyExistsError, response when "query_shard_exception", "parsing_exception"; raise QueryParsingError, response end raise RequestError, response end response end |
#head(path, params = {}) ⇒ Object
Internal: Sends an HTTP HEAD request to the server.
path - The path as a String params - Parameters Hash
Returns a Faraday::Response
176 177 178 |
# File 'lib/elastomer_client/client.rb', line 176 def head(path, params = {}) request :head, path, params end |
#index(name = nil) ⇒ Object
Provides access to index-level API commands. An index name is required for these API calls. If you want to operate on all indices - flushing all indices, for example - then you will need to use the “_all” index name.
You can override the index name for one-off calls by passing in the desired index name via the ‘:index` option.
name - The name of the index as a String or an Array of names
Returns an Index instance.
16 17 18 |
# File 'lib/elastomer_client/client/index.rb', line 16 def index(name = nil) Index.new self, name end |
#info ⇒ Object
Returns the information Hash from the attached Elasticsearch instance.
116 117 118 119 |
# File 'lib/elastomer_client/client.rb', line 116 def info response = get "/", action: "cluster.info" response.body end |
#inspect ⇒ Object
467 468 469 470 471 472 473 474 |
# File 'lib/elastomer_client/client.rb', line 467 def inspect public_vars = self.instance_variables.reject do |var| IVAR_NOISY_LIST.include?(var) end.map do |var| "#{var}=#{IVAR_BLACK_LIST.include?(var) ? "[FILTERED]" : instance_variable_get(var).inspect}" end.join(", ") "<##{self.class}:#{self.object_id.to_s(16)} #{public_vars}>" end |
#instrument(path, body, params) ⇒ Object
Internal: A noop method that simply yields to the block. This method will be replaced when the ‘elastomer_client/notifications’ module is included.
path - The full request path as a String body - The request body as a String or ‘nil` params - The request params Hash block - The block that will be instrumented
Returns the response from the block
391 392 393 |
# File 'lib/elastomer_client/client.rb', line 391 def instrument(path, body, params) yield end |
#is_ok?(item) ⇒ Boolean
Internal: Determine whether or not a response item has an HTTP status code in the range 200 to 299.
item - The bulk response item
Returns a boolean
84 85 86 |
# File 'lib/elastomer_client/client/bulk.rb', line 84 def is_ok?(item) item.values.first["status"].between?(200, 299) end |
#multi_percolate(body = nil, params = nil) ⇒ Object Also known as: mpercolate
Execute an array of percolate actions in bulk. Results are returned in an array in the order the actions were sent.
The ‘multi_percolate` method can be used in two ways. Without a block the method will perform an API call, and it requires a bulk request body and optional request parameters.
See www.elastic.co/guide/en/elasticsearch/reference/current/search-percolate.html#_multi_percolate_api
body - Request body as a String (required if a block is not given) params - Optional request parameters as a Hash block - Passed to a MultiPercolate instance which assembles the
percolate actions into a single request.
Examples
# index and type in request body
multi_percolate(request_body)
# index in URI
multi_percolate(request_body, index: 'default-index')
# block form
multi_percolate(index: 'default-index') do |m|
m.percolate({ author: "pea53" }, { type: 'default-type' })
m.count({ author: "pea53" }, { type: 'type2' })
...
end
Returns the response body as a Hash
36 37 38 39 40 41 42 43 44 45 46 47 48 |
# File 'lib/elastomer_client/client/multi_percolate.rb', line 36 def multi_percolate(body = nil, params = nil) if block_given? params, body = (body || {}), nil yield mpercolate_obj = MultiPercolate.new(self, params) mpercolate_obj.call else raise "multi_percolate request body cannot be nil" if body.nil? params ||= {} response = self.post "{/index}{/type}/_mpercolate", params.merge(body:, action: "mpercolate", rest_api: "mpercolate") response.body end end |
#multi_search(body = nil, params = nil) ⇒ Object Also known as: msearch
Execute an array of searches in bulk. Results are returned in an array in the order the queries were sent.
The ‘multi_search` method can be used in two ways. Without a block the method will perform an API call, and it requires a bulk request body and optional request parameters.
See www.elastic.co/guide/en/elasticsearch/reference/current/search-multi-search.html
body - Request body as a String (required if a block is not given) params - Optional request parameters as a Hash block - Passed to a MultiSearch instance which assembles the searches
into a single request.
Examples
# index and type in request body
multi_search(request_body)
# index in URI
multi_search(request_body, index: 'default-index')
# block form
multi_search(index: 'default-index') do |m|
m.search({query: {match_all: {}}, size: 0)
m.search({query: {field: {"foo" => "bar"}}}, type: 'default-type')
...
end
Returns the response body as a Hash
36 37 38 39 40 41 42 43 44 45 46 47 48 |
# File 'lib/elastomer_client/client/multi_search.rb', line 36 def multi_search(body = nil, params = nil) if block_given? params, body = (body || {}), nil yield msearch_obj = MultiSearch.new(self, params) msearch_obj.call else raise "multi_search request body cannot be nil" if body.nil? params ||= {} response = self.post "{/index}{/type}/_msearch", params.merge(body:, action: "msearch", rest_api: "msearch") response.body end end |
#native_delete_by_query(query, parameters = {}) ⇒ Object
Delete documents based on a query using the Elasticsearch _delete_by_query API.
query - The query body as a Hash params - Parameters Hash
Examples
# request body query
native_delete_by_query({query: {match_all: {}}}, type: 'tweet')
See www.elastic.co/guide/en/elasticsearch/reference/5.6/docs-delete-by-query.html
Returns a Hash containing the _delete_by_query response body.
18 19 20 |
# File 'lib/elastomer_client/client/native_delete_by_query.rb', line 18 def native_delete_by_query(query, parameters = {}) NativeDeleteByQuery.new(self, query, parameters).execute end |
#nodes(node_id = nil) ⇒ Object
Provides access to node-level API commands. The default node is set to nil which target all nodes. You can pass in “_all” (to get the same effect) or “_local” to target only the current node the client is connected to. And you can specify an individual node or multiple nodes.
node_id - The node ID as a String or an Array of node IDs
Returns a Nodes instance.
14 15 16 |
# File 'lib/elastomer_client/client/nodes.rb', line 14 def nodes(node_id = nil) Nodes.new self, node_id end |
#ping ⇒ Object Also known as: available?
Returns true if the server is available; returns false otherwise.
91 92 93 94 95 96 |
# File 'lib/elastomer_client/client.rb', line 91 def ping response = head "/", action: "cluster.ping" response.success? rescue StandardError false end |
#post(path, params = {}) ⇒ Object
Internal: Sends an HTTP POST request to the server.
path - The path as a String params - Parameters Hash
Returns a Faraday::Response Raises an ElastomerClient::Client::Error on 4XX and 5XX responses
209 210 211 |
# File 'lib/elastomer_client/client.rb', line 209 def post(path, params = {}) request :post, path, params end |
#put(path, params = {}) ⇒ Object
Internal: Sends an HTTP PUT request to the server.
path - The path as a String params - Parameters Hash
Returns a Faraday::Response Raises an ElastomerClient::Client::Error on 4XX and 5XX responses
198 199 200 |
# File 'lib/elastomer_client/client.rb', line 198 def put(path, params = {}) request :put, path, params end |
#reindex ⇒ Object
Returns a Reindex instance
7 8 9 |
# File 'lib/elastomer_client/client/reindex.rb', line 7 def reindex Reindex.new(self) end |
#repository(name = nil) ⇒ Object
Returns a Repository instance.
7 8 9 |
# File 'lib/elastomer_client/client/repository.rb', line 7 def repository(name = nil) Repository.new(self, name) end |
#request(method, path, params) ⇒ Object
Internal: Sends an HTTP request to the server. If the ‘params` Hash contains a :body key, it will be deleted from the Hash and the value will be used as the body of the request.
method - The HTTP method to send [:head, :get, :put, :post, :delete] path - The path as a String params - Parameters Hash
:body - Will be used as the request body
:read_timeout - Optional read timeout (in seconds) for the request
Returns a Faraday::Response Raises an ElastomerClient::Client::Error on 4XX and 5XX responses
236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 |
# File 'lib/elastomer_client/client.rb', line 236 def request(method, path, params) read_timeout = params.delete(:read_timeout) body = extract_body(params) path = (path, params) instrument(path, body, params) do begin response = case method when :head connection.head(path) { |req| req.[:timeout] = read_timeout if read_timeout } when :get connection.get(path) { |req| req.body = body if body req.[:timeout] = read_timeout if read_timeout } when :put connection.put(path, body) { |req| req.[:timeout] = read_timeout if read_timeout } when :post connection.post(path, body) { |req| req.[:timeout] = read_timeout if read_timeout } when :delete connection.delete(path) { |req| req.body = body if body req.[:timeout] = read_timeout if read_timeout } else raise ArgumentError, "unknown HTTP request method: #{method.inspect}" end handle_errors response # wrap Faraday errors with appropriate ElastomerClient::Client error classes rescue Faraday::Error => boom error = wrap_faraday_error(boom, method, path) raise error rescue OpaqueIdError => boom reset! raise boom end end end |
#reset! ⇒ Object
Internal: Reset the client by removing the current Faraday::Connection. A new connection will be established on the next request.
Returns this Client instance.
165 166 167 168 |
# File 'lib/elastomer_client/client.rb', line 165 def reset! @connection = nil self end |
#scan(query, opts = {}) ⇒ Object
Create a new Scroller instance for scrolling all results from a ‘query` via “scan” semantics by sorting by _doc.
query - The query to scan as a Hash or a JSON encoded String opts - Options Hash
:index - the name of the index to search
:type - the document type to search
:scroll - the keep alive time of the scrolling request (5 minutes by default)
:size - the number of documents per shard to fetch per scroll
Examples
scan = client.scan('{"query":{"match_all":{}}}', index: 'test')
scan.each_document do |document|
document['_id']
document['_source']
end
Returns a new Scroller instance
47 48 49 |
# File 'lib/elastomer_client/client/scroller.rb', line 47 def scan(query, opts = {}) Scroller.new(self, add_sort_by_doc(query), opts) end |
#scroll(query, opts = {}) ⇒ Object
Create a new Scroller instance for scrolling all results from a ‘query`.
query - The query to scroll as a Hash or a JSON encoded String opts - Options Hash
:index - the name of the index to search
:type - the document type to search
:scroll - the keep alive time of the scrolling request (5 minutes by default)
:size - the number of documents per shard to fetch per scroll
Examples
scroll = client.scroll('{"query":{"match_all":{}}}', index: 'test')
scroll.each_document do |document|
document['_id']
document['_source']
end
Returns a new Scroller instance
24 25 26 |
# File 'lib/elastomer_client/client/scroller.rb', line 24 def scroll(query, opts = {}) Scroller.new(self, query, opts) end |
#semantic_version ⇒ Object
Returns a Semantic::Version for the attached Elasticsearch instance. See rubygems.org/gems/semantic
111 112 113 |
# File 'lib/elastomer_client/client.rb', line 111 def semantic_version Semantic::Version.new(version) end |
#snapshot(repository = nil, name = nil) ⇒ Object
Provides access to snapshot API commands.
repository - The name of the repository as a String name - The name of the snapshot as a String
Returns a Snapshot instance.
12 13 14 |
# File 'lib/elastomer_client/client/snapshot.rb', line 12 def snapshot(repository = nil, name = nil) Snapshot.new self, repository, name end |
#start_scroll(opts = {}) ⇒ Object
Begin scrolling a query. See www.elastic.co/guide/en/elasticsearch/reference/current/search-request-scroll.html
opts - Options Hash
:body - the query to scroll as a Hash or JSON encoded String
:index - the name of the index to search
:type - the document type to search
:scroll - the keep alive time of the scrolling request (5 minutes by default)
:size - the number of documents per shard to fetch per scroll
Examples
h = client.start_scroll(body: '{"query":{"match_all":{}},"sort":{"created":"desc"}}', index: 'test')
scroll_id = h['_scroll_id']
h['hits']['hits'].each { |doc| ... }
h = client.continue_scroll(scroll_id)
scroll_id = h['_scroll_id']
h['hits']['hits'].each { |doc| ... }
# repeat until there are no more hits
Returns the response body as a Hash.
74 75 76 77 78 79 |
# File 'lib/elastomer_client/client/scroller.rb', line 74 def start_scroll(opts = {}) opts = opts.merge action: "search.start_scroll", rest_api: "search" opts.delete(:type) if version_support.es_version_8_plus? response = get "{/index}{/type}/_search", opts response.body end |
#tasks ⇒ Object
Returns a Tasks instance for querying the cluster bound to this client for metadata about internal tasks in flight, and to submit administrative requests (like cancellation) concerning those tasks.
Returns a new Tasks object associated with this client
11 12 13 |
# File 'lib/elastomer_client/client/tasks.rb', line 11 def tasks Tasks.new(self) end |
#template(name) ⇒ Object
Returns a Template instance.
7 8 9 |
# File 'lib/elastomer_client/client/template.rb', line 7 def template(name) Template.new self, name end |
#update_by_query(query, parameters = {}) ⇒ Object
Update documents based on a query using the Elasticsearch _update_by_query API.
query - The query body as a Hash params - Parameters Hash
Examples
# request body query
update_by_query({
"script": {
"source": "ctx._source.count++",
"lang": "painless"
},
"query": {
"term": {
"user.id": "kimchy"
}
}
})
See www.elastic.co/guide/en/elasticsearch/reference/8.7/docs-update-by-query.html
Returns a Hash containing the _update_by_query response body.
28 29 30 |
# File 'lib/elastomer_client/client/update_by_query.rb', line 28 def update_by_query(query, parameters = {}) UpdateByQuery.new(self, query, parameters).execute end |
#version ⇒ Object
Returns the version String of the attached Elasticsearch instance.
100 101 102 103 104 105 106 107 |
# File 'lib/elastomer_client/client.rb', line 100 def version return es_version unless es_version.nil? @version ||= begin response = get "/" response.body.dig("version", "number") end end |
#version_support ⇒ Object
463 464 465 |
# File 'lib/elastomer_client/client.rb', line 463 def version_support @version_support ||= VersionSupport.new(version) end |
#wrap_faraday_error(error, method, path) ⇒ Object
Internal: Returns a new ElastomerClient::Client error that wraps the given Faraday error. A generic Error is returned if we cannot wrap the given Faraday error.
error - The Faraday error
method - The request method
path - The request path
292 293 294 295 296 |
# File 'lib/elastomer_client/client.rb', line 292 def wrap_faraday_error(error, method, path) error_name = error.class.name.split("::").last error_class = ElastomerClient::Client.const_get(error_name) rescue ElastomerClient::Client::Error error_class.new(error, method.upcase, path) end |