Implements X-Opaque-Id

Author: picandocodigo

elasticsearch-transport/lib/elasticsearch/transport/client.rb

@@ -146,9 +146,16 @@ def initialize(arguments={}, &block)
       #
       def perform_request(method, path, params = {}, body = nil, headers = nil)
         method = @send_get_body_as if 'GET' == method && body
+        if @opaque_id
+          headers = {} if headers.nil?
+          headers.merge!('X-Opaque-Id' => @opaque_id)
+          @opaque_id = nil # Remove Opaque id after each request
+        end
         transport.perform_request(method, path, params, body, headers)
       end
 
+      attr_accessor :opaque_id
+
       private
 
       def set_api_key

elasticsearch-transport/spec/elasticsearch/transport/client_spec.rb

@@ -652,6 +652,21 @@
         expect(request).to be(true)
       end
     end
+
+    context 'when x-opaque-id is set before calling a method' do
+      let(:client) { described_class.new(host: hosts) }
+
+      it 'uses x-opaque-id on a request' do
+        client.opaque_id = '12345'
+        expect(client.perform_request('GET', '/').headers['x-opaque-id']).to eq('12345')
+      end
+
+      it 'deletes x-opaque-id on a second request' do
+        client.opaque_id = 'asdfg'
+        expect(client.perform_request('GET', '/').headers['x-opaque-id']).to eq('asdfg')
+        expect(client.perform_request('GET', '/').headers).not_to include('x-opaque-id')
+      end
+    end
   end
 
   context 'when the client connects to Elasticsearch' do