mwilliams-right_http_connection 1.2.100
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- data/History.txt +59 -0
- data/Manifest.txt +7 -0
- data/README.txt +54 -0
- data/Rakefile +49 -0
- data/VERSION +1 -0
- data/lib/net_fix.rb +160 -0
- data/lib/right_http_connection.rb +441 -0
- metadata +59 -0
data/History.txt
ADDED
@@ -0,0 +1,59 @@
|
|
1
|
+
== 0.0.1 2007-05-15
|
2
|
+
* 1 major enhancement:
|
3
|
+
* Initial release
|
4
|
+
|
5
|
+
== 0.1.2 2007-06-27
|
6
|
+
|
7
|
+
* No major changes.
|
8
|
+
|
9
|
+
== 0.1.3 2007-07-09
|
10
|
+
|
11
|
+
* No change.
|
12
|
+
|
13
|
+
== 0.1.4 2007-08-10
|
14
|
+
|
15
|
+
* r1442, todd, 2007-08-07 15:45:24
|
16
|
+
* # 373, Add support in right_http_connection for bailing out to a block while
|
17
|
+
reading the HTTP response (to support GET streaming...)
|
18
|
+
|
19
|
+
* r1411, todd, 2007-08-03 15:14:45
|
20
|
+
* # 373, Stream uploads (PUTs) if the source is a file, stream, or anything
|
21
|
+
read()-able
|
22
|
+
|
23
|
+
== 1.1.0 2007-08-15
|
24
|
+
Initial public release
|
25
|
+
|
26
|
+
== 1.2.0 2007-10-05
|
27
|
+
|
28
|
+
* r1867, konstantin, 2007-10-05 06:19:45
|
29
|
+
* # 220, (re)open connection to server if none exists or connection params
|
30
|
+
have changed
|
31
|
+
|
32
|
+
== 1.2.1
|
33
|
+
|
34
|
+
* r2648, konstantin, 01-24-08 11:12:00
|
35
|
+
* net_fix.rb moved from right_aws gem to fix the problem with uploading the streamable
|
36
|
+
objects to S3
|
37
|
+
|
38
|
+
* r2764, konstantin, 02-08-08 00:05:00 +03:00
|
39
|
+
* "RightAws: incompatible Net::HTTP monkey-patch" exception is raised if our net_fix
|
40
|
+
patch was overriden (by attachment_fu for example, to avoid this load attachment_fu
|
41
|
+
before loading the right_http_connection gem).
|
42
|
+
|
43
|
+
== 1.2.2
|
44
|
+
|
45
|
+
* r3524, konstantin, 2008-04-17 11:35:42 +0400
|
46
|
+
* Fixed a problem with incorrect error handling (connection retries always failed).
|
47
|
+
|
48
|
+
== 1.2.3
|
49
|
+
|
50
|
+
- Added support for setting retry & timeout parameters in the constructor
|
51
|
+
- Improve handling of data streams during upload: if there is a failure and a retry, reset
|
52
|
+
the seek pointer for the subsequent re-request
|
53
|
+
|
54
|
+
== 1.2.4
|
55
|
+
|
56
|
+
* r4984, konstantin, 2008-08-11 14:49:18 +0400
|
57
|
+
* fixed a bug: <NoMethodError: You have a nil object when you didn't expect it!
|
58
|
+
The error occurred while evaluating nil.body_stream>
|
59
|
+
|
data/Manifest.txt
ADDED
data/README.txt
ADDED
@@ -0,0 +1,54 @@
|
|
1
|
+
RightScale::HttpConnection
|
2
|
+
by RightScale, Inc.
|
3
|
+
www.RightScale.com
|
4
|
+
|
5
|
+
== DESCRIPTION:
|
6
|
+
|
7
|
+
Rightscale::HttpConnection is a robust HTTP/S library. It implements a retry
|
8
|
+
algorithm for low-level network errors.
|
9
|
+
|
10
|
+
== FEATURES:
|
11
|
+
|
12
|
+
- provides put/get streaming
|
13
|
+
- does configurable retries on connect and read timeouts, DNS failures, etc.
|
14
|
+
- HTTPS certificate checking
|
15
|
+
|
16
|
+
== SYNOPSIS:
|
17
|
+
|
18
|
+
|
19
|
+
== REQUIREMENTS:
|
20
|
+
|
21
|
+
- 2/11/08: If you use RightScale::HttpConnection in conjunction with attachment_fu, the
|
22
|
+
HttpConnection gem must be included (using the require statement) AFTER
|
23
|
+
attachment_fu.
|
24
|
+
This is due to a conflict between the HttpConnection gem and another
|
25
|
+
gem required by attachment_fu.
|
26
|
+
|
27
|
+
|
28
|
+
|
29
|
+
== INSTALL:
|
30
|
+
|
31
|
+
sudo gem install right_http_connection
|
32
|
+
|
33
|
+
== LICENSE:
|
34
|
+
|
35
|
+
Copyright (c) 2007-2008 RightScale, Inc.
|
36
|
+
|
37
|
+
Permission is hereby granted, free of charge, to any person obtaining
|
38
|
+
a copy of this software and associated documentation files (the
|
39
|
+
'Software'), to deal in the Software without restriction, including
|
40
|
+
without limitation the rights to use, copy, modify, merge, publish,
|
41
|
+
distribute, sublicense, and/or sell copies of the Software, and to
|
42
|
+
permit persons to whom the Software is furnished to do so, subject to
|
43
|
+
the following conditions:
|
44
|
+
|
45
|
+
The above copyright notice and this permission notice shall be
|
46
|
+
included in all copies or substantial portions of the Software.
|
47
|
+
|
48
|
+
THE SOFTWARE IS PROVIDED 'AS IS', WITHOUT WARRANTY OF ANY KIND,
|
49
|
+
EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
50
|
+
MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
|
51
|
+
IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
|
52
|
+
CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
|
53
|
+
TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
|
54
|
+
SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
data/Rakefile
ADDED
@@ -0,0 +1,49 @@
|
|
1
|
+
require 'rubygems'
|
2
|
+
require 'rake'
|
3
|
+
require 'rake/clean'
|
4
|
+
require 'rake/testtask'
|
5
|
+
require 'rake/packagetask'
|
6
|
+
require 'rake/gempackagetask'
|
7
|
+
require 'rake/rdoctask'
|
8
|
+
require 'rake/contrib/rubyforgepublisher'
|
9
|
+
require 'fileutils'
|
10
|
+
include FileUtils
|
11
|
+
require File.join(File.dirname(__FILE__), 'lib', 'right_http_connection')
|
12
|
+
|
13
|
+
AUTHOR = 'RightScale' # can also be an array of Authors
|
14
|
+
EMAIL = "rubygems@rightscale.com"
|
15
|
+
DESCRIPTION = "RightScale's robust HTTP/S connection module"
|
16
|
+
GEM_NAME = 'right_http_connection' # what ppl will type to install your gem
|
17
|
+
RUBYFORGE_PROJECT = 'rightscale' # The unix name for your project
|
18
|
+
HOMEPATH = "http://#{RUBYFORGE_PROJECT}.rubyforge.org"
|
19
|
+
DOWNLOAD_PATH = "http://rubyforge.org/projects/#{RUBYFORGE_PROJECT}"
|
20
|
+
|
21
|
+
NAME = "right_http_connection"
|
22
|
+
REV = nil # UNCOMMENT IF REQUIRED: File.read(".svn/entries")[/committed-rev="(d+)"/, 1] rescue nil
|
23
|
+
VERS = RightHttpConnection::VERSION::STRING + (REV ? ".#{REV}" : "")
|
24
|
+
CLEAN.include ['**/.*.sw?', '*.gem', '.config', '**/.DS_Store']
|
25
|
+
RDOC_OPTS = ['--quiet', '--title', 'right_http_connection documentation',
|
26
|
+
"--opname", "index.html",
|
27
|
+
"--line-numbers",
|
28
|
+
"--main", "README",
|
29
|
+
"--inline-source"]
|
30
|
+
|
31
|
+
begin
|
32
|
+
require 'jeweler'
|
33
|
+
Jeweler::Tasks.new do |s|
|
34
|
+
s.name = GEM_NAME
|
35
|
+
s.summary = DESCRIPTION
|
36
|
+
s.email = EMAIL
|
37
|
+
s.homepage = HOMEPATH
|
38
|
+
s.description = DESCRIPTION
|
39
|
+
s.authors = ["RightScale"]
|
40
|
+
s.files = FileList["[A-Z]*", "{bin,generators,lib,test}/**/*", 'lib/jeweler/templates/.gitignore']
|
41
|
+
end
|
42
|
+
rescue LoadError
|
43
|
+
puts "Jeweler, or one of its dependencies, is not available. Install it with: sudo gem install technicalpickles-jeweler -s http://gems.github.com"
|
44
|
+
end
|
45
|
+
|
46
|
+
|
47
|
+
|
48
|
+
|
49
|
+
|
data/VERSION
ADDED
@@ -0,0 +1 @@
|
|
1
|
+
1.2.100
|
data/lib/net_fix.rb
ADDED
@@ -0,0 +1,160 @@
|
|
1
|
+
#
|
2
|
+
# Copyright (c) 2008 RightScale Inc
|
3
|
+
#
|
4
|
+
# Permission is hereby granted, free of charge, to any person obtaining
|
5
|
+
# a copy of this software and associated documentation files (the
|
6
|
+
# "Software"), to deal in the Software without restriction, including
|
7
|
+
# without limitation the rights to use, copy, modify, merge, publish,
|
8
|
+
# distribute, sublicense, and/or sell copies of the Software, and to
|
9
|
+
# permit persons to whom the Software is furnished to do so, subject to
|
10
|
+
# the following conditions:
|
11
|
+
#
|
12
|
+
# The above copyright notice and this permission notice shall be
|
13
|
+
# included in all copies or substantial portions of the Software.
|
14
|
+
#
|
15
|
+
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
16
|
+
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
17
|
+
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
18
|
+
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
|
19
|
+
# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
|
20
|
+
# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
21
|
+
# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
22
|
+
#
|
23
|
+
#
|
24
|
+
|
25
|
+
# Net::HTTP and Net::HTTPGenericRequest fixes to support 100-continue on
|
26
|
+
# POST and PUT. The request must have 'expect' field set to '100-continue'.
|
27
|
+
|
28
|
+
|
29
|
+
module Net
|
30
|
+
|
31
|
+
class BufferedIO #:nodoc:
|
32
|
+
# Monkey-patch Net::BufferedIO to read > 1024 bytes from the socket at a time
|
33
|
+
|
34
|
+
# Default size (in bytes) of the max read from a socket into the user space read buffers for socket IO
|
35
|
+
DEFAULT_SOCKET_READ_SIZE = 16*1024
|
36
|
+
|
37
|
+
@@socket_read_size = DEFAULT_SOCKET_READ_SIZE
|
38
|
+
|
39
|
+
def self.socket_read_size=(readsize)
|
40
|
+
if(readsize <= 0)
|
41
|
+
return
|
42
|
+
end
|
43
|
+
@@socket_read_size = readsize
|
44
|
+
end
|
45
|
+
|
46
|
+
def self.socket_read_size?()
|
47
|
+
@@socket_read_size
|
48
|
+
end
|
49
|
+
|
50
|
+
def rbuf_fill
|
51
|
+
timeout(@read_timeout) {
|
52
|
+
@rbuf << @io.sysread(@@socket_read_size)
|
53
|
+
}
|
54
|
+
end
|
55
|
+
end
|
56
|
+
|
57
|
+
|
58
|
+
#-- Net::HTTPGenericRequest --
|
59
|
+
|
60
|
+
class HTTPGenericRequest
|
61
|
+
# Monkey-patch Net::HTTPGenericRequest to read > 1024 bytes from the local data
|
62
|
+
# source at a time (used in streaming PUTs)
|
63
|
+
|
64
|
+
# Default size (in bytes) of the max read from a local source (File, String,
|
65
|
+
# etc.) to the user space write buffers for socket IO.
|
66
|
+
DEFAULT_LOCAL_READ_SIZE = 16*1024
|
67
|
+
|
68
|
+
@@local_read_size = DEFAULT_LOCAL_READ_SIZE
|
69
|
+
|
70
|
+
def self.local_read_size=(readsize)
|
71
|
+
if(readsize <= 0)
|
72
|
+
return
|
73
|
+
end
|
74
|
+
@@local_read_size = readsize
|
75
|
+
end
|
76
|
+
|
77
|
+
def self.local_read_size?()
|
78
|
+
@@local_read_size
|
79
|
+
end
|
80
|
+
|
81
|
+
def exec(sock, ver, path, send_only=nil) #:nodoc: internal use only
|
82
|
+
if @body
|
83
|
+
send_request_with_body sock, ver, path, @body, send_only
|
84
|
+
elsif @body_stream
|
85
|
+
send_request_with_body_stream sock, ver, path, @body_stream, send_only
|
86
|
+
else
|
87
|
+
write_header(sock, ver, path)
|
88
|
+
end
|
89
|
+
end
|
90
|
+
|
91
|
+
private
|
92
|
+
|
93
|
+
def send_request_with_body(sock, ver, path, body, send_only=nil)
|
94
|
+
self.content_length = body.length
|
95
|
+
delete 'Transfer-Encoding'
|
96
|
+
supply_default_content_type
|
97
|
+
write_header(sock, ver, path) unless send_only == :body
|
98
|
+
sock.write(body) unless send_only == :header
|
99
|
+
end
|
100
|
+
|
101
|
+
def send_request_with_body_stream(sock, ver, path, f, send_only=nil)
|
102
|
+
unless content_length() or chunked?
|
103
|
+
raise ArgumentError,
|
104
|
+
"Content-Length not given and Transfer-Encoding is not `chunked'"
|
105
|
+
end
|
106
|
+
supply_default_content_type
|
107
|
+
write_header(sock, ver, path) unless send_only == :body
|
108
|
+
unless send_only == :header
|
109
|
+
if chunked?
|
110
|
+
while s = f.read(@@local_read_size)
|
111
|
+
sock.write(sprintf("%x\r\n", s.length) << s << "\r\n")
|
112
|
+
end
|
113
|
+
sock.write "0\r\n\r\n"
|
114
|
+
else
|
115
|
+
while s = f.read(@@local_read_size)
|
116
|
+
sock.write s
|
117
|
+
end
|
118
|
+
end
|
119
|
+
end
|
120
|
+
end
|
121
|
+
end
|
122
|
+
|
123
|
+
|
124
|
+
#-- Net::HTTP --
|
125
|
+
|
126
|
+
class HTTP
|
127
|
+
def request(req, body = nil, &block) # :yield: +response+
|
128
|
+
unless started?
|
129
|
+
start {
|
130
|
+
req['connection'] ||= 'close'
|
131
|
+
return request(req, body, &block)
|
132
|
+
}
|
133
|
+
end
|
134
|
+
if proxy_user()
|
135
|
+
unless use_ssl?
|
136
|
+
req.proxy_basic_auth proxy_user(), proxy_pass()
|
137
|
+
end
|
138
|
+
end
|
139
|
+
# set body
|
140
|
+
req.set_body_internal body
|
141
|
+
begin_transport req
|
142
|
+
# if we expect 100-continue then send a header first
|
143
|
+
send_only = ((req.is_a?(Post)||req.is_a?(Put)) && (req['expect']=='100-continue')) ? :header : nil
|
144
|
+
req.exec @socket, @curr_http_version, edit_path(req.path), send_only
|
145
|
+
begin
|
146
|
+
res = HTTPResponse.read_new(@socket)
|
147
|
+
# if we expected 100-continue then send a body
|
148
|
+
if res.is_a?(HTTPContinue) && send_only && req['content-length'].to_i > 0
|
149
|
+
req.exec @socket, @curr_http_version, edit_path(req.path), :body
|
150
|
+
end
|
151
|
+
end while res.kind_of?(HTTPContinue)
|
152
|
+
res.reading_body(@socket, req.response_body_permitted?) {
|
153
|
+
yield res if block_given?
|
154
|
+
}
|
155
|
+
end_transport req, res
|
156
|
+
res
|
157
|
+
end
|
158
|
+
end
|
159
|
+
|
160
|
+
end
|
@@ -0,0 +1,441 @@
|
|
1
|
+
#
|
2
|
+
# Copyright (c) 2007-2008 RightScale Inc
|
3
|
+
#
|
4
|
+
# Permission is hereby granted, free of charge, to any person obtaining
|
5
|
+
# a copy of this software and associated documentation files (the
|
6
|
+
# "Software"), to deal in the Software without restriction, including
|
7
|
+
# without limitation the rights to use, copy, modify, merge, publish,
|
8
|
+
# distribute, sublicense, and/or sell copies of the Software, and to
|
9
|
+
# permit persons to whom the Software is furnished to do so, subject to
|
10
|
+
# the following conditions:
|
11
|
+
#
|
12
|
+
# The above copyright notice and this permission notice shall be
|
13
|
+
# included in all copies or substantial portions of the Software.
|
14
|
+
#
|
15
|
+
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
16
|
+
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
17
|
+
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
18
|
+
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
|
19
|
+
# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
|
20
|
+
# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
21
|
+
# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
22
|
+
#
|
23
|
+
|
24
|
+
require "net/https"
|
25
|
+
require "uri"
|
26
|
+
require "time"
|
27
|
+
require "logger"
|
28
|
+
|
29
|
+
$:.unshift(File.dirname(__FILE__))
|
30
|
+
require "net_fix"
|
31
|
+
|
32
|
+
|
33
|
+
module RightHttpConnection #:nodoc:
|
34
|
+
module VERSION #:nodoc:
|
35
|
+
MAJOR = 1
|
36
|
+
MINOR = 2
|
37
|
+
TINY = 99 #4
|
38
|
+
|
39
|
+
STRING = [MAJOR, MINOR, TINY].join('.')
|
40
|
+
end
|
41
|
+
end
|
42
|
+
|
43
|
+
|
44
|
+
module Rightscale
|
45
|
+
|
46
|
+
=begin rdoc
|
47
|
+
HttpConnection maintains a persistent HTTP connection to a remote
|
48
|
+
server. Each instance maintains its own unique connection to the
|
49
|
+
HTTP server. HttpConnection makes a best effort to receive a proper
|
50
|
+
HTTP response from the server, although it does not guarantee that
|
51
|
+
this response contains a HTTP Success code.
|
52
|
+
|
53
|
+
On low-level errors (TCP/IP errors) HttpConnection invokes a reconnect
|
54
|
+
and retry algorithm. Note that although each HttpConnection object
|
55
|
+
has its own connection to the HTTP server, error handling is shared
|
56
|
+
across all connections to a server. For example, if there are three
|
57
|
+
connections to www.somehttpserver.com, a timeout error on one of those
|
58
|
+
connections will cause all three connections to break and reconnect.
|
59
|
+
A connection will not break and reconnect, however, unless a request
|
60
|
+
becomes active on it within a certain amount of time after the error
|
61
|
+
(as specified by HTTP_CONNECTION_RETRY_DELAY). An idle connection will not
|
62
|
+
break even if other connections to the same server experience errors.
|
63
|
+
|
64
|
+
A HttpConnection will retry a request a certain number of times (as
|
65
|
+
defined by HTTP_CONNNECTION_RETRY_COUNT). If all the retries fail,
|
66
|
+
an exception is thrown and all HttpConnections associated with a
|
67
|
+
server enter a probationary period defined by HTTP_CONNECTION_RETRY_DELAY.
|
68
|
+
If the user makes a new request subsequent to entering probation,
|
69
|
+
the request will fail immediately with the same exception thrown
|
70
|
+
on probation entry. This is so that if the HTTP server has gone
|
71
|
+
down, not every subsequent request must wait for a connect timeout
|
72
|
+
before failing. After the probation period expires, the internal
|
73
|
+
state of the HttpConnection is reset and subsequent requests have
|
74
|
+
the full number of potential reconnects and retries available to
|
75
|
+
them.
|
76
|
+
=end
|
77
|
+
|
78
|
+
class HttpConnection
|
79
|
+
|
80
|
+
# Number of times to retry the request after encountering the first error
|
81
|
+
HTTP_CONNECTION_RETRY_COUNT = 3
|
82
|
+
# Throw a Timeout::Error if a connection isn't established within this number of seconds
|
83
|
+
HTTP_CONNECTION_OPEN_TIMEOUT = 5
|
84
|
+
# Throw a Timeout::Error if no data have been read on this connnection within this number of seconds
|
85
|
+
HTTP_CONNECTION_READ_TIMEOUT = 120
|
86
|
+
# Length of the post-error probationary period during which all requests will fail
|
87
|
+
HTTP_CONNECTION_RETRY_DELAY = 15
|
88
|
+
|
89
|
+
#--------------------
|
90
|
+
# class methods
|
91
|
+
#--------------------
|
92
|
+
#
|
93
|
+
@@params = {}
|
94
|
+
@@params[:http_connection_retry_count] = HTTP_CONNECTION_RETRY_COUNT
|
95
|
+
@@params[:http_connection_open_timeout] = HTTP_CONNECTION_OPEN_TIMEOUT
|
96
|
+
@@params[:http_connection_read_timeout] = HTTP_CONNECTION_READ_TIMEOUT
|
97
|
+
@@params[:http_connection_retry_delay] = HTTP_CONNECTION_RETRY_DELAY
|
98
|
+
|
99
|
+
# Query the global (class-level) parameters:
|
100
|
+
#
|
101
|
+
# :user_agent => 'www.HostName.com' # String to report as HTTP User agent
|
102
|
+
# :ca_file => 'path_to_file' # Path to a CA certification file in PEM format. The file can contain several CA certificates. If this parameter isn't set, HTTPS certs won't be verified.
|
103
|
+
# :logger => Logger object # If omitted, HttpConnection logs to STDOUT
|
104
|
+
# :exception => Exception to raise # The type of exception to raise
|
105
|
+
# # if a request repeatedly fails. RuntimeError is raised if this parameter is omitted.
|
106
|
+
# :http_connection_retry_count # by default == Rightscale::HttpConnection::HTTP_CONNECTION_RETRY_COUNT
|
107
|
+
# :http_connection_open_timeout # by default == Rightscale::HttpConnection::HTTP_CONNECTION_OPEN_TIMEOUT
|
108
|
+
# :http_connection_read_timeout # by default == Rightscale::HttpConnection::HTTP_CONNECTION_READ_TIMEOUT
|
109
|
+
# :http_connection_retry_delay # by default == Rightscale::HttpConnection::HTTP_CONNECTION_RETRY_DELAY
|
110
|
+
def self.params
|
111
|
+
@@params
|
112
|
+
end
|
113
|
+
|
114
|
+
# Set the global (class-level) parameters
|
115
|
+
def self.params=(params)
|
116
|
+
@@params = params
|
117
|
+
end
|
118
|
+
|
119
|
+
#------------------
|
120
|
+
# instance methods
|
121
|
+
#------------------
|
122
|
+
attr_accessor :http
|
123
|
+
attr_accessor :server
|
124
|
+
attr_accessor :params # see @@params
|
125
|
+
attr_accessor :logger
|
126
|
+
|
127
|
+
# Params hash:
|
128
|
+
# :user_agent => 'www.HostName.com' # String to report as HTTP User agent
|
129
|
+
# :ca_file => 'path_to_file' # A path of a CA certification file in PEM format. The file can contain several CA certificates.
|
130
|
+
# :logger => Logger object # If omitted, HttpConnection logs to STDOUT
|
131
|
+
# :exception => Exception to raise # The type of exception to raise if a request repeatedly fails. RuntimeError is raised if this parameter is omitted.
|
132
|
+
# :http_connection_retry_count # by default == Rightscale::HttpConnection.params[:http_connection_retry_count]
|
133
|
+
# :http_connection_open_timeout # by default == Rightscale::HttpConnection.params[:http_connection_open_timeout]
|
134
|
+
# :http_connection_read_timeout # by default == Rightscale::HttpConnection.params[:http_connection_read_timeout]
|
135
|
+
# :http_connection_retry_delay # by default == Rightscale::HttpConnection.params[:http_connection_retry_delay]
|
136
|
+
#
|
137
|
+
def initialize(params={})
|
138
|
+
@params = params
|
139
|
+
@params[:http_connection_retry_count] ||= @@params[:http_connection_retry_count]
|
140
|
+
@params[:http_connection_open_timeout] ||= @@params[:http_connection_open_timeout]
|
141
|
+
@params[:http_connection_read_timeout] ||= @@params[:http_connection_read_timeout]
|
142
|
+
@params[:http_connection_retry_delay] ||= @@params[:http_connection_retry_delay]
|
143
|
+
@http = nil
|
144
|
+
@server = nil
|
145
|
+
@logger = get_param(:logger) ||
|
146
|
+
(RAILS_DEFAULT_LOGGER if defined?(RAILS_DEFAULT_LOGGER)) ||
|
147
|
+
Logger.new(STDOUT)
|
148
|
+
end
|
149
|
+
|
150
|
+
def get_param(name)
|
151
|
+
@params[name] || @@params[name]
|
152
|
+
end
|
153
|
+
|
154
|
+
# Query for the maximum size (in bytes) of a single read from the underlying
|
155
|
+
# socket. For bulk transfer, especially over fast links, this is value is
|
156
|
+
# critical to performance.
|
157
|
+
def socket_read_size?
|
158
|
+
Net::BufferedIO.socket_read_size?
|
159
|
+
end
|
160
|
+
|
161
|
+
# Set the maximum size (in bytes) of a single read from the underlying
|
162
|
+
# socket. For bulk transfer, especially over fast links, this is value is
|
163
|
+
# critical to performance.
|
164
|
+
def socket_read_size=(newsize)
|
165
|
+
Net::BufferedIO.socket_read_size=(newsize)
|
166
|
+
end
|
167
|
+
|
168
|
+
# Query for the maximum size (in bytes) of a single read from local data
|
169
|
+
# sources like files. This is important, for example, in a streaming PUT of a
|
170
|
+
# large buffer.
|
171
|
+
def local_read_size?
|
172
|
+
Net::HTTPGenericRequest.local_read_size?
|
173
|
+
end
|
174
|
+
|
175
|
+
# Set the maximum size (in bytes) of a single read from local data
|
176
|
+
# sources like files. This can be used to tune the performance of, for example, a streaming PUT of a
|
177
|
+
# large buffer.
|
178
|
+
def local_read_size=(newsize)
|
179
|
+
Net::HTTPGenericRequest.local_read_size=(newsize)
|
180
|
+
end
|
181
|
+
|
182
|
+
private
|
183
|
+
#--------------
|
184
|
+
# Retry state - Keep track of errors on a per-server basis
|
185
|
+
#--------------
|
186
|
+
@@state = {} # retry state indexed by server: consecutive error count, error time, and error
|
187
|
+
@@eof = {}
|
188
|
+
|
189
|
+
# number of consecutive errors seen for server, 0 all is ok
|
190
|
+
def error_count
|
191
|
+
@@state[@server] ? @@state[@server][:count] : 0
|
192
|
+
end
|
193
|
+
|
194
|
+
# time of last error for server, nil if all is ok
|
195
|
+
def error_time
|
196
|
+
@@state[@server] && @@state[@server][:time]
|
197
|
+
end
|
198
|
+
|
199
|
+
# message for last error for server, "" if all is ok
|
200
|
+
def error_message
|
201
|
+
@@state[@server] ? @@state[@server][:message] : ""
|
202
|
+
end
|
203
|
+
|
204
|
+
# add an error for a server
|
205
|
+
def error_add(message)
|
206
|
+
@@state[@server] = { :count => error_count+1, :time => Time.now, :message => message }
|
207
|
+
end
|
208
|
+
|
209
|
+
# reset the error state for a server (i.e. a request succeeded)
|
210
|
+
def error_reset
|
211
|
+
@@state.delete(@server)
|
212
|
+
end
|
213
|
+
|
214
|
+
# Error message stuff...
|
215
|
+
def banana_message
|
216
|
+
return "#{@server} temporarily unavailable: (#{error_message})"
|
217
|
+
end
|
218
|
+
|
219
|
+
def err_header
|
220
|
+
return "#{self.class.name} :"
|
221
|
+
end
|
222
|
+
|
223
|
+
# Adds new EOF timestamp.
|
224
|
+
# Returns the number of seconds to wait before new conection retry:
|
225
|
+
# 0.5, 1, 2, 4, 8
|
226
|
+
def add_eof
|
227
|
+
(@@eof[@server] ||= []).unshift Time.now
|
228
|
+
0.25 * 2 ** @@eof[@server].size
|
229
|
+
end
|
230
|
+
|
231
|
+
# Returns first EOF timestamp or nul if have no EOFs being tracked.
|
232
|
+
def eof_time
|
233
|
+
@@eof[@server] && @@eof[@server].last
|
234
|
+
end
|
235
|
+
|
236
|
+
# Returns true if we are receiving EOFs during last @params[:http_connection_retry_delay] seconds
|
237
|
+
# and there were no successful response from server
|
238
|
+
def raise_on_eof_exception?
|
239
|
+
@@eof[@server].blank? ? false : ( (Time.now.to_i-@params[:http_connection_retry_delay]) > @@eof[@server].last.to_i )
|
240
|
+
end
|
241
|
+
|
242
|
+
# Reset a list of EOFs for this server.
|
243
|
+
# This is being called when we have got an successful response from server.
|
244
|
+
def eof_reset
|
245
|
+
@@eof.delete(@server)
|
246
|
+
end
|
247
|
+
|
248
|
+
# Detects if an object is 'streamable' - can we read from it, and can we know the size?
|
249
|
+
def setup_streaming(request)
|
250
|
+
if(request.body && request.body.respond_to?(:read))
|
251
|
+
body = request.body
|
252
|
+
request.content_length = body.respond_to?(:lstat) ? body.lstat.size : body.size
|
253
|
+
request.body_stream = request.body
|
254
|
+
true
|
255
|
+
end
|
256
|
+
end
|
257
|
+
|
258
|
+
def get_fileptr_offset(request_params)
|
259
|
+
request_params[:request].body.pos
|
260
|
+
rescue Exception => e
|
261
|
+
# Probably caught this because the body doesn't support the pos() method, like if it is a socket.
|
262
|
+
# Just return 0 and get on with life.
|
263
|
+
0
|
264
|
+
end
|
265
|
+
|
266
|
+
def reset_fileptr_offset(request, offset = 0)
|
267
|
+
if(request.body_stream && request.body_stream.respond_to?(:pos))
|
268
|
+
begin
|
269
|
+
request.body_stream.pos = offset
|
270
|
+
rescue Exception => e
|
271
|
+
@logger.warn("Failed file pointer reset; aborting HTTP retries." +
|
272
|
+
" -- #{err_header} #{e.inspect}")
|
273
|
+
raise e
|
274
|
+
end
|
275
|
+
end
|
276
|
+
end
|
277
|
+
|
278
|
+
# Start a fresh connection. The object closes any existing connection and
|
279
|
+
# opens a new one.
|
280
|
+
def start(request_params)
|
281
|
+
# close the previous if exists
|
282
|
+
finish
|
283
|
+
# create new connection
|
284
|
+
@server = request_params[:server]
|
285
|
+
@port = request_params[:port]
|
286
|
+
@protocol = request_params[:protocol]
|
287
|
+
|
288
|
+
@logger.info("Opening new #{@protocol.upcase} connection to #@server:#@port")
|
289
|
+
# raise 1/0
|
290
|
+
|
291
|
+
@http = Net::HTTP.new(@server, @port)
|
292
|
+
@http.open_timeout = @params[:http_connection_open_timeout]
|
293
|
+
@http.read_timeout = @params[:http_connection_read_timeout]
|
294
|
+
|
295
|
+
if @protocol == 'https'
|
296
|
+
verifyCallbackProc = Proc.new{ |ok, x509_store_ctx|
|
297
|
+
code = x509_store_ctx.error
|
298
|
+
msg = x509_store_ctx.error_string
|
299
|
+
#debugger
|
300
|
+
@logger.warn("##### #{@server} certificate verify failed: #{msg}") unless code == 0
|
301
|
+
true
|
302
|
+
}
|
303
|
+
@http.use_ssl = true
|
304
|
+
ca_file = get_param(:ca_file)
|
305
|
+
if ca_file
|
306
|
+
@http.verify_mode = OpenSSL::SSL::VERIFY_PEER
|
307
|
+
@http.verify_callback = verifyCallbackProc
|
308
|
+
@http.ca_file = ca_file
|
309
|
+
end
|
310
|
+
end
|
311
|
+
# open connection
|
312
|
+
@http.start
|
313
|
+
end
|
314
|
+
|
315
|
+
public
|
316
|
+
|
317
|
+
=begin rdoc
|
318
|
+
Send HTTP request to server
|
319
|
+
|
320
|
+
request_params hash:
|
321
|
+
:server => 'www.HostName.com' # Hostname or IP address of HTTP server
|
322
|
+
:port => '80' # Port of HTTP server
|
323
|
+
:protocol => 'https' # http and https are supported on any port
|
324
|
+
:request => 'requeststring' # Fully-formed HTTP request to make
|
325
|
+
|
326
|
+
Raises RuntimeError, Interrupt, and params[:exception] (if specified in new).
|
327
|
+
|
328
|
+
=end
|
329
|
+
def request(request_params, &block)
|
330
|
+
# We save the offset here so that if we need to retry, we can return the file pointer to its initial position
|
331
|
+
mypos = get_fileptr_offset(request_params)
|
332
|
+
loop do
|
333
|
+
request_params[:protocol] ||= (request_params[:port] == 443 ? 'https' : 'http')
|
334
|
+
# (re)open connection to server if none exists or params has changed
|
335
|
+
same_server_as_before = @server == request_params[:server] &&
|
336
|
+
@port == request_params[:port] &&
|
337
|
+
@protocol == request_params[:protocol]
|
338
|
+
|
339
|
+
# if we are inside a delay between retries: no requests this time!
|
340
|
+
# (skip this step if the endpoint has changed)
|
341
|
+
if error_count > @params[:http_connection_retry_count] &&
|
342
|
+
error_time + @params[:http_connection_retry_delay] > Time.now &&
|
343
|
+
same_server_as_before
|
344
|
+
|
345
|
+
# store the message (otherwise it will be lost after error_reset and
|
346
|
+
# we will raise an exception with an empty text)
|
347
|
+
banana_message_text = banana_message
|
348
|
+
@logger.warn("#{err_header} re-raising same error: #{banana_message_text} " +
|
349
|
+
"-- error count: #{error_count}, error age: #{Time.now.to_i - error_time.to_i}")
|
350
|
+
exception = get_param(:exception) || RuntimeError
|
351
|
+
raise exception.new(banana_message_text)
|
352
|
+
end
|
353
|
+
|
354
|
+
# try to connect server(if connection does not exist) and get response data
|
355
|
+
begin
|
356
|
+
request = request_params[:request]
|
357
|
+
request['User-Agent'] = get_param(:user_agent) || ''
|
358
|
+
unless @http &&
|
359
|
+
@http.started? &&
|
360
|
+
same_server_as_before
|
361
|
+
start(request_params)
|
362
|
+
end
|
363
|
+
|
364
|
+
# Detect if the body is a streamable object like a file or socket. If so, stream that
|
365
|
+
# bad boy.
|
366
|
+
setup_streaming(request)
|
367
|
+
response = @http.request(request, &block)
|
368
|
+
|
369
|
+
error_reset
|
370
|
+
eof_reset
|
371
|
+
return response
|
372
|
+
|
373
|
+
# We treat EOF errors and the timeout/network errors differently. Both
|
374
|
+
# are tracked in different statistics blocks. Note below that EOF
|
375
|
+
# errors will sleep for a certain (exponentially increasing) period.
|
376
|
+
# Other errors don't sleep because there is already an inherent delay
|
377
|
+
# in them; connect and read timeouts (for example) have already
|
378
|
+
# 'slept'. It is still not clear which way we should treat errors
|
379
|
+
# like RST and resolution failures. For now, there is no additional
|
380
|
+
# delay for these errors although this may change in the future.
|
381
|
+
|
382
|
+
# EOFError means the server closed the connection on us.
|
383
|
+
rescue EOFError => e
|
384
|
+
@logger.debug("#{err_header} server #{@server} closed connection")
|
385
|
+
|
386
|
+
# if we have waited long enough - raise an exception...
|
387
|
+
if raise_on_eof_exception?
|
388
|
+
exception = get_param(:exception) || RuntimeError
|
389
|
+
@logger.warn("#{err_header} raising #{exception} due to permanent EOF being received from #{@server}, error age: #{Time.now.to_i - eof_time.to_i}")
|
390
|
+
# @http = nil (may be required)
|
391
|
+
raise exception.new("Permanent EOF is being received from #{@server}.")
|
392
|
+
else
|
393
|
+
# ... else just sleep a bit before new retry
|
394
|
+
sleep(add_eof)
|
395
|
+
# We will be retrying the request, so reset the file pointer
|
396
|
+
reset_fileptr_offset(request, mypos)
|
397
|
+
end
|
398
|
+
rescue Exception => e # See comment at bottom for the list of errors seen...
|
399
|
+
# if ctrl+c is pressed - we have to reraise exception to terminate proggy
|
400
|
+
if e.is_a?(Interrupt) && !( e.is_a?(Errno::ETIMEDOUT) || e.is_a?(Timeout::Error))
|
401
|
+
@logger.debug( "#{err_header} request to server #{@server} interrupted by ctrl-c")
|
402
|
+
raise
|
403
|
+
elsif e.is_a?(ArgumentError) && e.message.include?('wrong number of arguments (5 for 4)')
|
404
|
+
# seems our net_fix patch was overriden...
|
405
|
+
exception = get_param(:exception) || RuntimeError
|
406
|
+
# @http = nil (may be required)
|
407
|
+
raise exception.new('incompatible Net::HTTP monkey-patch')
|
408
|
+
end
|
409
|
+
# oops - we got a banana: log it
|
410
|
+
error_add(e.message)
|
411
|
+
@logger.warn("#{err_header} request failure count: #{error_count}, exception: #{e.inspect}")
|
412
|
+
|
413
|
+
# We will be retrying the request, so reset the file pointer
|
414
|
+
reset_fileptr_offset(request, mypos)
|
415
|
+
|
416
|
+
end
|
417
|
+
end
|
418
|
+
end
|
419
|
+
|
420
|
+
def finish(reason = '')
|
421
|
+
if @http && @http.started?
|
422
|
+
reason = ", reason: '#{reason}'" unless reason.blank?
|
423
|
+
@logger.info("Closing #{@http.use_ssl? ? 'HTTPS' : 'HTTP'} connection to #{@http.address}:#{@http.port}#{reason}")
|
424
|
+
@http.finish
|
425
|
+
end
|
426
|
+
@http = nil
|
427
|
+
end
|
428
|
+
|
429
|
+
# Errors received during testing:
|
430
|
+
#
|
431
|
+
# #<Timeout::Error: execution expired>
|
432
|
+
# #<Errno::ETIMEDOUT: Connection timed out - connect(2)>
|
433
|
+
# #<SocketError: getaddrinfo: Name or service not known>
|
434
|
+
# #<SocketError: getaddrinfo: Temporary failure in name resolution>
|
435
|
+
# #<EOFError: end of file reached>
|
436
|
+
# #<Errno::ECONNRESET: Connection reset by peer>
|
437
|
+
# #<OpenSSL::SSL::SSLError: SSL_write:: bad write retry>
|
438
|
+
end
|
439
|
+
|
440
|
+
end
|
441
|
+
|
metadata
ADDED
@@ -0,0 +1,59 @@
|
|
1
|
+
--- !ruby/object:Gem::Specification
|
2
|
+
name: mwilliams-right_http_connection
|
3
|
+
version: !ruby/object:Gem::Version
|
4
|
+
version: 1.2.100
|
5
|
+
platform: ruby
|
6
|
+
authors:
|
7
|
+
- RightScale
|
8
|
+
autorequire:
|
9
|
+
bindir: bin
|
10
|
+
cert_chain: []
|
11
|
+
|
12
|
+
date: 2009-07-14 00:00:00 -07:00
|
13
|
+
default_executable:
|
14
|
+
dependencies: []
|
15
|
+
|
16
|
+
description: RightScale's robust HTTP/S connection module
|
17
|
+
email: rubygems@rightscale.com
|
18
|
+
executables: []
|
19
|
+
|
20
|
+
extensions: []
|
21
|
+
|
22
|
+
extra_rdoc_files:
|
23
|
+
- README.txt
|
24
|
+
files:
|
25
|
+
- History.txt
|
26
|
+
- Manifest.txt
|
27
|
+
- README.txt
|
28
|
+
- Rakefile
|
29
|
+
- VERSION
|
30
|
+
- lib/net_fix.rb
|
31
|
+
- lib/right_http_connection.rb
|
32
|
+
has_rdoc: true
|
33
|
+
homepage: http://rightscale.rubyforge.org
|
34
|
+
post_install_message:
|
35
|
+
rdoc_options:
|
36
|
+
- --charset=UTF-8
|
37
|
+
require_paths:
|
38
|
+
- lib
|
39
|
+
required_ruby_version: !ruby/object:Gem::Requirement
|
40
|
+
requirements:
|
41
|
+
- - ">="
|
42
|
+
- !ruby/object:Gem::Version
|
43
|
+
version: "0"
|
44
|
+
version:
|
45
|
+
required_rubygems_version: !ruby/object:Gem::Requirement
|
46
|
+
requirements:
|
47
|
+
- - ">="
|
48
|
+
- !ruby/object:Gem::Version
|
49
|
+
version: "0"
|
50
|
+
version:
|
51
|
+
requirements: []
|
52
|
+
|
53
|
+
rubyforge_project:
|
54
|
+
rubygems_version: 1.2.0
|
55
|
+
signing_key:
|
56
|
+
specification_version: 2
|
57
|
+
summary: RightScale's robust HTTP/S connection module
|
58
|
+
test_files: []
|
59
|
+
|