Class: Excon::Connection

Inherits:
Object
  • Object
show all
Defined in:
lib/excon/connection.rb

Instance Method Summary collapse

Constructor Details

#initialize(url) ⇒ Connection

Returns a new instance of Connection.



4
5
6
7
 
# File 'lib/excon/connection.rb', line 4

def initialize(url)
  @uri = URI.parse(url)
  reset_socket
end

Instance Method Details

#request(params) ⇒ Object 



9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
 
# File 'lib/excon/connection.rb', line 9

def request(params)
  begin
    params[:path] ||= @uri.path
    unless params[:path][0..0] == '/'
      params[:path] = "/#{params[:path]}"
    end
    if (params[:query] && !params[:query].empty?) || @uri.query
      params[:path] << "?#{params[:query]}"
    end
    request = "#{params[:method]} #{params[:path]} HTTP/1.1\r\n"
    params[:headers] ||= {}
    params[:headers]['Host'] = params[:host] || @uri.host
    if params[:body] && !params[:headers]['Content-Length']
      params[:headers]['Content-Length'] = params[:body].length
    end
    for key, value in params[:headers]
      request << "#{key}: #{value}\r\n"
    end
    request << "\r\n"
    socket.write(request)

    if params[:body]
      if params[:body].is_a?(String)
        socket.write(params[:body])
      else
        while chunk = params[:body].read(CHUNK_SIZE)
          socket.write(chunk)
        end
      end
    end

    response = Excon::Response.new
    response.status = socket.readline[9..11].to_i
    while true
      data = socket.readline.chop!
      unless data.empty?
        key, value = data.split(': ')
        response.headers[key] = value
      else
        break
      end
    end

    unless params[:method] == 'HEAD'
      block = if !params[:block] || (params[:expects] && ![*params[:expects]].include?(response.status))
        response.body = ''
        lambda { |chunk| response.body << chunk }
      else
        params[:block]
      end

      if response.headers['Connection'] == 'close'
        block.call(socket.read)
        reset_socket
      elsif response.headers['Content-Length']
        remaining = response.headers['Content-Length'].to_i
        while remaining > 0
          block.call(socket.read([CHUNK_SIZE, remaining].min))
          remaining -= CHUNK_SIZE
        end
      elsif response.headers['Transfer-Encoding'] == 'chunked'
        while true
          chunk_size = socket.readline.chop!.to_i(16)
          chunk = socket.read(chunk_size + 2).chop! # 2 == "/r/n".length
          if chunk_size > 0
            block.call(chunk)
          else
            break
          end
        end
      end
    end
  rescue => socket_error
    reset_socket
    raise(socket_error)
  end

  if params[:expects] && ![*params[:expects]].include?(response.status)
    reset_socket
    raise(Excon::Errors.status_error(params, response))
  else
    response
  end

rescue => request_error
  if params[:idempotent] &&
      (!request_error.is_a?(Excon::Errors::Error) || response.status != 404)
    retries_remaining ||= 4
    retries_remaining -= 1
    if retries_remaining > 0
      retry
    end
  else
    raise(request_error)
  end
end