fblee-typhoeus 0.1.31
Sign up to get free protection for your applications and to get access to all the features.
- data/.gitignore +2 -0
- data/CHANGELOG.markdown +27 -0
- data/README.textile +333 -0
- data/Rakefile +39 -0
- data/VERSION +1 -0
- data/benchmarks/profile.rb +25 -0
- data/benchmarks/vs_nethttp.rb +35 -0
- data/examples/twitter.rb +21 -0
- data/ext/typhoeus/.gitignore +5 -0
- data/ext/typhoeus/Makefile +157 -0
- data/ext/typhoeus/extconf.rb +65 -0
- data/ext/typhoeus/native.c +11 -0
- data/ext/typhoeus/native.h +21 -0
- data/ext/typhoeus/typhoeus_easy.c +207 -0
- data/ext/typhoeus/typhoeus_easy.h +19 -0
- data/ext/typhoeus/typhoeus_multi.c +225 -0
- data/ext/typhoeus/typhoeus_multi.h +16 -0
- data/lib/typhoeus/.gitignore +1 -0
- data/lib/typhoeus/easy.rb +348 -0
- data/lib/typhoeus/filter.rb +28 -0
- data/lib/typhoeus/hydra.rb +243 -0
- data/lib/typhoeus/multi.rb +35 -0
- data/lib/typhoeus/remote.rb +306 -0
- data/lib/typhoeus/remote_method.rb +108 -0
- data/lib/typhoeus/remote_proxy_object.rb +48 -0
- data/lib/typhoeus/request.rb +172 -0
- data/lib/typhoeus/response.rb +49 -0
- data/lib/typhoeus/service.rb +20 -0
- data/lib/typhoeus.rb +55 -0
- data/profilers/valgrind.rb +24 -0
- data/spec/fixtures/result_set.xml +60 -0
- data/spec/servers/app.rb +84 -0
- data/spec/spec.opts +2 -0
- data/spec/spec_helper.rb +11 -0
- data/spec/typhoeus/easy_spec.rb +249 -0
- data/spec/typhoeus/filter_spec.rb +35 -0
- data/spec/typhoeus/hydra_spec.rb +311 -0
- data/spec/typhoeus/multi_spec.rb +74 -0
- data/spec/typhoeus/remote_method_spec.rb +141 -0
- data/spec/typhoeus/remote_proxy_object_spec.rb +65 -0
- data/spec/typhoeus/remote_spec.rb +695 -0
- data/spec/typhoeus/request_spec.rb +195 -0
- data/spec/typhoeus/response_spec.rb +63 -0
- data/typhoeus.gemspec +113 -0
- metadata +188 -0
@@ -0,0 +1,65 @@
|
|
1
|
+
ENV['RC_ARCHS'] = '' if RUBY_PLATFORM =~ /darwin/
|
2
|
+
|
3
|
+
# :stopdoc:
|
4
|
+
|
5
|
+
require 'mkmf'
|
6
|
+
|
7
|
+
ROOT = File.expand_path(File.join(File.dirname(__FILE__), '..', '..'))
|
8
|
+
specified_curl = ARGV[0] =~ /^--with-curl/ ? ARGV[0].split("=")[1] : nil
|
9
|
+
LIBDIR = specified_curl ? "#{specified_curl}/lib": Config::CONFIG['libdir']
|
10
|
+
INCLUDEDIR = specified_curl ? "#{specified_curl}/include" : Config::CONFIG['includedir']
|
11
|
+
|
12
|
+
if defined?(RUBY_ENGINE) && RUBY_ENGINE == 'macruby'
|
13
|
+
$LIBRUBYARG_STATIC.gsub!(/-static/, '')
|
14
|
+
end
|
15
|
+
|
16
|
+
$CFLAGS << " #{ENV["CFLAGS"]}"
|
17
|
+
if Config::CONFIG['target_os'] == 'mingw32'
|
18
|
+
$CFLAGS << " -DXP_WIN -DXP_WIN32 -DUSE_INCLUDED_VASPRINTF"
|
19
|
+
elsif Config::CONFIG['target_os'] == 'solaris2'
|
20
|
+
$CFLAGS << " -DUSE_INCLUDED_VASPRINTF"
|
21
|
+
else
|
22
|
+
$CFLAGS << " -g -DXP_UNIX"
|
23
|
+
end
|
24
|
+
|
25
|
+
use_macports = !(defined?(RUBY_ENGINE) && RUBY_ENGINE != 'ruby')
|
26
|
+
$LIBPATH << "/opt/local/lib" if use_macports
|
27
|
+
|
28
|
+
$CFLAGS << " -O3 -Wall -Wcast-qual -Wwrite-strings -Wconversion -Wmissing-noreturn -Winline"
|
29
|
+
|
30
|
+
if Config::CONFIG['target_os'] == 'mingw32'
|
31
|
+
header = File.join(ROOT, 'cross', 'curl-7.19.4.win32', 'include')
|
32
|
+
unless find_header('curl/curl.h', header)
|
33
|
+
abort "need libcurl"
|
34
|
+
end
|
35
|
+
else
|
36
|
+
HEADER_DIRS = [
|
37
|
+
File.join(INCLUDEDIR, "curl"),
|
38
|
+
INCLUDEDIR,
|
39
|
+
'/usr/include/curl',
|
40
|
+
'/usr/local/include/curl'
|
41
|
+
]
|
42
|
+
|
43
|
+
[
|
44
|
+
'/opt/local/include/curl',
|
45
|
+
'/opt/local/include',
|
46
|
+
].each { |x| HEADER_DIRS.unshift(x) } if use_macports
|
47
|
+
|
48
|
+
unless find_header('curl/curl.h', *HEADER_DIRS)
|
49
|
+
abort "need libcurl"
|
50
|
+
end
|
51
|
+
end
|
52
|
+
|
53
|
+
if Config::CONFIG['target_os'] == 'mingw32'
|
54
|
+
find_library('curl', 'curl_easy_init',
|
55
|
+
File.join(ROOT, 'cross', 'curl-7.19.4.win32', 'bin'))
|
56
|
+
else
|
57
|
+
find_library('curl', 'curl_easy_init',
|
58
|
+
LIBDIR,
|
59
|
+
'/opt/local/lib',
|
60
|
+
'/usr/local/lib',
|
61
|
+
'/usr/lib'
|
62
|
+
)
|
63
|
+
end
|
64
|
+
|
65
|
+
create_makefile("typhoeus/native")
|
@@ -0,0 +1,21 @@
|
|
1
|
+
#ifndef TYPHOEUS_NATIVE
|
2
|
+
#define TYPHOEUS_NATIVE
|
3
|
+
|
4
|
+
#include <ruby.h>
|
5
|
+
#include <curl/curl.h>
|
6
|
+
#include <curl/easy.h>
|
7
|
+
#include <curl/multi.h>
|
8
|
+
|
9
|
+
void Init_native();
|
10
|
+
extern VALUE mTyphoeus;
|
11
|
+
extern void init_typhoeus_easy();
|
12
|
+
extern void init_typhoeus_multi();
|
13
|
+
|
14
|
+
#endif
|
15
|
+
|
16
|
+
#ifndef RSTRING_PTR
|
17
|
+
|
18
|
+
#define RSTRING_PTR(s) (RSTRING(s)->ptr)
|
19
|
+
#define RSTRING_LEN(s) (RSTRING(s)->len)
|
20
|
+
|
21
|
+
#endif
|
@@ -0,0 +1,207 @@
|
|
1
|
+
#include <typhoeus_easy.h>
|
2
|
+
|
3
|
+
static VALUE idAppend;
|
4
|
+
VALUE cTyphoeusEasy;
|
5
|
+
|
6
|
+
static void dealloc(CurlEasy *curl_easy) {
|
7
|
+
if (curl_easy->request_chunk != NULL) {
|
8
|
+
free(curl_easy->request_chunk);
|
9
|
+
}
|
10
|
+
|
11
|
+
if (curl_easy->headers != NULL) {
|
12
|
+
curl_slist_free_all(curl_easy->headers);
|
13
|
+
}
|
14
|
+
|
15
|
+
curl_easy_cleanup(curl_easy->curl);
|
16
|
+
|
17
|
+
free(curl_easy);
|
18
|
+
}
|
19
|
+
|
20
|
+
static VALUE easy_setopt_string(VALUE self, VALUE opt_name, VALUE parameter) {
|
21
|
+
CurlEasy *curl_easy;
|
22
|
+
Data_Get_Struct(self, CurlEasy, curl_easy);
|
23
|
+
|
24
|
+
long opt = NUM2LONG(opt_name);
|
25
|
+
curl_easy_setopt(curl_easy->curl, opt, StringValuePtr(parameter));
|
26
|
+
return opt_name;
|
27
|
+
}
|
28
|
+
|
29
|
+
static VALUE easy_setopt_long(VALUE self, VALUE opt_name, VALUE parameter) {
|
30
|
+
CurlEasy *curl_easy;
|
31
|
+
Data_Get_Struct(self, CurlEasy, curl_easy);
|
32
|
+
|
33
|
+
long opt = NUM2LONG(opt_name);
|
34
|
+
curl_easy_setopt(curl_easy->curl, opt, NUM2LONG(parameter));
|
35
|
+
return opt_name;
|
36
|
+
}
|
37
|
+
|
38
|
+
static VALUE easy_getinfo_string(VALUE self, VALUE info) {
|
39
|
+
char *info_string;
|
40
|
+
CurlEasy *curl_easy;
|
41
|
+
Data_Get_Struct(self, CurlEasy, curl_easy);
|
42
|
+
|
43
|
+
long opt = NUM2LONG(info);
|
44
|
+
curl_easy_getinfo(curl_easy->curl, opt, &info_string);
|
45
|
+
|
46
|
+
return rb_str_new2(info_string);
|
47
|
+
}
|
48
|
+
|
49
|
+
static VALUE easy_getinfo_long(VALUE self, VALUE info) {
|
50
|
+
long info_long;
|
51
|
+
CurlEasy *curl_easy;
|
52
|
+
Data_Get_Struct(self, CurlEasy, curl_easy);
|
53
|
+
|
54
|
+
long opt = NUM2LONG(info);
|
55
|
+
curl_easy_getinfo(curl_easy->curl, opt, &info_long);
|
56
|
+
|
57
|
+
return LONG2NUM(info_long);
|
58
|
+
}
|
59
|
+
|
60
|
+
static VALUE easy_getinfo_double(VALUE self, VALUE info) {
|
61
|
+
double info_double = 0;
|
62
|
+
CurlEasy *curl_easy;
|
63
|
+
Data_Get_Struct(self, CurlEasy, curl_easy);
|
64
|
+
|
65
|
+
long opt = NUM2LONG(info);
|
66
|
+
curl_easy_getinfo(curl_easy->curl, opt, &info_double);
|
67
|
+
|
68
|
+
return rb_float_new(info_double);
|
69
|
+
}
|
70
|
+
|
71
|
+
static VALUE easy_perform(VALUE self) {
|
72
|
+
CurlEasy *curl_easy;
|
73
|
+
Data_Get_Struct(self, CurlEasy, curl_easy);
|
74
|
+
curl_easy_perform(curl_easy->curl);
|
75
|
+
|
76
|
+
return Qnil;
|
77
|
+
}
|
78
|
+
|
79
|
+
static size_t write_data_handler(char *stream, size_t size, size_t nmemb, VALUE val) {
|
80
|
+
long stream_size = (long)(size * nmemb);
|
81
|
+
rb_funcall(val, idAppend, 1, rb_str_new(stream, stream_size));
|
82
|
+
return size * nmemb;
|
83
|
+
}
|
84
|
+
|
85
|
+
static size_t read_callback(void *ptr, size_t size, size_t nmemb, void *data) {
|
86
|
+
size_t realsize = size * nmemb;
|
87
|
+
RequestChunk *mem = (RequestChunk *)data;
|
88
|
+
|
89
|
+
if (realsize > mem->size - mem->read) {
|
90
|
+
realsize = mem->size - mem->read;
|
91
|
+
}
|
92
|
+
|
93
|
+
if (realsize != 0) {
|
94
|
+
memcpy(ptr, &(mem->memory[mem->read]), realsize);
|
95
|
+
mem->read += realsize;
|
96
|
+
}
|
97
|
+
|
98
|
+
return realsize;
|
99
|
+
}
|
100
|
+
|
101
|
+
static void set_response_handlers(VALUE easy, CURL *curl) {
|
102
|
+
rb_iv_set(easy, "@response_body", rb_str_new2(""));
|
103
|
+
rb_iv_set(easy, "@response_header", rb_str_new2(""));
|
104
|
+
|
105
|
+
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, (curl_write_callback)&write_data_handler);
|
106
|
+
curl_easy_setopt(curl, CURLOPT_WRITEDATA, rb_iv_get(easy, "@response_body"));
|
107
|
+
curl_easy_setopt(curl, CURLOPT_HEADERFUNCTION, (curl_write_callback)&write_data_handler);
|
108
|
+
curl_easy_setopt(curl, CURLOPT_HEADERDATA, rb_iv_get(easy, "@response_header"));
|
109
|
+
}
|
110
|
+
|
111
|
+
static VALUE easy_reset(VALUE self) {
|
112
|
+
CurlEasy *curl_easy;
|
113
|
+
Data_Get_Struct(self, CurlEasy, curl_easy);
|
114
|
+
|
115
|
+
if (curl_easy->request_chunk != NULL) {
|
116
|
+
free(curl_easy->request_chunk);
|
117
|
+
curl_easy->request_chunk = NULL;
|
118
|
+
}
|
119
|
+
|
120
|
+
if (curl_easy->headers != NULL) {
|
121
|
+
curl_slist_free_all(curl_easy->headers);
|
122
|
+
curl_easy->headers = NULL;
|
123
|
+
}
|
124
|
+
|
125
|
+
curl_easy_reset(curl_easy->curl);
|
126
|
+
|
127
|
+
set_response_handlers(self, curl_easy->curl);
|
128
|
+
|
129
|
+
return Qnil;
|
130
|
+
}
|
131
|
+
|
132
|
+
static VALUE easy_add_header(VALUE self, VALUE header) {
|
133
|
+
CurlEasy *curl_easy;
|
134
|
+
Data_Get_Struct(self, CurlEasy, curl_easy);
|
135
|
+
|
136
|
+
curl_easy->headers = curl_slist_append(curl_easy->headers, RSTRING_PTR(header));
|
137
|
+
return header;
|
138
|
+
}
|
139
|
+
|
140
|
+
static VALUE easy_set_headers(VALUE self) {
|
141
|
+
CurlEasy *curl_easy;
|
142
|
+
Data_Get_Struct(self, CurlEasy, curl_easy);
|
143
|
+
|
144
|
+
curl_easy_setopt(curl_easy->curl, CURLOPT_HTTPHEADER, curl_easy->headers);
|
145
|
+
|
146
|
+
return Qnil;
|
147
|
+
}
|
148
|
+
|
149
|
+
static VALUE easy_set_request_body(VALUE self, VALUE data, VALUE content_length_header) {
|
150
|
+
CurlEasy *curl_easy;
|
151
|
+
Data_Get_Struct(self, CurlEasy, curl_easy);
|
152
|
+
|
153
|
+
curl_easy->request_chunk = ALLOC(RequestChunk);
|
154
|
+
curl_easy->request_chunk->size = RSTRING_LEN(data);
|
155
|
+
curl_easy->request_chunk->memory = StringValuePtr(data);
|
156
|
+
curl_easy->request_chunk->read = 0;
|
157
|
+
|
158
|
+
curl_easy_setopt(curl_easy->curl, CURLOPT_READFUNCTION, (curl_read_callback)read_callback);
|
159
|
+
curl_easy_setopt(curl_easy->curl, CURLOPT_READDATA, curl_easy->request_chunk);
|
160
|
+
curl_easy_setopt(curl_easy->curl, CURLOPT_INFILESIZE, RSTRING_LEN(data));
|
161
|
+
|
162
|
+
return Qnil;
|
163
|
+
}
|
164
|
+
|
165
|
+
static VALUE easy_escape(VALUE self, VALUE data, VALUE length) {
|
166
|
+
CurlEasy *curl_easy;
|
167
|
+
Data_Get_Struct(self, CurlEasy, curl_easy);
|
168
|
+
|
169
|
+
return rb_str_new2(curl_easy_escape(curl_easy->curl, StringValuePtr(data), NUM2INT(length)));
|
170
|
+
}
|
171
|
+
|
172
|
+
static VALUE version(VALUE self) {
|
173
|
+
return rb_str_new2(curl_version());
|
174
|
+
}
|
175
|
+
|
176
|
+
static VALUE new(int argc, VALUE *argv, VALUE klass) {
|
177
|
+
CURL *curl = curl_easy_init();
|
178
|
+
CurlEasy *curl_easy = ALLOC(CurlEasy);
|
179
|
+
curl_easy->curl = curl;
|
180
|
+
curl_easy->headers = NULL;
|
181
|
+
curl_easy->request_chunk = NULL;
|
182
|
+
VALUE easy = Data_Wrap_Struct(cTyphoeusEasy, 0, dealloc, curl_easy);
|
183
|
+
|
184
|
+
set_response_handlers(easy, curl);
|
185
|
+
|
186
|
+
rb_obj_call_init(easy, argc, argv);
|
187
|
+
|
188
|
+
return easy;
|
189
|
+
}
|
190
|
+
|
191
|
+
void init_typhoeus_easy() {
|
192
|
+
VALUE klass = cTyphoeusEasy = rb_define_class_under(mTyphoeus, "Easy", rb_cObject);
|
193
|
+
idAppend = rb_intern("<<");
|
194
|
+
rb_define_singleton_method(klass, "new", new, -1);
|
195
|
+
rb_define_private_method(klass, "easy_setopt_string", easy_setopt_string, 2);
|
196
|
+
rb_define_private_method(klass, "easy_setopt_long", easy_setopt_long, 2);
|
197
|
+
rb_define_private_method(klass, "easy_getinfo_string", easy_getinfo_string, 1);
|
198
|
+
rb_define_private_method(klass, "easy_getinfo_long", easy_getinfo_long, 1);
|
199
|
+
rb_define_private_method(klass, "easy_getinfo_double", easy_getinfo_double, 1);
|
200
|
+
rb_define_private_method(klass, "easy_perform", easy_perform, 0);
|
201
|
+
rb_define_private_method(klass, "easy_reset", easy_reset, 0);
|
202
|
+
rb_define_private_method(klass, "easy_set_request_body", easy_set_request_body, 1);
|
203
|
+
rb_define_private_method(klass, "easy_set_headers", easy_set_headers, 0);
|
204
|
+
rb_define_private_method(klass, "easy_add_header", easy_add_header, 1);
|
205
|
+
rb_define_private_method(klass, "easy_escape", easy_escape, 2);
|
206
|
+
rb_define_private_method(klass, "version", version, 0);
|
207
|
+
}
|
@@ -0,0 +1,19 @@
|
|
1
|
+
#ifndef TYPHOEUS_EASY
|
2
|
+
#define TYPHOEUS_EASY
|
3
|
+
|
4
|
+
#include <native.h>
|
5
|
+
|
6
|
+
void init_typhoeus_easy();
|
7
|
+
typedef struct {
|
8
|
+
const char *memory;
|
9
|
+
int size;
|
10
|
+
int read;
|
11
|
+
} RequestChunk;
|
12
|
+
|
13
|
+
typedef struct {
|
14
|
+
RequestChunk *request_chunk;
|
15
|
+
CURL *curl;
|
16
|
+
struct curl_slist *headers;
|
17
|
+
} CurlEasy;
|
18
|
+
|
19
|
+
#endif
|
@@ -0,0 +1,225 @@
|
|
1
|
+
#include <typhoeus_multi.h>
|
2
|
+
|
3
|
+
static void multi_read_info(VALUE self, CURLM *multi_handle);
|
4
|
+
|
5
|
+
static void dealloc(CurlMulti *curl_multi) {
|
6
|
+
curl_multi_cleanup(curl_multi->multi);
|
7
|
+
free(curl_multi);
|
8
|
+
}
|
9
|
+
|
10
|
+
static VALUE multi_add_handle(VALUE self, VALUE easy) {
|
11
|
+
CurlEasy *curl_easy;
|
12
|
+
Data_Get_Struct(easy, CurlEasy, curl_easy);
|
13
|
+
CurlMulti *curl_multi;
|
14
|
+
Data_Get_Struct(self, CurlMulti, curl_multi);
|
15
|
+
CURLMcode mcode;
|
16
|
+
|
17
|
+
mcode = curl_multi_add_handle(curl_multi->multi, curl_easy->curl);
|
18
|
+
if (mcode != CURLM_CALL_MULTI_PERFORM && mcode != CURLM_OK) {
|
19
|
+
rb_raise((VALUE)mcode, "An error occured adding the handle");
|
20
|
+
}
|
21
|
+
|
22
|
+
curl_easy_setopt(curl_easy->curl, CURLOPT_PRIVATE, easy);
|
23
|
+
curl_multi->active++;
|
24
|
+
|
25
|
+
if (mcode == CURLM_CALL_MULTI_PERFORM) {
|
26
|
+
curl_multi_perform(curl_multi->multi, &(curl_multi->running));
|
27
|
+
}
|
28
|
+
//
|
29
|
+
// if (curl_multi->running) {
|
30
|
+
// printf("call read_info on add<br/>");
|
31
|
+
// multi_read_info(self, curl_multi->multi);
|
32
|
+
// }
|
33
|
+
|
34
|
+
return easy;
|
35
|
+
}
|
36
|
+
|
37
|
+
static VALUE multi_remove_handle(VALUE self, VALUE easy) {
|
38
|
+
CurlEasy *curl_easy;
|
39
|
+
Data_Get_Struct(easy, CurlEasy, curl_easy);
|
40
|
+
CurlMulti *curl_multi;
|
41
|
+
Data_Get_Struct(self, CurlMulti, curl_multi);
|
42
|
+
|
43
|
+
curl_multi->active--;
|
44
|
+
curl_multi_remove_handle(curl_multi->multi, curl_easy->curl);
|
45
|
+
|
46
|
+
return easy;
|
47
|
+
}
|
48
|
+
|
49
|
+
static void multi_read_info(VALUE self, CURLM *multi_handle) {
|
50
|
+
int msgs_left, result;
|
51
|
+
CURLMsg *msg;
|
52
|
+
CURLcode ecode;
|
53
|
+
CURL *easy_handle;
|
54
|
+
VALUE easy;
|
55
|
+
|
56
|
+
/* check for finished easy handles and remove from the multi handle */
|
57
|
+
while ((msg = curl_multi_info_read(multi_handle, &msgs_left))) {
|
58
|
+
|
59
|
+
if (msg->msg != CURLMSG_DONE) {
|
60
|
+
continue;
|
61
|
+
}
|
62
|
+
|
63
|
+
easy_handle = msg->easy_handle;
|
64
|
+
result = msg->data.result;
|
65
|
+
if (easy_handle) {
|
66
|
+
ecode = curl_easy_getinfo(easy_handle, CURLINFO_PRIVATE, &easy);
|
67
|
+
if (ecode != 0) {
|
68
|
+
rb_raise(ecode, "error getting easy object");
|
69
|
+
}
|
70
|
+
|
71
|
+
long response_code = -1;
|
72
|
+
curl_easy_getinfo(easy_handle, CURLINFO_RESPONSE_CODE, &response_code);
|
73
|
+
|
74
|
+
// TODO: find out what the real problem is here and fix it.
|
75
|
+
// this next bit is a horrible hack. For some reason my tests against a local server on my laptop
|
76
|
+
// fail intermittently and return this result number. However, it will succeed if you try it a few
|
77
|
+
// more times. Also noteworthy is that this doens't happen when hitting an external server. WTF?!
|
78
|
+
|
79
|
+
// Sandofsky says:
|
80
|
+
// This is caused by OS X first attempting to resolve using IPV6.
|
81
|
+
// Hack solution: connect to yourself with 127.0.0.1, not localhost
|
82
|
+
// http://curl.haxx.se/mail/tracker-2009-09/0018.html
|
83
|
+
if (result == 7) {
|
84
|
+
VALUE max_retries = rb_funcall(easy, rb_intern("max_retries?"), 0);
|
85
|
+
if (max_retries != Qtrue) {
|
86
|
+
multi_remove_handle(self, easy);
|
87
|
+
multi_add_handle(self, easy);
|
88
|
+
CurlMulti *curl_multi;
|
89
|
+
Data_Get_Struct(self, CurlMulti, curl_multi);
|
90
|
+
curl_multi_perform(curl_multi->multi, &(curl_multi->running));
|
91
|
+
|
92
|
+
rb_funcall(easy, rb_intern("increment_retries"), 0);
|
93
|
+
|
94
|
+
continue;
|
95
|
+
}
|
96
|
+
}
|
97
|
+
multi_remove_handle(self, easy);
|
98
|
+
|
99
|
+
if (result != 0) {
|
100
|
+
rb_funcall(easy, rb_intern("failure"), 0);
|
101
|
+
}
|
102
|
+
else if ((response_code >= 200 && response_code < 300) || response_code == 0) {
|
103
|
+
rb_funcall(easy, rb_intern("success"), 0);
|
104
|
+
}
|
105
|
+
else if (response_code >= 300 && response_code < 600) {
|
106
|
+
rb_funcall(easy, rb_intern("failure"), 0);
|
107
|
+
}
|
108
|
+
}
|
109
|
+
}
|
110
|
+
}
|
111
|
+
|
112
|
+
/* called by multi_perform and fire_and_forget */
|
113
|
+
static void rb_curl_multi_run(VALUE self, CURLM *multi_handle, int *still_running) {
|
114
|
+
CURLMcode mcode;
|
115
|
+
|
116
|
+
do {
|
117
|
+
mcode = curl_multi_perform(multi_handle, still_running);
|
118
|
+
} while (mcode == CURLM_CALL_MULTI_PERFORM);
|
119
|
+
|
120
|
+
if (mcode != CURLM_OK) {
|
121
|
+
rb_raise((VALUE)mcode, "an error occured while running perform");
|
122
|
+
}
|
123
|
+
|
124
|
+
multi_read_info( self, multi_handle );
|
125
|
+
}
|
126
|
+
|
127
|
+
static VALUE fire_and_forget(VALUE self) {
|
128
|
+
CurlMulti *curl_multi;
|
129
|
+
Data_Get_Struct(self, CurlMulti, curl_multi);
|
130
|
+
rb_curl_multi_run( self, curl_multi->multi, &(curl_multi->running) );
|
131
|
+
}
|
132
|
+
|
133
|
+
static VALUE multi_perform(VALUE self) {
|
134
|
+
CURLMcode mcode;
|
135
|
+
CurlMulti *curl_multi;
|
136
|
+
int maxfd, rc;
|
137
|
+
fd_set fdread, fdwrite, fdexcep;
|
138
|
+
|
139
|
+
long timeout;
|
140
|
+
struct timeval tv = {0, 0};
|
141
|
+
|
142
|
+
Data_Get_Struct(self, CurlMulti, curl_multi);
|
143
|
+
|
144
|
+
rb_curl_multi_run( self, curl_multi->multi, &(curl_multi->running) );
|
145
|
+
while(curl_multi->running) {
|
146
|
+
FD_ZERO(&fdread);
|
147
|
+
FD_ZERO(&fdwrite);
|
148
|
+
FD_ZERO(&fdexcep);
|
149
|
+
|
150
|
+
/* get the curl suggested time out */
|
151
|
+
mcode = curl_multi_timeout(curl_multi->multi, &timeout);
|
152
|
+
if (mcode != CURLM_OK) {
|
153
|
+
rb_raise((VALUE)mcode, "an error occured getting the timeout");
|
154
|
+
}
|
155
|
+
|
156
|
+
if (timeout == 0) { /* no delay */
|
157
|
+
rb_curl_multi_run( self, curl_multi->multi, &(curl_multi->running) );
|
158
|
+
continue;
|
159
|
+
}
|
160
|
+
else if (timeout < 0) {
|
161
|
+
timeout = 1;
|
162
|
+
}
|
163
|
+
|
164
|
+
tv.tv_sec = timeout / 1000;
|
165
|
+
tv.tv_usec = (timeout * 1000) % 1000000;
|
166
|
+
|
167
|
+
/* load the fd sets from the multi handle */
|
168
|
+
mcode = curl_multi_fdset(curl_multi->multi, &fdread, &fdwrite, &fdexcep, &maxfd);
|
169
|
+
if (mcode != CURLM_OK) {
|
170
|
+
rb_raise((VALUE)mcode, "an error occured getting the fdset");
|
171
|
+
}
|
172
|
+
|
173
|
+
rc = rb_thread_select(maxfd+1, &fdread, &fdwrite, &fdexcep, &tv);
|
174
|
+
if (rc < 0) {
|
175
|
+
rb_raise(rb_eRuntimeError, "error on thread select");
|
176
|
+
}
|
177
|
+
rb_curl_multi_run( self, curl_multi->multi, &(curl_multi->running) );
|
178
|
+
|
179
|
+
}
|
180
|
+
|
181
|
+
return Qnil;
|
182
|
+
}
|
183
|
+
|
184
|
+
static VALUE active_handle_count(VALUE self) {
|
185
|
+
CurlMulti *curl_multi;
|
186
|
+
Data_Get_Struct(self, CurlMulti, curl_multi);
|
187
|
+
|
188
|
+
return INT2NUM(curl_multi->active);
|
189
|
+
}
|
190
|
+
|
191
|
+
static VALUE multi_cleanup(VALUE self) {
|
192
|
+
CurlMulti *curl_multi;
|
193
|
+
Data_Get_Struct(self, CurlMulti, curl_multi);
|
194
|
+
|
195
|
+
curl_multi_cleanup(curl_multi->multi);
|
196
|
+
curl_multi->active = 0;
|
197
|
+
curl_multi->running = 0;
|
198
|
+
|
199
|
+
return Qnil;
|
200
|
+
}
|
201
|
+
|
202
|
+
static VALUE new(int argc, VALUE *argv, VALUE klass) {
|
203
|
+
CurlMulti *curl_multi = ALLOC(CurlMulti);
|
204
|
+
curl_multi->multi = curl_multi_init();
|
205
|
+
curl_multi->active = 0;
|
206
|
+
curl_multi->running = 0;
|
207
|
+
|
208
|
+
VALUE multi = Data_Wrap_Struct(cTyphoeusMulti, 0, dealloc, curl_multi);
|
209
|
+
|
210
|
+
rb_obj_call_init(multi, argc, argv);
|
211
|
+
|
212
|
+
return multi;
|
213
|
+
}
|
214
|
+
|
215
|
+
void init_typhoeus_multi() {
|
216
|
+
VALUE klass = cTyphoeusMulti = rb_define_class_under(mTyphoeus, "Multi", rb_cObject);
|
217
|
+
|
218
|
+
rb_define_singleton_method(klass, "new", new, -1);
|
219
|
+
rb_define_private_method(klass, "multi_add_handle", multi_add_handle, 1);
|
220
|
+
rb_define_private_method(klass, "multi_remove_handle", multi_remove_handle, 1);
|
221
|
+
rb_define_private_method(klass, "multi_perform", multi_perform, 0);
|
222
|
+
rb_define_private_method(klass, "multi_cleanup", multi_cleanup, 0);
|
223
|
+
rb_define_private_method(klass, "active_handle_count", active_handle_count, 0);
|
224
|
+
rb_define_method(klass, "fire_and_forget", fire_and_forget, 0);
|
225
|
+
}
|
@@ -0,0 +1,16 @@
|
|
1
|
+
#ifndef TYPHOEUS_MULTI
|
2
|
+
#define TYPHOEUS_MULTI
|
3
|
+
|
4
|
+
#include <native.h>
|
5
|
+
#include <typhoeus_easy.h>
|
6
|
+
|
7
|
+
VALUE cTyphoeusMulti;
|
8
|
+
typedef struct {
|
9
|
+
int running;
|
10
|
+
int active;
|
11
|
+
CURLM *multi;
|
12
|
+
} CurlMulti;
|
13
|
+
|
14
|
+
void init_typhoeus_multi();
|
15
|
+
|
16
|
+
#endif
|
@@ -0,0 +1 @@
|
|
1
|
+
native.bundle
|