cool.io 0.9.0
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/.gitignore +25 -0
- data/CHANGES +199 -0
- data/LICENSE +20 -0
- data/README.markdown +4 -0
- data/Rakefile +98 -0
- data/VERSION +1 -0
- data/examples/echo_client.rb +38 -0
- data/examples/echo_server.rb +27 -0
- data/examples/google.rb +9 -0
- data/examples/httpclient.rb +38 -0
- data/ext/cool.io/.gitignore +5 -0
- data/ext/cool.io/cool.io.h +58 -0
- data/ext/cool.io/cool.io_ext.c +25 -0
- data/ext/cool.io/ev_wrap.h +8 -0
- data/ext/cool.io/extconf.rb +69 -0
- data/ext/cool.io/iowatcher.c +189 -0
- data/ext/cool.io/libev.c +8 -0
- data/ext/cool.io/loop.c +303 -0
- data/ext/cool.io/stat_watcher.c +191 -0
- data/ext/cool.io/timer_watcher.c +219 -0
- data/ext/cool.io/utils.c +122 -0
- data/ext/cool.io/watcher.c +264 -0
- data/ext/cool.io/watcher.h +71 -0
- data/ext/http11_client/.gitignore +5 -0
- data/ext/http11_client/ext_help.h +14 -0
- data/ext/http11_client/extconf.rb +6 -0
- data/ext/http11_client/http11_client.c +300 -0
- data/ext/http11_client/http11_parser.c +403 -0
- data/ext/http11_client/http11_parser.h +48 -0
- data/ext/http11_client/http11_parser.rl +173 -0
- data/ext/libev/Changes +364 -0
- data/ext/libev/LICENSE +36 -0
- data/ext/libev/README +58 -0
- data/ext/libev/README.embed +3 -0
- data/ext/libev/ev.c +3867 -0
- data/ext/libev/ev.h +826 -0
- data/ext/libev/ev_epoll.c +234 -0
- data/ext/libev/ev_kqueue.c +198 -0
- data/ext/libev/ev_poll.c +148 -0
- data/ext/libev/ev_port.c +164 -0
- data/ext/libev/ev_select.c +307 -0
- data/ext/libev/ev_vars.h +197 -0
- data/ext/libev/ev_win32.c +153 -0
- data/ext/libev/ev_wrap.h +186 -0
- data/ext/libev/test_libev_win32.c +123 -0
- data/ext/libev/update_ev_wrap +19 -0
- data/lib/.gitignore +2 -0
- data/lib/cool.io.rb +30 -0
- data/lib/cool.io/async_watcher.rb +43 -0
- data/lib/cool.io/dns_resolver.rb +220 -0
- data/lib/cool.io/eventmachine.rb +234 -0
- data/lib/cool.io/http_client.rb +419 -0
- data/lib/cool.io/io.rb +174 -0
- data/lib/cool.io/iowatcher.rb +17 -0
- data/lib/cool.io/listener.rb +93 -0
- data/lib/cool.io/loop.rb +130 -0
- data/lib/cool.io/meta.rb +49 -0
- data/lib/cool.io/server.rb +74 -0
- data/lib/cool.io/socket.rb +224 -0
- data/lib/cool.io/timer_watcher.rb +17 -0
- data/lib/coolio.rb +2 -0
- data/lib/rev.rb +4 -0
- data/spec/async_watcher_spec.rb +57 -0
- data/spec/possible_tests/schedules_other_threads.rb +48 -0
- data/spec/possible_tests/test_on_resolve_failed.rb +9 -0
- data/spec/possible_tests/test_resolves.rb +27 -0
- data/spec/possible_tests/test_write_during_resolve.rb +27 -0
- data/spec/possible_tests/works_straight.rb +71 -0
- data/spec/spec_helper.rb +5 -0
- data/spec/timer_watcher_spec.rb +55 -0
- data/spec/unix_listener_spec.rb +25 -0
- data/spec/unix_server_spec.rb +25 -0
- metadata +184 -0
@@ -0,0 +1,58 @@
|
|
1
|
+
/*
|
2
|
+
* Copyright (C) 2007-10 Tony Arcieri
|
3
|
+
* You may redistribute this under the terms of the Ruby license.
|
4
|
+
* See LICENSE for details
|
5
|
+
*/
|
6
|
+
|
7
|
+
#ifndef COOLIO_H
|
8
|
+
#define COOLIO_H
|
9
|
+
|
10
|
+
#include "ruby.h"
|
11
|
+
#include "rubyio.h"
|
12
|
+
|
13
|
+
#ifdef GetReadFile
|
14
|
+
#define FPTR_TO_FD(fptr) (fileno(GetReadFile(fptr)))
|
15
|
+
#else
|
16
|
+
|
17
|
+
#if !HAVE_RB_IO_T || (RUBY_VERSION_MAJOR == 1 && RUBY_VERSION_MINOR == 8)
|
18
|
+
#define FPTR_TO_FD(fptr) fileno(fptr->f)
|
19
|
+
#else
|
20
|
+
#define FPTR_TO_FD(fptr) fptr->fd
|
21
|
+
#endif
|
22
|
+
|
23
|
+
#endif
|
24
|
+
|
25
|
+
struct Coolio_Event
|
26
|
+
{
|
27
|
+
/* These values are used to extract events from libev callbacks */
|
28
|
+
VALUE watcher;
|
29
|
+
int revents;
|
30
|
+
};
|
31
|
+
|
32
|
+
struct Coolio_Loop
|
33
|
+
{
|
34
|
+
struct ev_loop *ev_loop;
|
35
|
+
|
36
|
+
int running;
|
37
|
+
int events_received;
|
38
|
+
int eventbuf_size;
|
39
|
+
struct Coolio_Event *eventbuf;
|
40
|
+
};
|
41
|
+
|
42
|
+
struct Coolio_Watcher
|
43
|
+
{
|
44
|
+
union {
|
45
|
+
struct ev_io ev_io;
|
46
|
+
struct ev_timer ev_timer;
|
47
|
+
struct ev_stat ev_stat;
|
48
|
+
} event_types;
|
49
|
+
|
50
|
+
int enabled;
|
51
|
+
VALUE loop;
|
52
|
+
|
53
|
+
void (*dispatch_callback)(VALUE self, int revents);
|
54
|
+
};
|
55
|
+
|
56
|
+
void Coolio_Loop_process_event(VALUE watcher, int revents);
|
57
|
+
|
58
|
+
#endif
|
@@ -0,0 +1,25 @@
|
|
1
|
+
/*
|
2
|
+
* Copyright (C) 2007-10 Tony Arcieri
|
3
|
+
* You may redistribute this under the terms of the Ruby license.
|
4
|
+
* See LICENSE for details
|
5
|
+
*/
|
6
|
+
|
7
|
+
|
8
|
+
#include "ruby.h"
|
9
|
+
|
10
|
+
#include "ev_wrap.h"
|
11
|
+
#include "cool.io.h"
|
12
|
+
|
13
|
+
static VALUE mCoolio = Qnil;
|
14
|
+
|
15
|
+
/* Initialize the coolness */
|
16
|
+
void Init_cool()
|
17
|
+
{
|
18
|
+
/* Initializers for other modules */
|
19
|
+
Init_coolio_loop();
|
20
|
+
Init_coolio_watcher();
|
21
|
+
Init_coolio_iowatcher();
|
22
|
+
Init_coolio_timer_watcher();
|
23
|
+
Init_coolio_stat_watcher();
|
24
|
+
Init_coolio_utils();
|
25
|
+
}
|
@@ -0,0 +1,8 @@
|
|
1
|
+
#define EV_STANDALONE /* keeps ev from requiring config.h */
|
2
|
+
#ifdef _WIN32
|
3
|
+
# define EV_SELECT_IS_WINSOCKET 1 /* configure libev for windows select */
|
4
|
+
# define FD_SETSIZE 2048 /* wishful thinking, as msvcrt6 [?] seems to only allow 512 fd's and 256 sockets max */
|
5
|
+
#endif
|
6
|
+
|
7
|
+
#include "../libev/ev.h"
|
8
|
+
|
@@ -0,0 +1,69 @@
|
|
1
|
+
require 'mkmf'
|
2
|
+
|
3
|
+
libs = []
|
4
|
+
|
5
|
+
$defs << "-DRUBY_VERSION_CODE=#{RUBY_VERSION.gsub(/\D/, '')}"
|
6
|
+
|
7
|
+
if have_func('rb_thread_blocking_region')
|
8
|
+
$defs << '-DHAVE_RB_THREAD_BLOCKING_REGION'
|
9
|
+
end
|
10
|
+
|
11
|
+
if have_func('rb_thread_alone')
|
12
|
+
$defs << '-DHAVE_RB_THREAD_ALONE'
|
13
|
+
end
|
14
|
+
|
15
|
+
if have_func('rb_str_set_len')
|
16
|
+
$defs << '-DHAVE_RB_STR_SET_LEN'
|
17
|
+
end
|
18
|
+
|
19
|
+
if have_library('rt', 'clock_gettime')
|
20
|
+
libs << "-lrt"
|
21
|
+
end
|
22
|
+
|
23
|
+
if have_header('sys/select.h')
|
24
|
+
$defs << '-DEV_USE_SELECT'
|
25
|
+
end
|
26
|
+
|
27
|
+
if have_header('poll.h')
|
28
|
+
$defs << '-DEV_USE_POLL'
|
29
|
+
end
|
30
|
+
|
31
|
+
if have_header('sys/epoll.h')
|
32
|
+
$defs << '-DEV_USE_EPOLL'
|
33
|
+
end
|
34
|
+
|
35
|
+
if have_header('sys/event.h') and have_header('sys/queue.h')
|
36
|
+
$defs << '-DEV_USE_KQUEUE'
|
37
|
+
end
|
38
|
+
|
39
|
+
if have_header('port.h')
|
40
|
+
$defs << '-DEV_USE_PORT'
|
41
|
+
end
|
42
|
+
|
43
|
+
if have_header('sys/resource.h')
|
44
|
+
$defs << '-DHAVE_SYS_RESOURCE_H'
|
45
|
+
end
|
46
|
+
|
47
|
+
# ncpu detection specifics
|
48
|
+
case RUBY_PLATFORM
|
49
|
+
when /linux/
|
50
|
+
$defs << '-DHAVE_LINUX_PROCFS'
|
51
|
+
else
|
52
|
+
if have_func('sysctlbyname', ['sys/param.h', 'sys/sysctl.h'])
|
53
|
+
$defs << '-DHAVE_SYSCTLBYNAME'
|
54
|
+
end
|
55
|
+
end
|
56
|
+
|
57
|
+
$LIBS << ' ' << libs.join(' ')
|
58
|
+
|
59
|
+
dir_config('cool.io_ext')
|
60
|
+
create_makefile('cool.io_ext')
|
61
|
+
|
62
|
+
# win32 needs to link in "just the right order" for some reason or ioctlsocket will be mapped to an [inverted] ruby specific version. See libev mailing list for (not so helpful discussion--true cause I'm not sure, but this overcomes the symptom)
|
63
|
+
if RUBY_PLATFORM =~ /mingw|win32/
|
64
|
+
makefile_contents = File.read 'Makefile'
|
65
|
+
|
66
|
+
makefile_contents.gsub! 'LIBS = $(LIBRUBYARG_SHARED)', 'LIBS = -lws2_32 $(LIBRUBYARG_SHARED)'
|
67
|
+
File.open('Makefile', 'w') { |f| f.write makefile_contents }
|
68
|
+
end
|
69
|
+
|
@@ -0,0 +1,189 @@
|
|
1
|
+
/*
|
2
|
+
* Copyright (C) 2007-10 Tony Arcieri
|
3
|
+
* You may redistribute this under the terms of the Ruby license.
|
4
|
+
* See LICENSE for details
|
5
|
+
*/
|
6
|
+
|
7
|
+
#include "ruby.h"
|
8
|
+
#include "rubyio.h"
|
9
|
+
|
10
|
+
#include "ev_wrap.h"
|
11
|
+
|
12
|
+
#include "cool.io.h"
|
13
|
+
#include "watcher.h"
|
14
|
+
|
15
|
+
static VALUE mCoolio = Qnil;
|
16
|
+
static VALUE cCoolio_Watcher = Qnil;
|
17
|
+
static VALUE cCoolio_Loop = Qnil;
|
18
|
+
static VALUE cCoolio_IOWatcher = Qnil;
|
19
|
+
|
20
|
+
static VALUE Coolio_IOWatcher_initialize(int argc, VALUE *argv, VALUE self);
|
21
|
+
static VALUE Coolio_IOWatcher_attach(VALUE self, VALUE loop);
|
22
|
+
static VALUE Coolio_IOWatcher_detach(VALUE self);
|
23
|
+
static VALUE Coolio_IOWatcher_enable(VALUE self);
|
24
|
+
static VALUE Coolio_IOWatcher_disable(VALUE self);
|
25
|
+
static VALUE Coolio_IOWatcher_on_readable(VALUE self);
|
26
|
+
static VALUE Coolio_IOWatcher_on_writable(VALUE self);
|
27
|
+
|
28
|
+
static void Coolio_IOWatcher_libev_callback(struct ev_loop *ev_loop, struct ev_io *io, int revents);
|
29
|
+
static void Coolio_IOWatcher_dispatch_callback(VALUE self, int revents);
|
30
|
+
|
31
|
+
/*
|
32
|
+
* Coolio::IOWatcher monitors Ruby IO objects for readability or writability.
|
33
|
+
* This allows your application to block while the kernel is writing out
|
34
|
+
* data and fill the read or write buffer whenever there is space available.
|
35
|
+
*/
|
36
|
+
void Init_coolio_iowatcher()
|
37
|
+
{
|
38
|
+
mCoolio = rb_define_module("Coolio");
|
39
|
+
cCoolio_Watcher = rb_define_class_under(mCoolio, "Watcher", rb_cObject);
|
40
|
+
cCoolio_IOWatcher = rb_define_class_under(mCoolio, "IOWatcher", cCoolio_Watcher);
|
41
|
+
cCoolio_Loop = rb_define_class_under(mCoolio, "Loop", rb_cObject);
|
42
|
+
|
43
|
+
rb_define_method(cCoolio_IOWatcher, "initialize", Coolio_IOWatcher_initialize, -1);
|
44
|
+
rb_define_method(cCoolio_IOWatcher, "attach", Coolio_IOWatcher_attach, 1);
|
45
|
+
rb_define_method(cCoolio_IOWatcher, "detach", Coolio_IOWatcher_detach, 0);
|
46
|
+
rb_define_method(cCoolio_IOWatcher, "enable", Coolio_IOWatcher_enable, 0);
|
47
|
+
rb_define_method(cCoolio_IOWatcher, "disable", Coolio_IOWatcher_disable, 0);
|
48
|
+
rb_define_method(cCoolio_IOWatcher, "on_readable", Coolio_IOWatcher_on_readable, 0);
|
49
|
+
rb_define_method(cCoolio_IOWatcher, "on_writable", Coolio_IOWatcher_on_writable, 0);
|
50
|
+
}
|
51
|
+
|
52
|
+
/**
|
53
|
+
* call-seq:
|
54
|
+
* Coolio::IOWatcher.initialize(IO, events = 'r') -> Coolio::IOWatcher
|
55
|
+
*
|
56
|
+
* Create a new Coolio::IOWatcher for the given IO object and add it to the given Coolio::Loop
|
57
|
+
*/
|
58
|
+
static VALUE Coolio_IOWatcher_initialize(int argc, VALUE *argv, VALUE self)
|
59
|
+
{
|
60
|
+
VALUE io, flags;
|
61
|
+
char *flags_str;
|
62
|
+
int events;
|
63
|
+
struct Coolio_Watcher *watcher_data;
|
64
|
+
#if HAVE_RB_IO_T
|
65
|
+
rb_io_t *fptr;
|
66
|
+
#else
|
67
|
+
OpenFile *fptr;
|
68
|
+
#endif
|
69
|
+
|
70
|
+
rb_scan_args(argc, argv, "11", &io, &flags);
|
71
|
+
|
72
|
+
if(flags != Qnil)
|
73
|
+
flags_str = RSTRING_PTR(rb_String(flags));
|
74
|
+
else
|
75
|
+
flags_str = "r";
|
76
|
+
|
77
|
+
if(!strcmp(flags_str, "r"))
|
78
|
+
events = EV_READ;
|
79
|
+
else if(!strcmp(flags_str, "w"))
|
80
|
+
events = EV_WRITE;
|
81
|
+
else if(!strcmp(flags_str, "rw"))
|
82
|
+
events = EV_READ | EV_WRITE;
|
83
|
+
else
|
84
|
+
rb_raise(rb_eArgError, "invalid event type: '%s' (must be 'r', 'w', or 'rw')", flags_str);
|
85
|
+
|
86
|
+
Data_Get_Struct(self, struct Coolio_Watcher, watcher_data);
|
87
|
+
GetOpenFile(rb_convert_type(io, T_FILE, "IO", "to_io"), fptr);
|
88
|
+
|
89
|
+
watcher_data->dispatch_callback = Coolio_IOWatcher_dispatch_callback;
|
90
|
+
ev_io_init(&watcher_data->event_types.ev_io, Coolio_IOWatcher_libev_callback, FPTR_TO_FD(fptr), events);
|
91
|
+
watcher_data->event_types.ev_io.data = (void *)self;
|
92
|
+
|
93
|
+
return Qnil;
|
94
|
+
}
|
95
|
+
|
96
|
+
/**
|
97
|
+
* call-seq:
|
98
|
+
* Coolio::IOWatcher.attach(loop) -> Coolio::IOWatcher
|
99
|
+
*
|
100
|
+
* Attach the IO watcher to the given Coolio::Loop. If the watcher is already attached
|
101
|
+
* to a loop, detach it from the old one and attach it to the new one.
|
102
|
+
*/
|
103
|
+
static VALUE Coolio_IOWatcher_attach(VALUE self, VALUE loop)
|
104
|
+
{
|
105
|
+
Watcher_Attach(io, Coolio_IOWatcher_detach, self, loop);
|
106
|
+
|
107
|
+
return self;
|
108
|
+
}
|
109
|
+
|
110
|
+
/**
|
111
|
+
* call-seq:
|
112
|
+
* Coolio::IOWatcher.detach -> Coolio::IOWatcher
|
113
|
+
*
|
114
|
+
* Detach the IO watcher from its current Coolio::Loop.
|
115
|
+
*/
|
116
|
+
static VALUE Coolio_IOWatcher_detach(VALUE self)
|
117
|
+
{
|
118
|
+
Watcher_Detach(io, self);
|
119
|
+
|
120
|
+
return self;
|
121
|
+
}
|
122
|
+
|
123
|
+
/**
|
124
|
+
* call-seq:
|
125
|
+
* Coolio::IOWatcher.enable -> Coolio::IOWatcher
|
126
|
+
*
|
127
|
+
* Re-enable an IO watcher which has been temporarily disabled. See the
|
128
|
+
* disable method for a more thorough explanation.
|
129
|
+
*/
|
130
|
+
static VALUE Coolio_IOWatcher_enable(VALUE self)
|
131
|
+
{
|
132
|
+
Watcher_Enable(io, self);
|
133
|
+
|
134
|
+
return self;
|
135
|
+
}
|
136
|
+
|
137
|
+
/**
|
138
|
+
* call-seq:
|
139
|
+
* Coolio::IOWatcher.disable -> Coolio::IOWatcher
|
140
|
+
*
|
141
|
+
* Temporarily disable an IO watcher which is attached to a loop.
|
142
|
+
* This is useful if you wish to toggle event monitoring on and off.
|
143
|
+
*/
|
144
|
+
static VALUE Coolio_IOWatcher_disable(VALUE self)
|
145
|
+
{
|
146
|
+
Watcher_Disable(io, self);
|
147
|
+
|
148
|
+
return self;
|
149
|
+
}
|
150
|
+
|
151
|
+
/**
|
152
|
+
* call-seq:
|
153
|
+
* Coolio::IOWatcher#on_readable -> nil
|
154
|
+
*
|
155
|
+
* Called whenever the IO object associated with the IOWatcher is readable
|
156
|
+
*/
|
157
|
+
static VALUE Coolio_IOWatcher_on_readable(VALUE self)
|
158
|
+
{
|
159
|
+
return Qnil;
|
160
|
+
}
|
161
|
+
|
162
|
+
/**
|
163
|
+
* call-seq:
|
164
|
+
* Coolio::IOWatcher#on_writable -> nil
|
165
|
+
*
|
166
|
+
* Called whenever the IO object associated with the IOWatcher is writable
|
167
|
+
*/
|
168
|
+
|
169
|
+
static VALUE Coolio_IOWatcher_on_writable(VALUE self)
|
170
|
+
{
|
171
|
+
return Qnil;
|
172
|
+
}
|
173
|
+
|
174
|
+
/* libev callback */
|
175
|
+
static void Coolio_IOWatcher_libev_callback(struct ev_loop *ev_loop, struct ev_io *io, int revents)
|
176
|
+
{
|
177
|
+
Coolio_Loop_process_event((VALUE)io->data, revents);
|
178
|
+
}
|
179
|
+
|
180
|
+
/* Coolio::Loop dispatch callback */
|
181
|
+
static void Coolio_IOWatcher_dispatch_callback(VALUE self, int revents)
|
182
|
+
{
|
183
|
+
if(revents & EV_READ)
|
184
|
+
rb_funcall(self, rb_intern("on_readable"), 0, 0);
|
185
|
+
else if(revents & EV_WRITE)
|
186
|
+
rb_funcall(self, rb_intern("on_writable"), 0, 0);
|
187
|
+
else
|
188
|
+
rb_raise(rb_eRuntimeError, "unknown revents value for ev_io: %d", revents);
|
189
|
+
}
|
data/ext/cool.io/libev.c
ADDED
data/ext/cool.io/loop.c
ADDED
@@ -0,0 +1,303 @@
|
|
1
|
+
/*
|
2
|
+
* Copyright (C) 2007-10 Tony Arcieri
|
3
|
+
* You may redistribute this under the terms of the Ruby license.
|
4
|
+
* See LICENSE for details
|
5
|
+
*/
|
6
|
+
|
7
|
+
#include <assert.h>
|
8
|
+
#include "ruby.h"
|
9
|
+
#include "rubysig.h"
|
10
|
+
#include "ev_wrap.h"
|
11
|
+
|
12
|
+
#include "cool.io.h"
|
13
|
+
|
14
|
+
#if defined(HAVE_RB_THREAD_BLOCKING_REGION)
|
15
|
+
# define Coolio_Loop_may_block_safely() (1)
|
16
|
+
#elif defined(HAVE_RB_THREAD_ALONE)
|
17
|
+
# define Coolio_Loop_may_block_safely() (rb_thread_alone())
|
18
|
+
#else /* just in case Ruby changes: */
|
19
|
+
# define Coolio_Loop_may_block_safely() (0)
|
20
|
+
#endif
|
21
|
+
|
22
|
+
static VALUE mCoolio = Qnil;
|
23
|
+
static VALUE cCoolio_Loop = Qnil;
|
24
|
+
|
25
|
+
static VALUE Coolio_Loop_allocate(VALUE klass);
|
26
|
+
static void Coolio_Loop_mark(struct Coolio_Loop *loop);
|
27
|
+
static void Coolio_Loop_free(struct Coolio_Loop *loop);
|
28
|
+
|
29
|
+
static VALUE Coolio_Loop_initialize(VALUE self);
|
30
|
+
static VALUE Coolio_Loop_ev_loop_new(VALUE self, VALUE flags);
|
31
|
+
static VALUE Coolio_Loop_run_once(VALUE self);
|
32
|
+
static VALUE Coolio_Loop_run_nonblock(VALUE self);
|
33
|
+
|
34
|
+
static void Coolio_Loop_ev_loop_oneshot(struct Coolio_Loop *loop_data);
|
35
|
+
static void Coolio_Loop_dispatch_events(struct Coolio_Loop *loop_data);
|
36
|
+
|
37
|
+
#define DEFAULT_EVENTBUF_SIZE 32
|
38
|
+
#define RUN_LOOP(loop_data, options) \
|
39
|
+
loop_data->running = 1; \
|
40
|
+
ev_loop(loop_data->ev_loop, options); \
|
41
|
+
loop_data->running = 0;
|
42
|
+
|
43
|
+
/*
|
44
|
+
* Coolio::Loop represents an event loop. Event watchers can be attached and
|
45
|
+
* unattached. When an event loop is run, all currently attached watchers
|
46
|
+
* are monitored for events, and their respective callbacks are signaled
|
47
|
+
* whenever events occur.
|
48
|
+
*/
|
49
|
+
void Init_coolio_loop()
|
50
|
+
{
|
51
|
+
mCoolio = rb_define_module("Coolio");
|
52
|
+
cCoolio_Loop = rb_define_class_under(mCoolio, "Loop", rb_cObject);
|
53
|
+
rb_define_alloc_func(cCoolio_Loop, Coolio_Loop_allocate);
|
54
|
+
|
55
|
+
rb_define_method(cCoolio_Loop, "initialize", Coolio_Loop_initialize, 0);
|
56
|
+
rb_define_private_method(cCoolio_Loop, "ev_loop_new", Coolio_Loop_ev_loop_new, 1);
|
57
|
+
rb_define_method(cCoolio_Loop, "run_once", Coolio_Loop_run_once, 0);
|
58
|
+
rb_define_method(cCoolio_Loop, "run_nonblock", Coolio_Loop_run_nonblock, 0);
|
59
|
+
}
|
60
|
+
|
61
|
+
static VALUE Coolio_Loop_allocate(VALUE klass)
|
62
|
+
{
|
63
|
+
struct Coolio_Loop *loop = (struct Coolio_Loop *)xmalloc(sizeof(struct Coolio_Loop));
|
64
|
+
|
65
|
+
loop->ev_loop = 0;
|
66
|
+
loop->running = 0;
|
67
|
+
loop->events_received = 0;
|
68
|
+
loop->eventbuf_size = DEFAULT_EVENTBUF_SIZE;
|
69
|
+
loop->eventbuf = (struct Coolio_Event *)xmalloc(sizeof(struct Coolio_Event) * DEFAULT_EVENTBUF_SIZE);
|
70
|
+
|
71
|
+
return Data_Wrap_Struct(klass, Coolio_Loop_mark, Coolio_Loop_free, loop);
|
72
|
+
}
|
73
|
+
|
74
|
+
static void Coolio_Loop_mark(struct Coolio_Loop *loop)
|
75
|
+
{
|
76
|
+
}
|
77
|
+
|
78
|
+
static void Coolio_Loop_free(struct Coolio_Loop *loop)
|
79
|
+
{
|
80
|
+
if(!loop->ev_loop)
|
81
|
+
return;
|
82
|
+
|
83
|
+
ev_loop_destroy(loop->ev_loop);
|
84
|
+
|
85
|
+
xfree(loop->eventbuf);
|
86
|
+
xfree(loop);
|
87
|
+
}
|
88
|
+
|
89
|
+
static VALUE Coolio_Loop_initialize(VALUE self)
|
90
|
+
{
|
91
|
+
Coolio_Loop_ev_loop_new(self, INT2NUM(0));
|
92
|
+
}
|
93
|
+
|
94
|
+
/* Wrapper for populating a Coolio_Loop struct with a new event loop */
|
95
|
+
static VALUE Coolio_Loop_ev_loop_new(VALUE self, VALUE flags)
|
96
|
+
{
|
97
|
+
struct Coolio_Loop *loop_data;
|
98
|
+
Data_Get_Struct(self, struct Coolio_Loop, loop_data);
|
99
|
+
|
100
|
+
if(loop_data->ev_loop)
|
101
|
+
rb_raise(rb_eRuntimeError, "loop already initialized");
|
102
|
+
|
103
|
+
loop_data->ev_loop = ev_loop_new(NUM2INT(flags));
|
104
|
+
|
105
|
+
return Qnil;
|
106
|
+
}
|
107
|
+
|
108
|
+
/* libev callback for receiving events */
|
109
|
+
void Coolio_Loop_process_event(VALUE watcher, int revents)
|
110
|
+
{
|
111
|
+
struct Coolio_Loop *loop_data;
|
112
|
+
struct Coolio_Watcher *watcher_data;
|
113
|
+
|
114
|
+
/* The Global VM lock isn't held right now, but hopefully
|
115
|
+
* we can still do this safely */
|
116
|
+
Data_Get_Struct(watcher, struct Coolio_Watcher, watcher_data);
|
117
|
+
Data_Get_Struct(watcher_data->loop, struct Coolio_Loop, loop_data);
|
118
|
+
|
119
|
+
/* Well, what better place to explain how this all works than
|
120
|
+
* where the most wonky and convoluted stuff is going on!
|
121
|
+
*
|
122
|
+
* Our call path up to here looks a little something like:
|
123
|
+
*
|
124
|
+
* -> release GVL -> event syscall -> libev callback
|
125
|
+
* (GVL = Global VM Lock) ^^^ You are here
|
126
|
+
*
|
127
|
+
* We released the GVL in the Coolio_Loop_run_once() function
|
128
|
+
* so other Ruby threads can run while we make a blocking
|
129
|
+
* system call (one of epoll, kqueue, port, poll, or select,
|
130
|
+
* depending on the platform).
|
131
|
+
*
|
132
|
+
* More specifically, this is a libev callback abstraction
|
133
|
+
* called from a real libev callback in every watcher,
|
134
|
+
* hence this function not being static. The real libev
|
135
|
+
* callbacks are event-specific and handled in a watcher.
|
136
|
+
*
|
137
|
+
* For syscalls like epoll and kqueue, the kernel tells libev
|
138
|
+
* a pointer (to a structure with a pointer) to the watcher
|
139
|
+
* object. No data structure lookups are required at all
|
140
|
+
* (beyond structs), it's smooth O(1) sailing the entire way.
|
141
|
+
* Then libev calls out to the watcher's callback, which
|
142
|
+
* calls this function.
|
143
|
+
*
|
144
|
+
* Now, you may be curious: if the watcher already knew what
|
145
|
+
* event fired, why the hell is it telling the loop? Why
|
146
|
+
* doesn't it just rb_funcall() the appropriate callback?
|
147
|
+
*
|
148
|
+
* Well, the problem is the Global VM Lock isn't held right
|
149
|
+
* now, so we can't rb_funcall() anything. In order to get
|
150
|
+
* it back we have to:
|
151
|
+
*
|
152
|
+
* stash event and return -> acquire GVL -> dispatch to Ruby
|
153
|
+
*
|
154
|
+
* Which is kinda ugly and confusing, but still gives us
|
155
|
+
* an O(1) event loop whose heart is in the kernel itself. w00t!
|
156
|
+
*
|
157
|
+
* So, stash the event in the loop's data struct. When we return
|
158
|
+
* the ev_loop() call being made in the Coolio_Loop_run_once_blocking()
|
159
|
+
* function below will also return, at which point the GVL is
|
160
|
+
* reacquired and we can call out to Ruby */
|
161
|
+
|
162
|
+
/* Grow the event buffer if it's too small */
|
163
|
+
if(loop_data->events_received >= loop_data->eventbuf_size) {
|
164
|
+
loop_data->eventbuf_size *= 2;
|
165
|
+
loop_data->eventbuf = (struct Coolio_Event *)xrealloc(
|
166
|
+
loop_data->eventbuf,
|
167
|
+
sizeof(struct Coolio_Event) * loop_data->eventbuf_size
|
168
|
+
);
|
169
|
+
}
|
170
|
+
|
171
|
+
loop_data->eventbuf[loop_data->events_received].watcher = watcher;
|
172
|
+
loop_data->eventbuf[loop_data->events_received].revents = revents;
|
173
|
+
|
174
|
+
loop_data->events_received++;
|
175
|
+
}
|
176
|
+
|
177
|
+
/**
|
178
|
+
* call-seq:
|
179
|
+
* Coolio::Loop.run_once -> nil
|
180
|
+
*
|
181
|
+
* Run the Coolio::Loop once, blocking until events are received.
|
182
|
+
*/
|
183
|
+
static VALUE Coolio_Loop_run_once(VALUE self)
|
184
|
+
{
|
185
|
+
VALUE nevents;
|
186
|
+
|
187
|
+
if (Coolio_Loop_may_block_safely()) {
|
188
|
+
struct Coolio_Loop *loop_data;
|
189
|
+
|
190
|
+
Data_Get_Struct(self, struct Coolio_Loop, loop_data);
|
191
|
+
|
192
|
+
assert(loop_data->ev_loop && !loop_data->events_received);
|
193
|
+
|
194
|
+
Coolio_Loop_ev_loop_oneshot(loop_data);
|
195
|
+
Coolio_Loop_dispatch_events(loop_data);
|
196
|
+
|
197
|
+
nevents = INT2NUM(loop_data->events_received);
|
198
|
+
loop_data->events_received = 0;
|
199
|
+
|
200
|
+
} else {
|
201
|
+
nevents = Coolio_Loop_run_nonblock(self);
|
202
|
+
rb_thread_schedule();
|
203
|
+
}
|
204
|
+
|
205
|
+
return nevents;
|
206
|
+
}
|
207
|
+
|
208
|
+
/* Ruby 1.9 supports blocking system calls through rb_thread_blocking_region() */
|
209
|
+
#ifdef HAVE_RB_THREAD_BLOCKING_REGION
|
210
|
+
#define HAVE_EV_LOOP_ONESHOT
|
211
|
+
static VALUE Coolio_Loop_ev_loop_oneshot_blocking(void *ptr)
|
212
|
+
{
|
213
|
+
/* The libev loop has now escaped through the Global VM Lock unscathed! */
|
214
|
+
struct Coolio_Loop *loop_data = (struct Coolio_Loop *)ptr;
|
215
|
+
|
216
|
+
RUN_LOOP(loop_data, EVLOOP_ONESHOT);
|
217
|
+
|
218
|
+
return Qnil;
|
219
|
+
}
|
220
|
+
|
221
|
+
static void Coolio_Loop_ev_loop_oneshot(struct Coolio_Loop *loop_data)
|
222
|
+
{
|
223
|
+
/* Use Ruby 1.9's rb_thread_blocking_region call to make a blocking system call */
|
224
|
+
rb_thread_blocking_region(Coolio_Loop_ev_loop_oneshot_blocking, loop_data, RUBY_UBF_IO, 0);
|
225
|
+
}
|
226
|
+
#endif
|
227
|
+
|
228
|
+
/* Ruby 1.8 requires us to periodically run the event loop then defer back to
|
229
|
+
* the green threads scheduler */
|
230
|
+
#ifndef HAVE_EV_LOOP_ONESHOT
|
231
|
+
#define BLOCKING_INTERVAL 0.01 /* Block for 10ms at a time */
|
232
|
+
|
233
|
+
/* Stub for scheduler's ev_timer callback */
|
234
|
+
static void timer_callback(struct ev_loop *ev_loop, struct ev_timer *timer, int revents)
|
235
|
+
{
|
236
|
+
ev_timer_again (ev_loop, timer);
|
237
|
+
}
|
238
|
+
|
239
|
+
/* Run the event loop, calling rb_thread_schedule every 10ms */
|
240
|
+
static void Coolio_Loop_ev_loop_oneshot(struct Coolio_Loop *loop_data)
|
241
|
+
{
|
242
|
+
struct ev_timer timer;
|
243
|
+
struct timeval tv;
|
244
|
+
|
245
|
+
/* Set up an ev_timer to unblock the loop every 10ms */
|
246
|
+
ev_timer_init(&timer, timer_callback, BLOCKING_INTERVAL, BLOCKING_INTERVAL);
|
247
|
+
ev_timer_start(loop_data->ev_loop, &timer);
|
248
|
+
|
249
|
+
/* Loop until we receive events */
|
250
|
+
while(!loop_data->events_received) {
|
251
|
+
TRAP_BEG;
|
252
|
+
RUN_LOOP(loop_data, EVLOOP_ONESHOT);
|
253
|
+
TRAP_END;
|
254
|
+
|
255
|
+
rb_thread_schedule();
|
256
|
+
}
|
257
|
+
|
258
|
+
ev_timer_stop(loop_data->ev_loop, &timer);
|
259
|
+
}
|
260
|
+
#endif
|
261
|
+
|
262
|
+
/**
|
263
|
+
* call-seq:
|
264
|
+
* Coolio::Loop.run_nonblock -> nil
|
265
|
+
*
|
266
|
+
* Run the Coolio::Loop once, but return immediately if there are no pending events.
|
267
|
+
*/
|
268
|
+
static VALUE Coolio_Loop_run_nonblock(VALUE self)
|
269
|
+
{
|
270
|
+
struct Coolio_Loop *loop_data;
|
271
|
+
VALUE nevents;
|
272
|
+
|
273
|
+
Data_Get_Struct(self, struct Coolio_Loop, loop_data);
|
274
|
+
|
275
|
+
assert(loop_data->ev_loop && !loop_data->events_received);
|
276
|
+
|
277
|
+
TRAP_BEG;
|
278
|
+
RUN_LOOP(loop_data, EVLOOP_NONBLOCK);
|
279
|
+
TRAP_END;
|
280
|
+
Coolio_Loop_dispatch_events(loop_data);
|
281
|
+
|
282
|
+
nevents = INT2NUM(loop_data->events_received);
|
283
|
+
loop_data->events_received = 0;
|
284
|
+
|
285
|
+
return nevents;
|
286
|
+
}
|
287
|
+
|
288
|
+
static void Coolio_Loop_dispatch_events(struct Coolio_Loop *loop_data)
|
289
|
+
{
|
290
|
+
int i;
|
291
|
+
struct Coolio_Watcher *watcher_data;
|
292
|
+
|
293
|
+
for(i = 0; i < loop_data->events_received; i++) {
|
294
|
+
/* A watcher with pending events may have been detached from the loop
|
295
|
+
* during the dispatch process. If so, the watcher clears the pending
|
296
|
+
* events, so skip over them */
|
297
|
+
if(loop_data->eventbuf[i].watcher == Qnil)
|
298
|
+
continue;
|
299
|
+
|
300
|
+
Data_Get_Struct(loop_data->eventbuf[i].watcher, struct Coolio_Watcher, watcher_data);
|
301
|
+
watcher_data->dispatch_callback(loop_data->eventbuf[i].watcher, loop_data->eventbuf[i].revents);
|
302
|
+
}
|
303
|
+
}
|