ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/IO-AIO/AIO.pm
(Generate patch)

Comparing IO-AIO/AIO.pm (file contents):
Revision 1.1 by root, Sun Jul 10 17:07:44 2005 UTC vs.
Revision 1.82 by root, Fri Oct 27 20:10:06 2006 UTC

4 4
5=head1 SYNOPSIS 5=head1 SYNOPSIS
6 6
7 use IO::AIO; 7 use IO::AIO;
8 8
9 aio_open "/etc/passwd", O_RDONLY, 0, sub {
10 my ($fh) = @_;
11 ...
12 };
13
14 aio_unlink "/tmp/file", sub { };
15
16 aio_read $fh, 30000, 1024, $buffer, 0, sub {
17 $_[0] > 0 or die "read error: $!";
18 };
19
20 # version 2+ has request and group objects
21 use IO::AIO 2;
22
23 aioreq_pri 4; # give next request a very high priority
24 my $req = aio_unlink "/tmp/file", sub { };
25 $req->cancel; # cancel request if still in queue
26
27 my $grp = aio_group sub { print "all stats done\n" };
28 add $grp aio_stat "..." for ...;
29
30 # AnyEvent integration
31 open my $fh, "<&=" . IO::AIO::poll_fileno or die "$!";
32 my $w = AnyEvent->io (fh => $fh, poll => 'r', cb => sub { IO::AIO::poll_cb });
33
34 # Event integration
35 Event->io (fd => IO::AIO::poll_fileno,
36 poll => 'r',
37 cb => \&IO::AIO::poll_cb);
38
39 # Glib/Gtk2 integration
40 add_watch Glib::IO IO::AIO::poll_fileno,
41 in => sub { IO::AIO::poll_cb; 1 };
42
43 # Tk integration
44 Tk::Event::IO->fileevent (IO::AIO::poll_fileno, "",
45 readable => \&IO::AIO::poll_cb);
46
47 # Danga::Socket integration
48 Danga::Socket->AddOtherFds (IO::AIO::poll_fileno =>
49 \&IO::AIO::poll_cb);
50
9=head1 DESCRIPTION 51=head1 DESCRIPTION
10 52
11This module implements asynchronous I/O using whatever means your 53This module implements asynchronous I/O using whatever means your
12operating system supports. Currently, it falls back to Linux::AIO if that 54operating system supports.
13module is available, or uses pthreads to emulato aio functionality.
14 55
15Currently, in this module a number of threads are started that execute 56In this version, a number of threads are started that execute your
16your read/writes and signal their completion. You don't need thread 57requests and signal their completion. You don't need thread support
17support in your libc or perl, and the threads created by this module will 58in perl, and the threads created by this module will not be visible
18not be visible to the pthreads library. 59to perl. In the future, this module might make use of the native aio
60functions available on many operating systems. However, they are often
61not well-supported or restricted (Linux doesn't allow them on normal
62files currently, for example), and they would only support aio_read and
63aio_write, so the remaining functionality would have to be implemented
64using threads anyway.
19 65
20Although the module will work with in the presence of other threads, it is 66Although the module will work with in the presence of other (Perl-)
21not reentrant, so use appropriate locking yourself. 67threads, it is currently not reentrant in any way, so use appropriate
68locking yourself, always call C<poll_cb> from within the same thread, or
69never call C<poll_cb> (or other C<aio_> functions) recursively.
22 70
23=head2 API NOTES 71=head1 REQUEST ANATOMY AND LIFETIME
72
73Every C<aio_*> function creates a request. which is a C data structure not
74directly visible to Perl.
75
76If called in non-void context, every request function returns a Perl
77object representing the request. In void context, nothing is returned,
78which saves a bit of memory.
79
80The perl object is a fairly standard ref-to-hash object. The hash contents
81are not used by IO::AIO so you are free to store anything you like in it.
82
83During their existance, aio requests travel through the following states,
84in order:
85
86=over 4
87
88=item ready
89
90Immediately after a request is created it is put into the ready state,
91waiting for a thread to execute it.
92
93=item execute
94
95A thread has accepted the request for processing and is currently
96executing it (e.g. blocking in read).
97
98=item pending
99
100The request has been executed and is waiting for result processing.
101
102While request submission and execution is fully asynchronous, result
103processing is not and relies on the perl interpreter calling C<poll_cb>
104(or another function with the same effect).
105
106=item result
107
108The request results are processed synchronously by C<poll_cb>.
109
110The C<poll_cb> function will process all outstanding aio requests by
111calling their callbacks, freeing memory associated with them and managing
112any groups they are contained in.
113
114=item done
115
116Request has reached the end of its lifetime and holds no resources anymore
117(except possibly for the Perl object, but its connection to the actual
118aio request is severed and calling its methods will either do nothing or
119result in a runtime error).
120
121=cut
122
123package IO::AIO;
124
125no warnings;
126use strict 'vars';
127
128use base 'Exporter';
129
130BEGIN {
131 our $VERSION = '2.0';
132
133 our @AIO_REQ = qw(aio_sendfile aio_read aio_write aio_open aio_close aio_stat
134 aio_lstat aio_unlink aio_rmdir aio_readdir aio_scandir aio_symlink
135 aio_fsync aio_fdatasync aio_readahead aio_rename aio_link aio_move
136 aio_copy aio_group aio_nop aio_mknod);
137 our @EXPORT = (@AIO_REQ, qw(aioreq_pri aioreq_nice));
138 our @EXPORT_OK = qw(poll_fileno poll_cb poll_wait flush
139 min_parallel max_parallel nreqs nready npending);
140
141 @IO::AIO::GRP::ISA = 'IO::AIO::REQ';
142
143 require XSLoader;
144 XSLoader::load ("IO::AIO", $VERSION);
145}
146
147=head1 FUNCTIONS
148
149=head2 AIO FUNCTIONS
24 150
25All the C<aio_*> calls are more or less thin wrappers around the syscall 151All the C<aio_*> calls are more or less thin wrappers around the syscall
26with the same name (sans C<aio_>). The arguments are similar or identical, 152with the same name (sans C<aio_>). The arguments are similar or identical,
27and they all accept an additional C<$callback> argument which must be 153and they all accept an additional (and optional) C<$callback> argument
28a code reference. This code reference will get called with the syscall 154which must be a code reference. This code reference will get called with
29return code (e.g. most syscalls return C<-1> on error, unlike perl, which 155the syscall return code (e.g. most syscalls return C<-1> on error, unlike
30usually delivers "false") as it's sole argument when the given syscall has 156perl, which usually delivers "false") as it's sole argument when the given
31been executed asynchronously. 157syscall has been executed asynchronously.
32 158
33All functions that expect a filehandle will also accept a file descriptor. 159All functions expecting a filehandle keep a copy of the filehandle
160internally until the request has finished.
34 161
162All requests return objects of type L<IO::AIO::REQ> that allow further
163manipulation of those requests while they are in-flight.
164
35The filenames you pass to these routines I<must> be absolute. The reason 165The pathnames you pass to these routines I<must> be absolute and
36is that at the time the request is being executed, the current working 166encoded in byte form. The reason for the former is that at the time the
37directory could have changed. Alternatively, you can make sure that you 167request is being executed, the current working directory could have
168changed. Alternatively, you can make sure that you never change the
38never change the current working directory. 169current working directory.
170
171To encode pathnames to byte form, either make sure you either: a)
172always pass in filenames you got from outside (command line, readdir
173etc.), b) are ASCII or ISO 8859-1, c) use the Encode module and encode
174your pathnames to the locale (or other) encoding in effect in the user
175environment, d) use Glib::filename_from_unicode on unicode filenames or e)
176use something else.
39 177
40=over 4 178=over 4
41 179
42=cut 180=item $prev_pri = aioreq_pri [$pri]
43 181
44package IO::AIO; 182Returns the priority value that would be used for the next request and, if
183C<$pri> is given, sets the priority for the next aio request.
45 184
46use base 'Exporter'; 185The default priority is C<0>, the minimum and maximum priorities are C<-4>
186and C<4>, respectively. Requests with higher priority will be serviced
187first.
47 188
48BEGIN { 189The priority will be reset to C<0> after each call to one of the C<aio_*>
49 $VERSION = 0.1; 190functions.
50 191
51 @EXPORT = qw(aio_read aio_write aio_open aio_close aio_stat aio_lstat aio_unlink 192Example: open a file with low priority, then read something from it with
52 aio_fsync aio_fdatasync aio_readahead); 193higher priority so the read request is serviced before other low priority
53 @EXPORT_OK = qw(poll_fileno poll_cb min_parallel max_parallel nreqs); 194open requests (potentially spamming the cache):
54 195
55 require XSLoader; 196 aioreq_pri -3;
56 XSLoader::load IO::AIO, $VERSION; 197 aio_open ..., sub {
57} 198 return unless $_[0];
58 199
59=item IO::AIO::min_parallel $nthreads 200 aioreq_pri -2;
201 aio_read $_[0], ..., sub {
202 ...
203 };
204 };
60 205
61Set the minimum number of AIO threads to C<$nthreads>. The default is 206=item aioreq_nice $pri_adjust
62C<1>, which means a single asynchronous operation can be done at one time
63(the number of outstanding operations, however, is unlimited).
64 207
65It is recommended to keep the number of threads low, as some linux 208Similar to C<aioreq_pri>, but subtracts the given value from the current
66kernel versions will scale negatively with the number of threads (higher 209priority, so effects are cumulative.
67parallelity => MUCH higher latency).
68 210
69Under normal circumstances you don't need to call this function, as this
70module automatically starts a single async thread.
71
72=item IO::AIO::max_parallel $nthreads
73
74Sets the maximum number of AIO threads to C<$nthreads>. If more than
75the specified number of threads are currently running, kill them. This
76function blocks until the limit is reached.
77
78This module automatically runs C<max_parallel 0> at program end, to ensure
79that all threads are killed and that there are no outstanding requests.
80
81Under normal circumstances you don't need to call this function.
82
83=item $fileno = IO::AIO::poll_fileno
84
85Return the I<request result pipe filehandle>. This filehandle must be
86polled for reading by some mechanism outside this module (e.g. Event
87or select, see below). If the pipe becomes readable you have to call
88C<poll_cb> to check the results.
89
90See C<poll_cb> for an example.
91
92=item IO::AIO::poll_cb
93
94Process all outstanding events on the result pipe. You have to call this
95regularly. Returns the number of events processed. Returns immediately
96when no events are outstanding.
97
98You can use Event to multiplex, e.g.:
99
100 Event->io (fd => IO::AIO::poll_fileno,
101 poll => 'r', async => 1,
102 cb => \&IO::AIO::poll_cb);
103
104=item IO::AIO::poll_wait
105
106Wait till the result filehandle becomes ready for reading (simply does a
107select on the filehandle. This is useful if you want to synchronously wait
108for some requests to finish).
109
110See C<nreqs> for an example.
111
112=item IO::AIO::nreqs
113
114Returns the number of requests currently outstanding.
115
116Example: wait till there are no outstanding requests anymore:
117
118 IO::AIO::poll_wait, IO::AIO::poll_cb
119 while IO::AIO::nreqs;
120
121=item aio_open $pathname, $flags, $mode, $callback 211=item aio_open $pathname, $flags, $mode, $callback->($fh)
122 212
123Asynchronously open or create a file and call the callback with the 213Asynchronously open or create a file and call the callback with a newly
124filedescriptor (NOT a perl filehandle, sorry for that, but watch out, this 214created filehandle for the file.
125might change in the future).
126 215
127The pathname passed to C<aio_open> must be absolute. See API NOTES, above, 216The pathname passed to C<aio_open> must be absolute. See API NOTES, above,
128for an explanation. 217for an explanation.
129 218
130The C<$mode> argument is a bitmask. See the C<Fcntl> module for a 219The C<$flags> argument is a bitmask. See the C<Fcntl> module for a
131list. They are the same as used in C<sysopen>. 220list. They are the same as used by C<sysopen>.
221
222Likewise, C<$mode> specifies the mode of the newly created file, if it
223didn't exist and C<O_CREAT> has been given, just like perl's C<sysopen>,
224except that it is mandatory (i.e. use C<0> if you don't create new files,
225and C<0666> or C<0777> if you do).
132 226
133Example: 227Example:
134 228
135 aio_open "/etc/passwd", O_RDONLY, 0, sub { 229 aio_open "/etc/passwd", O_RDONLY, 0, sub {
136 if ($_[0] >= 0) { 230 if ($_[0]) {
137 open my $fh, "<&$_[0]"; # create a copy for perl
138 aio_close $_[0], sub { }; # close the aio handle
139 print "open successful, fh is $fh\n"; 231 print "open successful, fh is $_[0]\n";
140 ... 232 ...
141 } else { 233 } else {
142 die "open failed: $!\n"; 234 die "open failed: $!\n";
143 } 235 }
144 }; 236 };
145 237
146=item aio_close $fh, $callback 238=item aio_close $fh, $callback->($status)
147 239
148Asynchronously close a file and call the callback with the result code. 240Asynchronously close a file and call the callback with the result
241code. I<WARNING:> although accepted, you should not pass in a perl
242filehandle here, as perl will likely close the file descriptor another
243time when the filehandle is destroyed. Normally, you can safely call perls
244C<close> or just let filehandles go out of scope.
149 245
246This is supposed to be a bug in the API, so that might change. It's
247therefore best to avoid this function.
248
150=item aio_read $fh,$offset,$length, $data,$dataoffset,$callback 249=item aio_read $fh,$offset,$length, $data,$dataoffset, $callback->($retval)
151 250
152=item aio_write $fh,$offset,$length, $data,$dataoffset,$callback 251=item aio_write $fh,$offset,$length, $data,$dataoffset, $callback->($retval)
153 252
154Reads or writes C<length> bytes from the specified C<fh> and C<offset> 253Reads or writes C<length> bytes from the specified C<fh> and C<offset>
155into the scalar given by C<data> and offset C<dataoffset> and calls the 254into the scalar given by C<data> and offset C<dataoffset> and calls the
156callback without the actual number of bytes read (or -1 on error, just 255callback without the actual number of bytes read (or -1 on error, just
157like the syscall). 256like the syscall).
158 257
258The C<$data> scalar I<MUST NOT> be modified in any way while the request
259is outstanding. Modifying it can result in segfaults or WW3 (if the
260necessary/optional hardware is installed).
261
159Example: Read 15 bytes at offset 7 into scalar C<$buffer>, strating at 262Example: Read 15 bytes at offset 7 into scalar C<$buffer>, starting at
160offset C<0> within the scalar: 263offset C<0> within the scalar:
161 264
162 aio_read $fh, 7, 15, $buffer, 0, sub { 265 aio_read $fh, 7, 15, $buffer, 0, sub {
163 $_[0] >= 0 or die "read error: $!"; 266 $_[0] > 0 or die "read error: $!";
164 print "read <$buffer>\n"; 267 print "read $_[0] bytes: <$buffer>\n";
165 }; 268 };
166 269
270=item aio_sendfile $out_fh, $in_fh, $in_offset, $length, $callback->($retval)
271
272Tries to copy C<$length> bytes from C<$in_fh> to C<$out_fh>. It starts
273reading at byte offset C<$in_offset>, and starts writing at the current
274file offset of C<$out_fh>. Because of that, it is not safe to issue more
275than one C<aio_sendfile> per C<$out_fh>, as they will interfere with each
276other.
277
278This call tries to make use of a native C<sendfile> syscall to provide
279zero-copy operation. For this to work, C<$out_fh> should refer to a
280socket, and C<$in_fh> should refer to mmap'able file.
281
282If the native sendfile call fails or is not implemented, it will be
283emulated, so you can call C<aio_sendfile> on any type of filehandle
284regardless of the limitations of the operating system.
285
286Please note, however, that C<aio_sendfile> can read more bytes from
287C<$in_fh> than are written, and there is no way to find out how many
288bytes have been read from C<aio_sendfile> alone, as C<aio_sendfile> only
289provides the number of bytes written to C<$out_fh>. Only if the result
290value equals C<$length> one can assume that C<$length> bytes have been
291read.
292
167=item aio_readahead $fh,$offset,$length, $callback 293=item aio_readahead $fh,$offset,$length, $callback->($retval)
168 294
169Asynchronously reads the specified byte range into the page cache, using
170the C<readahead> syscall. If that syscall doesn't exist the status will be
171C<-1> and C<$!> is set to ENOSYS.
172
173readahead() populates the page cache with data from a file so that 295C<aio_readahead> populates the page cache with data from a file so that
174subsequent reads from that file will not block on disk I/O. The C<$offset> 296subsequent reads from that file will not block on disk I/O. The C<$offset>
175argument specifies the starting point from which data is to be read and 297argument specifies the starting point from which data is to be read and
176C<$length> specifies the number of bytes to be read. I/O is performed in 298C<$length> specifies the number of bytes to be read. I/O is performed in
177whole pages, so that offset is effectively rounded down to a page boundary 299whole pages, so that offset is effectively rounded down to a page boundary
178and bytes are read up to the next page boundary greater than or equal to 300and bytes are read up to the next page boundary greater than or equal to
179(off-set+length). aio_readahead() does not read beyond the end of the 301(off-set+length). C<aio_readahead> does not read beyond the end of the
180file. The current file offset of the file is left unchanged. 302file. The current file offset of the file is left unchanged.
181 303
304If that syscall doesn't exist (likely if your OS isn't Linux) it will be
305emulated by simply reading the data, which would have a similar effect.
306
182=item aio_stat $fh_or_path, $callback 307=item aio_stat $fh_or_path, $callback->($status)
183 308
184=item aio_lstat $fh, $callback 309=item aio_lstat $fh, $callback->($status)
185 310
186Works like perl's C<stat> or C<lstat> in void context. The callback will 311Works like perl's C<stat> or C<lstat> in void context. The callback will
187be called after the stat and the results will be available using C<stat _> 312be called after the stat and the results will be available using C<stat _>
188or C<-s _> etc... 313or C<-s _> etc...
189 314
199 aio_stat "/etc/passwd", sub { 324 aio_stat "/etc/passwd", sub {
200 $_[0] and die "stat failed: $!"; 325 $_[0] and die "stat failed: $!";
201 print "size is ", -s _, "\n"; 326 print "size is ", -s _, "\n";
202 }; 327 };
203 328
204=item aio_unlink $pathname, $callback 329=item aio_unlink $pathname, $callback->($status)
205 330
206Asynchronously unlink (delete) a file and call the callback with the 331Asynchronously unlink (delete) a file and call the callback with the
207result code. 332result code.
208 333
334=item aio_mknod $path, $mode, $dev, $callback->($status)
335
336Asynchronously create a device node (or fifo). See mknod(2): the only
337portable value for C<$mode> is C<S_IFIFO> ored with permissions, and C<0>
338for C<$dev>.
339
340=item aio_link $srcpath, $dstpath, $callback->($status)
341
342Asynchronously create a new link to the existing object at C<$srcpath> at
343the path C<$dstpath> and call the callback with the result code.
344
345=item aio_symlink $srcpath, $dstpath, $callback->($status)
346
347Asynchronously create a new symbolic link to the existing object at C<$srcpath> at
348the path C<$dstpath> and call the callback with the result code.
349
350=item aio_rename $srcpath, $dstpath, $callback->($status)
351
352Asynchronously rename the object at C<$srcpath> to C<$dstpath>, just as
353rename(2) and call the callback with the result code.
354
355=item aio_rmdir $pathname, $callback->($status)
356
357Asynchronously rmdir (delete) a directory and call the callback with the
358result code.
359
360=item aio_readdir $pathname, $callback->($entries)
361
362Unlike the POSIX call of the same name, C<aio_readdir> reads an entire
363directory (i.e. opendir + readdir + closedir). The entries will not be
364sorted, and will B<NOT> include the C<.> and C<..> entries.
365
366The callback a single argument which is either C<undef> or an array-ref
367with the filenames.
368
369=item aio_copy $srcpath, $dstpath, $callback->($status)
370
371Try to copy the I<file> (directories not supported as either source or
372destination) from C<$srcpath> to C<$dstpath> and call the callback with
373the C<0> (error) or C<-1> ok.
374
375This is a composite request that it creates the destination file with
376mode 0200 and copies the contents of the source file into it using
377C<aio_sendfile>, followed by restoring atime, mtime, access mode and
378uid/gid, in that order.
379
380If an error occurs, the partial destination file will be unlinked, if
381possible, except when setting atime, mtime, access mode and uid/gid, where
382errors are being ignored.
383
384=cut
385
386sub aio_copy($$;$) {
387 my ($src, $dst, $cb) = @_;
388
389 my $pri = aioreq_pri;
390 my $grp = aio_group $cb;
391
392 aioreq_pri $pri;
393 add $grp aio_open $src, O_RDONLY, 0, sub {
394 if (my $src_fh = $_[0]) {
395 my @stat = stat $src_fh;
396
397 aioreq_pri $pri;
398 add $grp aio_open $dst, O_CREAT | O_WRONLY | O_TRUNC, 0200, sub {
399 if (my $dst_fh = $_[0]) {
400 aioreq_pri $pri;
401 add $grp aio_sendfile $dst_fh, $src_fh, 0, $stat[7], sub {
402 if ($_[0] == $stat[7]) {
403 $grp->result (0);
404 close $src_fh;
405
406 # those should not normally block. should. should.
407 utime $stat[8], $stat[9], $dst;
408 chmod $stat[2] & 07777, $dst_fh;
409 chown $stat[4], $stat[5], $dst_fh;
410 close $dst_fh;
411 } else {
412 $grp->result (-1);
413 close $src_fh;
414 close $dst_fh;
415
416 aioreq $pri;
417 add $grp aio_unlink $dst;
418 }
419 };
420 } else {
421 $grp->result (-1);
422 }
423 },
424
425 } else {
426 $grp->result (-1);
427 }
428 };
429
430 $grp
431}
432
433=item aio_move $srcpath, $dstpath, $callback->($status)
434
435Try to move the I<file> (directories not supported as either source or
436destination) from C<$srcpath> to C<$dstpath> and call the callback with
437the C<0> (error) or C<-1> ok.
438
439This is a composite request that tries to rename(2) the file first. If
440rename files with C<EXDEV>, it copies the file with C<aio_copy> and, if
441that is successful, unlinking the C<$srcpath>.
442
443=cut
444
445sub aio_move($$;$) {
446 my ($src, $dst, $cb) = @_;
447
448 my $pri = aioreq_pri;
449 my $grp = aio_group $cb;
450
451 aioreq_pri $pri;
452 add $grp aio_rename $src, $dst, sub {
453 if ($_[0] && $! == EXDEV) {
454 aioreq_pri $pri;
455 add $grp aio_copy $src, $dst, sub {
456 $grp->result ($_[0]);
457
458 if (!$_[0]) {
459 aioreq_pri $pri;
460 add $grp aio_unlink $src;
461 }
462 };
463 } else {
464 $grp->result ($_[0]);
465 }
466 };
467
468 $grp
469}
470
471=item aio_scandir $path, $maxreq, $callback->($dirs, $nondirs)
472
473Scans a directory (similar to C<aio_readdir>) but additionally tries to
474efficiently separate the entries of directory C<$path> into two sets of
475names, directories you can recurse into (directories), and ones you cannot
476recurse into (everything else, including symlinks to directories).
477
478C<aio_scandir> is a composite request that creates of many sub requests_
479C<$maxreq> specifies the maximum number of outstanding aio requests that
480this function generates. If it is C<< <= 0 >>, then a suitable default
481will be chosen (currently 4).
482
483On error, the callback is called without arguments, otherwise it receives
484two array-refs with path-relative entry names.
485
486Example:
487
488 aio_scandir $dir, 0, sub {
489 my ($dirs, $nondirs) = @_;
490 print "real directories: @$dirs\n";
491 print "everything else: @$nondirs\n";
492 };
493
494Implementation notes.
495
496The C<aio_readdir> cannot be avoided, but C<stat()>'ing every entry can.
497
498After reading the directory, the modification time, size etc. of the
499directory before and after the readdir is checked, and if they match (and
500isn't the current time), the link count will be used to decide how many
501entries are directories (if >= 2). Otherwise, no knowledge of the number
502of subdirectories will be assumed.
503
504Then entries will be sorted into likely directories (everything without
505a non-initial dot currently) and likely non-directories (everything
506else). Then every entry plus an appended C</.> will be C<stat>'ed,
507likely directories first. If that succeeds, it assumes that the entry
508is a directory or a symlink to directory (which will be checked
509seperately). This is often faster than stat'ing the entry itself because
510filesystems might detect the type of the entry without reading the inode
511data (e.g. ext2fs filetype feature).
512
513If the known number of directories (link count - 2) has been reached, the
514rest of the entries is assumed to be non-directories.
515
516This only works with certainty on POSIX (= UNIX) filesystems, which
517fortunately are the vast majority of filesystems around.
518
519It will also likely work on non-POSIX filesystems with reduced efficiency
520as those tend to return 0 or 1 as link counts, which disables the
521directory counting heuristic.
522
523=cut
524
525sub aio_scandir($$$) {
526 my ($path, $maxreq, $cb) = @_;
527
528 my $pri = aioreq_pri;
529
530 my $grp = aio_group $cb;
531
532 $maxreq = 4 if $maxreq <= 0;
533
534 # stat once
535 aioreq_pri $pri;
536 add $grp aio_stat $path, sub {
537 return $grp->result () if $_[0];
538 my $now = time;
539 my $hash1 = join ":", (stat _)[0,1,3,7,9];
540
541 # read the directory entries
542 aioreq_pri $pri;
543 add $grp aio_readdir $path, sub {
544 my $entries = shift
545 or return $grp->result ();
546
547 # stat the dir another time
548 aioreq_pri $pri;
549 add $grp aio_stat $path, sub {
550 my $hash2 = join ":", (stat _)[0,1,3,7,9];
551
552 my $ndirs;
553
554 # take the slow route if anything looks fishy
555 if ($hash1 ne $hash2 or (stat _)[9] == $now) {
556 $ndirs = -1;
557 } else {
558 # if nlink == 2, we are finished
559 # on non-posix-fs's, we rely on nlink < 2
560 $ndirs = (stat _)[3] - 2
561 or return $grp->result ([], $entries);
562 }
563
564 # sort into likely dirs and likely nondirs
565 # dirs == files without ".", short entries first
566 $entries = [map $_->[0],
567 sort { $b->[1] cmp $a->[1] }
568 map [$_, sprintf "%s%04d", (/.\./ ? "1" : "0"), length],
569 @$entries];
570
571 my (@dirs, @nondirs);
572
573 my $statgrp = add $grp aio_group sub {
574 $grp->result (\@dirs, \@nondirs);
575 };
576
577 limit $statgrp $maxreq;
578 feed $statgrp sub {
579 return unless @$entries;
580 my $entry = pop @$entries;
581
582 aioreq_pri $pri;
583 add $statgrp aio_stat "$path/$entry/.", sub {
584 if ($_[0] < 0) {
585 push @nondirs, $entry;
586 } else {
587 # need to check for real directory
588 aioreq_pri $pri;
589 add $statgrp aio_lstat "$path/$entry", sub {
590 if (-d _) {
591 push @dirs, $entry;
592
593 unless (--$ndirs) {
594 push @nondirs, @$entries;
595 feed $statgrp;
596 }
597 } else {
598 push @nondirs, $entry;
599 }
600 }
601 }
602 };
603 };
604 };
605 };
606 };
607
608 $grp
609}
610
209=item aio_fsync $fh, $callback 611=item aio_fsync $fh, $callback->($status)
210 612
211Asynchronously call fsync on the given filehandle and call the callback 613Asynchronously call fsync on the given filehandle and call the callback
212with the fsync result code. 614with the fsync result code.
213 615
214=item aio_fdatasync $fh, $callback 616=item aio_fdatasync $fh, $callback->($status)
215 617
216Asynchronously call fdatasync on the given filehandle and call the 618Asynchronously call fdatasync on the given filehandle and call the
217callback with the fdatasync result code. 619callback with the fdatasync result code.
218 620
621If this call isn't available because your OS lacks it or it couldn't be
622detected, it will be emulated by calling C<fsync> instead.
623
624=item aio_group $callback->(...)
625
626This is a very special aio request: Instead of doing something, it is a
627container for other aio requests, which is useful if you want to bundle
628many requests into a single, composite, request with a definite callback
629and the ability to cancel the whole request with its subrequests.
630
631Returns an object of class L<IO::AIO::GRP>. See its documentation below
632for more info.
633
634Example:
635
636 my $grp = aio_group sub {
637 print "all stats done\n";
638 };
639
640 add $grp
641 (aio_stat ...),
642 (aio_stat ...),
643 ...;
644
645=item aio_nop $callback->()
646
647This is a special request - it does nothing in itself and is only used for
648side effects, such as when you want to add a dummy request to a group so
649that finishing the requests in the group depends on executing the given
650code.
651
652While this request does nothing, it still goes through the execution
653phase and still requires a worker thread. Thus, the callback will not
654be executed immediately but only after other requests in the queue have
655entered their execution phase. This can be used to measure request
656latency.
657
658=item IO::AIO::aio_busy $fractional_seconds, $callback->() *NOT EXPORTED*
659
660Mainly used for debugging and benchmarking, this aio request puts one of
661the request workers to sleep for the given time.
662
663While it is theoretically handy to have simple I/O scheduling requests
664like sleep and file handle readable/writable, the overhead this creates is
665immense (it blocks a thread for a long time) so do not use this function
666except to put your application under artificial I/O pressure.
667
668=back
669
670=head2 IO::AIO::REQ CLASS
671
672All non-aggregate C<aio_*> functions return an object of this class when
673called in non-void context.
674
675=over 4
676
677=item cancel $req
678
679Cancels the request, if possible. Has the effect of skipping execution
680when entering the B<execute> state and skipping calling the callback when
681entering the the B<result> state, but will leave the request otherwise
682untouched. That means that requests that currently execute will not be
683stopped and resources held by the request will not be freed prematurely.
684
685=item cb $req $callback->(...)
686
687Replace (or simply set) the callback registered to the request.
688
689=back
690
691=head2 IO::AIO::GRP CLASS
692
693This class is a subclass of L<IO::AIO::REQ>, so all its methods apply to
694objects of this class, too.
695
696A IO::AIO::GRP object is a special request that can contain multiple other
697aio requests.
698
699You create one by calling the C<aio_group> constructing function with a
700callback that will be called when all contained requests have entered the
701C<done> state:
702
703 my $grp = aio_group sub {
704 print "all requests are done\n";
705 };
706
707You add requests by calling the C<add> method with one or more
708C<IO::AIO::REQ> objects:
709
710 $grp->add (aio_unlink "...");
711
712 add $grp aio_stat "...", sub {
713 $_[0] or return $grp->result ("error");
714
715 # add another request dynamically, if first succeeded
716 add $grp aio_open "...", sub {
717 $grp->result ("ok");
718 };
719 };
720
721This makes it very easy to create composite requests (see the source of
722C<aio_move> for an application) that work and feel like simple requests.
723
724=over 4
725
726=item * The IO::AIO::GRP objects will be cleaned up during calls to
727C<IO::AIO::poll_cb>, just like any other request.
728
729=item * They can be canceled like any other request. Canceling will cancel not
730only the request itself, but also all requests it contains.
731
732=item * They can also can also be added to other IO::AIO::GRP objects.
733
734=item * You must not add requests to a group from within the group callback (or
735any later time).
736
737=back
738
739Their lifetime, simplified, looks like this: when they are empty, they
740will finish very quickly. If they contain only requests that are in the
741C<done> state, they will also finish. Otherwise they will continue to
742exist.
743
744That means after creating a group you have some time to add requests. And
745in the callbacks of those requests, you can add further requests to the
746group. And only when all those requests have finished will the the group
747itself finish.
748
749=over 4
750
751=item add $grp ...
752
753=item $grp->add (...)
754
755Add one or more requests to the group. Any type of L<IO::AIO::REQ> can
756be added, including other groups, as long as you do not create circular
757dependencies.
758
759Returns all its arguments.
760
761=item $grp->cancel_subs
762
763Cancel all subrequests and clears any feeder, but not the group request
764itself. Useful when you queued a lot of events but got a result early.
765
766=item $grp->result (...)
767
768Set the result value(s) that will be passed to the group callback when all
769subrequests have finished and set thre groups errno to the current value
770of errno (just like calling C<errno> without an error number). By default,
771no argument will be passed and errno is zero.
772
773=item $grp->errno ([$errno])
774
775Sets the group errno value to C<$errno>, or the current value of errno
776when the argument is missing.
777
778Every aio request has an associated errno value that is restored when
779the callback is invoked. This method lets you change this value from its
780default (0).
781
782Calling C<result> will also set errno, so make sure you either set C<$!>
783before the call to C<result>, or call c<errno> after it.
784
785=item feed $grp $callback->($grp)
786
787Sets a feeder/generator on this group: every group can have an attached
788generator that generates requests if idle. The idea behind this is that,
789although you could just queue as many requests as you want in a group,
790this might starve other requests for a potentially long time. For
791example, C<aio_scandir> might generate hundreds of thousands C<aio_stat>
792requests, delaying any later requests for a long time.
793
794To avoid this, and allow incremental generation of requests, you can
795instead a group and set a feeder on it that generates those requests. The
796feed callback will be called whenever there are few enough (see C<limit>,
797below) requests active in the group itself and is expected to queue more
798requests.
799
800The feed callback can queue as many requests as it likes (i.e. C<add> does
801not impose any limits).
802
803If the feed does not queue more requests when called, it will be
804automatically removed from the group.
805
806If the feed limit is C<0>, it will be set to C<2> automatically.
807
808Example:
809
810 # stat all files in @files, but only ever use four aio requests concurrently:
811
812 my $grp = aio_group sub { print "finished\n" };
813 limit $grp 4;
814 feed $grp sub {
815 my $file = pop @files
816 or return;
817
818 add $grp aio_stat $file, sub { ... };
819 };
820
821=item limit $grp $num
822
823Sets the feeder limit for the group: The feeder will be called whenever
824the group contains less than this many requests.
825
826Setting the limit to C<0> will pause the feeding process.
827
828=back
829
830=head2 SUPPORT FUNCTIONS
831
832=over 4
833
834=item $fileno = IO::AIO::poll_fileno
835
836Return the I<request result pipe file descriptor>. This filehandle must be
837polled for reading by some mechanism outside this module (e.g. Event or
838select, see below or the SYNOPSIS). If the pipe becomes readable you have
839to call C<poll_cb> to check the results.
840
841See C<poll_cb> for an example.
842
843=item IO::AIO::poll_cb
844
845Process all outstanding events on the result pipe. You have to call this
846regularly. Returns the number of events processed. Returns immediately
847when no events are outstanding.
848
849If not all requests were processed for whatever reason, the filehandle
850will still be ready when C<poll_cb> returns.
851
852Example: Install an Event watcher that automatically calls
853IO::AIO::poll_cb with high priority:
854
855 Event->io (fd => IO::AIO::poll_fileno,
856 poll => 'r', async => 1,
857 cb => \&IO::AIO::poll_cb);
858
859=item IO::AIO::poll_some $max_requests
860
861Similar to C<poll_cb>, but only processes up to C<$max_requests> requests
862at a time.
863
864Useful if you want to ensure some level of interactiveness when perl is
865not fast enough to process all requests in time.
866
867Example: Install an Event watcher that automatically calls
868IO::AIO::poll_some with low priority, to ensure that other parts of the
869program get the CPU sometimes even under high AIO load.
870
871 Event->io (fd => IO::AIO::poll_fileno,
872 poll => 'r', nice => 1,
873 cb => sub { IO::AIO::poll_some 256 });
874
875=item IO::AIO::poll_wait
876
877Wait till the result filehandle becomes ready for reading (simply does a
878C<select> on the filehandle. This is useful if you want to synchronously wait
879for some requests to finish).
880
881See C<nreqs> for an example.
882
883=item IO::AIO::nreqs
884
885Returns the number of requests currently in the ready, execute or pending
886states (i.e. for which their callback has not been invoked yet).
887
888Example: wait till there are no outstanding requests anymore:
889
890 IO::AIO::poll_wait, IO::AIO::poll_cb
891 while IO::AIO::nreqs;
892
893=item IO::AIO::nready
894
895Returns the number of requests currently in the ready state (not yet
896executed).
897
898=item IO::AIO::npending
899
900Returns the number of requests currently in the pending state (executed,
901but not yet processed by poll_cb).
902
903=item IO::AIO::flush
904
905Wait till all outstanding AIO requests have been handled.
906
907Strictly equivalent to:
908
909 IO::AIO::poll_wait, IO::AIO::poll_cb
910 while IO::AIO::nreqs;
911
912=item IO::AIO::poll
913
914Waits until some requests have been handled.
915
916Strictly equivalent to:
917
918 IO::AIO::poll_wait, IO::AIO::poll_cb
919 if IO::AIO::nreqs;
920
921=item IO::AIO::min_parallel $nthreads
922
923Set the minimum number of AIO threads to C<$nthreads>. The current
924default is C<8>, which means eight asynchronous operations can execute
925concurrently at any one time (the number of outstanding requests,
926however, is unlimited).
927
928IO::AIO starts threads only on demand, when an AIO request is queued and
929no free thread exists.
930
931It is recommended to keep the number of threads relatively low, as some
932Linux kernel versions will scale negatively with the number of threads
933(higher parallelity => MUCH higher latency). With current Linux 2.6
934versions, 4-32 threads should be fine.
935
936Under most circumstances you don't need to call this function, as the
937module selects a default that is suitable for low to moderate load.
938
939=item IO::AIO::max_parallel $nthreads
940
941Sets the maximum number of AIO threads to C<$nthreads>. If more than the
942specified number of threads are currently running, this function kills
943them. This function blocks until the limit is reached.
944
945While C<$nthreads> are zero, aio requests get queued but not executed
946until the number of threads has been increased again.
947
948This module automatically runs C<max_parallel 0> at program end, to ensure
949that all threads are killed and that there are no outstanding requests.
950
951Under normal circumstances you don't need to call this function.
952
953=item $oldmaxreqs = IO::AIO::max_outstanding $maxreqs
954
955This is a very bad function to use in interactive programs because it
956blocks, and a bad way to reduce concurrency because it is inexact: Better
957use an C<aio_group> together with a feed callback.
958
959Sets the maximum number of outstanding requests to C<$nreqs>. If you
960to queue up more than this number of requests, the next call to the
961C<poll_cb> (and C<poll_some> and other functions calling C<poll_cb>)
962function will block until the limit is no longer exceeded.
963
964The default value is very large, so there is no practical limit on the
965number of outstanding requests.
966
967You can still queue as many requests as you want. Therefore,
968C<max_oustsanding> is mainly useful in simple scripts (with low values) or
969as a stop gap to shield against fatal memory overflow (with large values).
970
971=back
972
219=cut 973=cut
220 974
975# support function to convert a fd into a perl filehandle
976sub _fd2fh {
977 return undef if $_[0] < 0;
978
979 # try to generate nice filehandles
980 my $sym = "IO::AIO::fd#$_[0]";
981 local *$sym;
982
983 open *$sym, "+<&=$_[0]" # usually works under any unix
984 or open *$sym, "<&=$_[0]" # cygwin needs this
985 or open *$sym, ">&=$_[0]" # or this
986 or return undef;
987
988 *$sym
989}
990
221min_parallel 4; 991min_parallel 8;
222 992
223END { 993END {
224 max_parallel 0; 994 flush;
225} 995};
226 996
2271; 9971;
228 998
229=back 999=head2 FORK BEHAVIOUR
230 1000
1001This module should do "the right thing" when the process using it forks:
1002
1003Before the fork, IO::AIO enters a quiescent state where no requests
1004can be added in other threads and no results will be processed. After
1005the fork the parent simply leaves the quiescent state and continues
1006request/result processing, while the child frees the request/result queue
1007(so that the requests started before the fork will only be handled in the
1008parent). Threads will be started on demand until the limit set in the
1009parent process has been reached again.
1010
1011In short: the parent will, after a short pause, continue as if fork had
1012not been called, while the child will act as if IO::AIO has not been used
1013yet.
1014
1015=head2 MEMORY USAGE
1016
1017Per-request usage:
1018
1019Each aio request uses - depending on your architecture - around 100-200
1020bytes of memory. In addition, stat requests need a stat buffer (possibly
1021a few hundred bytes), readdir requires a result buffer and so on. Perl
1022scalars and other data passed into aio requests will also be locked and
1023will consume memory till the request has entered the done state.
1024
1025This is now awfully much, so queuing lots of requests is not usually a
1026problem.
1027
1028Per-thread usage:
1029
1030In the execution phase, some aio requests require more memory for
1031temporary buffers, and each thread requires a stack and other data
1032structures (usually around 16k-128k, depending on the OS).
1033
231=head1 BUGS 1034=head1 KNOWN BUGS
232 1035
233 - aio_open gives a fd, but all other functions expect a perl filehandle. 1036Known bugs will be fixed in the next release.
234 1037
235=head1 SEE ALSO 1038=head1 SEE ALSO
236 1039
237L<Coro>, L<Linux::AIO>. 1040L<Coro::AIO>.
238 1041
239=head1 AUTHOR 1042=head1 AUTHOR
240 1043
241 Marc Lehmann <schmorp@schmorp.de> 1044 Marc Lehmann <schmorp@schmorp.de>
242 http://home.schmorp.de/ 1045 http://home.schmorp.de/

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines