Sha256: 4cae5fcea6d022aebbcd91609e46be01bebc30f27660a4261f2e71367dad6a28

Contents?: true

Size: 1.94 KB

Versions: 6

Compression:

Stored size: 1.94 KB

Contents

# Copyright 2010 Sean Cribbs, Sonian Inc., and Basho Technologies, Inc.
#
#    Licensed under the Apache License, Version 2.0 (the "License");
#    you may not use this file except in compliance with the License.
#    You may obtain a copy of the License at
#
#        http://www.apache.org/licenses/LICENSE-2.0
#
#    Unless required by applicable law or agreed to in writing, software
#    distributed under the License is distributed on an "AS IS" BASIS,
#    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#    See the License for the specific language governing permissions and
#    limitations under the License.
require 'riak'

module Riak
  class Client
    # Uses the Ruby standard library Net::HTTP to connect to Riak.
    # We recommend using the CurbBackend, which will
    # be preferred when the 'curb' library is available.
    # Conforms to the Riak::Client::HTTPBackend interface.
    class NetHTTPBackend < HTTPBackend
      private
      def perform(method, uri, headers, expect, data=nil) #:nodoc:
        Net::HTTP.start(uri.host, uri.port) do |http|
          request = Net::HTTP.const_get(method.to_s.camelize).new(uri.request_uri, headers)
          case data
          when String
            request.body = data
          when IO
            request.body_stream = data
          end

          {}.tap do |result|
            http.request(request) do |response|
              if valid_response?(expect, response.code)
                result.merge!({:headers => response.to_hash, :code => response.code.to_i})
                response.read_body {|chunk| yield chunk } if block_given?
                if return_body?(method, response.code, block_given?)
                  result[:body] = response.body
                end
              else
                raise FailedRequest.new(method, expect, response.code, response.to_hash, response.body)
              end
            end
          end
        end
      end
    end
  end
end

Version data entries

6 entries across 6 versions & 1 rubygems

Version Path
riak-client-0.8.2 lib/riak/client/net_http_backend.rb
riak-client-0.8.1 lib/riak/client/net_http_backend.rb
riak-client-0.8.0 lib/riak/client/net_http_backend.rb
riak-client-0.8.0.beta2 lib/riak/client/net_http_backend.rb
riak-client-0.8.0.beta lib/riak/client/net_http_backend.rb
riak-client-0.7.1 lib/riak/client/net_http_backend.rb