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.85 by root, Sat Oct 28 01:40:30 2006 UTC vs.
Revision 1.117 by root, Sat Oct 6 14:05:19 2007 UTC

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 { 9 aio_open "/etc/passwd", O_RDONLY, 0, sub {
10 my ($fh) = @_; 10 my $fh = shift
11 or die "/etc/passwd: $!";
11 ... 12 ...
12 }; 13 };
13 14
14 aio_unlink "/tmp/file", sub { }; 15 aio_unlink "/tmp/file", sub { };
15 16
61etc.), but can also be used to easily do operations in parallel that are 62etc.), but can also be used to easily do operations in parallel that are
62normally done sequentially, e.g. stat'ing many files, which is much faster 63normally done sequentially, e.g. stat'ing many files, which is much faster
63on a RAID volume or over NFS when you do a number of stat operations 64on a RAID volume or over NFS when you do a number of stat operations
64concurrently. 65concurrently.
65 66
66While this works on all types of file descriptors (for example sockets), 67While most of this works on all types of file descriptors (for
67using these functions on file descriptors that support nonblocking 68example sockets), using these functions on file descriptors that
68operation (again, sockets, pipes etc.) is very inefficient. Use an event 69support nonblocking operation (again, sockets, pipes etc.) is very
69loop for that (such as the L<Event|Event> module): IO::AIO will naturally 70inefficient. Use an event loop for that (such as the L<Event|Event>
70fit into such an event loop itself. 71module): IO::AIO will naturally fit into such an event loop itself.
71 72
72In this version, a number of threads are started that execute your 73In this version, a number of threads are started that execute your
73requests and signal their completion. You don't need thread support 74requests and signal their completion. You don't need thread support
74in perl, and the threads created by this module will not be visible 75in perl, and the threads created by this module will not be visible
75to perl. In the future, this module might make use of the native aio 76to perl. In the future, this module might make use of the native aio
77not well-supported or restricted (GNU/Linux doesn't allow them on normal 78not well-supported or restricted (GNU/Linux doesn't allow them on normal
78files currently, for example), and they would only support aio_read and 79files currently, for example), and they would only support aio_read and
79aio_write, so the remaining functionality would have to be implemented 80aio_write, so the remaining functionality would have to be implemented
80using threads anyway. 81using threads anyway.
81 82
82Although the module will work with in the presence of other (Perl-) 83Although the module will work in the presence of other (Perl-) threads,
83threads, it is currently not reentrant in any way, so use appropriate 84it is currently not reentrant in any way, so use appropriate locking
84locking yourself, always call C<poll_cb> from within the same thread, or 85yourself, always call C<poll_cb> from within the same thread, or never
85never call C<poll_cb> (or other C<aio_> functions) recursively. 86call C<poll_cb> (or other C<aio_> functions) recursively.
87
88=head2 EXAMPLE
89
90This is a simple example that uses the Event module and loads
91F</etc/passwd> asynchronously:
92
93 use Fcntl;
94 use Event;
95 use IO::AIO;
96
97 # register the IO::AIO callback with Event
98 Event->io (fd => IO::AIO::poll_fileno,
99 poll => 'r',
100 cb => \&IO::AIO::poll_cb);
101
102 # queue the request to open /etc/passwd
103 aio_open "/etc/passwd", O_RDONLY, 0, sub {
104 my $fh = shift
105 or die "error while opening: $!";
106
107 # stat'ing filehandles is generally non-blocking
108 my $size = -s $fh;
109
110 # queue a request to read the file
111 my $contents;
112 aio_read $fh, 0, $size, $contents, 0, sub {
113 $_[0] == $size
114 or die "short read: $!";
115
116 close $fh;
117
118 # file contents now in $contents
119 print $contents;
120
121 # exit event loop and program
122 Event::unloop;
123 };
124 };
125
126 # possibly queue up other requests, or open GUI windows,
127 # check for sockets etc. etc.
128
129 # process events as long as there are some:
130 Event::loop;
86 131
87=head1 REQUEST ANATOMY AND LIFETIME 132=head1 REQUEST ANATOMY AND LIFETIME
88 133
89Every C<aio_*> function creates a request. which is a C data structure not 134Every C<aio_*> function creates a request. which is a C data structure not
90directly visible to Perl. 135directly visible to Perl.
132Request has reached the end of its lifetime and holds no resources anymore 177Request has reached the end of its lifetime and holds no resources anymore
133(except possibly for the Perl object, but its connection to the actual 178(except possibly for the Perl object, but its connection to the actual
134aio request is severed and calling its methods will either do nothing or 179aio request is severed and calling its methods will either do nothing or
135result in a runtime error). 180result in a runtime error).
136 181
182=back
183
137=cut 184=cut
138 185
139package IO::AIO; 186package IO::AIO;
187
188use Carp ();
140 189
141no warnings; 190no warnings;
142use strict 'vars'; 191use strict 'vars';
143 192
144use base 'Exporter'; 193use base 'Exporter';
145 194
146BEGIN { 195BEGIN {
147 our $VERSION = '2.0'; 196 our $VERSION = '2.51';
148 197
149 our @AIO_REQ = qw(aio_sendfile aio_read aio_write aio_open aio_close aio_stat 198 our @AIO_REQ = qw(aio_sendfile aio_read aio_write aio_open aio_close aio_stat
150 aio_lstat aio_unlink aio_rmdir aio_readdir aio_scandir aio_symlink 199 aio_lstat aio_unlink aio_rmdir aio_readdir aio_scandir aio_symlink
151 aio_fsync aio_fdatasync aio_readahead aio_rename aio_link aio_move 200 aio_readlink aio_fsync aio_fdatasync aio_readahead aio_rename aio_link
152 aio_copy aio_group aio_nop aio_mknod); 201 aio_move aio_copy aio_group aio_nop aio_mknod aio_load aio_rmtree aio_mkdir
202 aio_chown aio_chmod aio_utime aio_truncate);
153 our @EXPORT = (@AIO_REQ, qw(aioreq_pri aioreq_nice)); 203 our @EXPORT = (@AIO_REQ, qw(aioreq_pri aioreq_nice aio_block));
154 our @EXPORT_OK = qw(poll_fileno poll_cb poll_wait flush 204 our @EXPORT_OK = qw(poll_fileno poll_cb poll_wait flush
155 min_parallel max_parallel nreqs nready npending); 205 min_parallel max_parallel max_idle
206 nreqs nready npending nthreads
207 max_poll_time max_poll_reqs);
156 208
157 @IO::AIO::GRP::ISA = 'IO::AIO::REQ'; 209 @IO::AIO::GRP::ISA = 'IO::AIO::REQ';
158 210
159 require XSLoader; 211 require XSLoader;
160 XSLoader::load ("IO::AIO", $VERSION); 212 XSLoader::load ("IO::AIO", $VERSION);
161} 213}
162 214
163=head1 FUNCTIONS 215=head1 FUNCTIONS
164 216
165=head2 AIO FUNCTIONS 217=head2 AIO REQUEST FUNCTIONS
166 218
167All the C<aio_*> calls are more or less thin wrappers around the syscall 219All the C<aio_*> calls are more or less thin wrappers around the syscall
168with the same name (sans C<aio_>). The arguments are similar or identical, 220with the same name (sans C<aio_>). The arguments are similar or identical,
169and they all accept an additional (and optional) C<$callback> argument 221and they all accept an additional (and optional) C<$callback> argument
170which must be a code reference. This code reference will get called with 222which must be a code reference. This code reference will get called with
173syscall has been executed asynchronously. 225syscall has been executed asynchronously.
174 226
175All functions expecting a filehandle keep a copy of the filehandle 227All functions expecting a filehandle keep a copy of the filehandle
176internally until the request has finished. 228internally until the request has finished.
177 229
178All requests return objects of type L<IO::AIO::REQ> that allow further 230All functions return request objects of type L<IO::AIO::REQ> that allow
179manipulation of those requests while they are in-flight. 231further manipulation of those requests while they are in-flight.
180 232
181The pathnames you pass to these routines I<must> be absolute and 233The pathnames you pass to these routines I<must> be absolute and
182encoded in byte form. The reason for the former is that at the time the 234encoded as octets. The reason for the former is that at the time the
183request is being executed, the current working directory could have 235request is being executed, the current working directory could have
184changed. Alternatively, you can make sure that you never change the 236changed. Alternatively, you can make sure that you never change the
185current working directory. 237current working directory anywhere in the program and then use relative
238paths.
186 239
187To encode pathnames to byte form, either make sure you either: a) 240To encode pathnames as octets, either make sure you either: a) always pass
188always pass in filenames you got from outside (command line, readdir 241in filenames you got from outside (command line, readdir etc.) without
189etc.), b) are ASCII or ISO 8859-1, c) use the Encode module and encode 242tinkering, b) are ASCII or ISO 8859-1, c) use the Encode module and encode
190your pathnames to the locale (or other) encoding in effect in the user 243your pathnames to the locale (or other) encoding in effect in the user
191environment, d) use Glib::filename_from_unicode on unicode filenames or e) 244environment, d) use Glib::filename_from_unicode on unicode filenames or e)
192use something else. 245use something else to ensure your scalar has the correct contents.
246
247This works, btw. independent of the internal UTF-8 bit, which IO::AIO
248handles correctly wether it is set or not.
193 249
194=over 4 250=over 4
195 251
196=item $prev_pri = aioreq_pri [$pri] 252=item $prev_pri = aioreq_pri [$pri]
197 253
217 aio_read $_[0], ..., sub { 273 aio_read $_[0], ..., sub {
218 ... 274 ...
219 }; 275 };
220 }; 276 };
221 277
278
222=item aioreq_nice $pri_adjust 279=item aioreq_nice $pri_adjust
223 280
224Similar to C<aioreq_pri>, but subtracts the given value from the current 281Similar to C<aioreq_pri>, but subtracts the given value from the current
225priority, so effects are cumulative. 282priority, so the effect is cumulative.
283
226 284
227=item aio_open $pathname, $flags, $mode, $callback->($fh) 285=item aio_open $pathname, $flags, $mode, $callback->($fh)
228 286
229Asynchronously open or create a file and call the callback with a newly 287Asynchronously open or create a file and call the callback with a newly
230created filehandle for the file. 288created filehandle for the file.
236list. They are the same as used by C<sysopen>. 294list. They are the same as used by C<sysopen>.
237 295
238Likewise, C<$mode> specifies the mode of the newly created file, if it 296Likewise, C<$mode> specifies the mode of the newly created file, if it
239didn't exist and C<O_CREAT> has been given, just like perl's C<sysopen>, 297didn't exist and C<O_CREAT> has been given, just like perl's C<sysopen>,
240except that it is mandatory (i.e. use C<0> if you don't create new files, 298except that it is mandatory (i.e. use C<0> if you don't create new files,
241and C<0666> or C<0777> if you do). 299and C<0666> or C<0777> if you do). Note that the C<$mode> will be modified
300by the umask in effect then the request is being executed, so better never
301change the umask.
242 302
243Example: 303Example:
244 304
245 aio_open "/etc/passwd", O_RDONLY, 0, sub { 305 aio_open "/etc/passwd", O_RDONLY, 0, sub {
246 if ($_[0]) { 306 if ($_[0]) {
249 } else { 309 } else {
250 die "open failed: $!\n"; 310 die "open failed: $!\n";
251 } 311 }
252 }; 312 };
253 313
314
254=item aio_close $fh, $callback->($status) 315=item aio_close $fh, $callback->($status)
255 316
256Asynchronously close a file and call the callback with the result 317Asynchronously close a file and call the callback with the result
257code. I<WARNING:> although accepted, you should not pass in a perl 318code.
258filehandle here, as perl will likely close the file descriptor another
259time when the filehandle is destroyed. Normally, you can safely call perls
260C<close> or just let filehandles go out of scope.
261 319
262This is supposed to be a bug in the API, so that might change. It's 320Unfortunately, you can't do this to perl. Perl I<insists> very strongly on
263therefore best to avoid this function. 321closing the file descriptor associated with the filehandle itself. Here is
322what aio_close will try:
323
324 1. dup()licate the fd
325 2. asynchronously close() the duplicated fd
326 3. dup()licate the fd once more
327 4. let perl close() the filehandle
328 5. asynchronously close the duplicated fd
329
330The idea is that the first close() flushes stuff to disk that closing an
331fd will flush, so when perl closes the fd, nothing much will need to be
332flushed. The second async. close() will then flush stuff to disk that
333closing the last fd to the file will flush.
334
335Just FYI, SuSv3 has this to say on close:
336
337 All outstanding record locks owned by the process on the file
338 associated with the file descriptor shall be removed.
339
340 If fildes refers to a socket, close() shall cause the socket to be
341 destroyed. ... close() shall block for up to the current linger
342 interval until all data is transmitted.
343 [this actually sounds like a specification bug, but who knows]
344
345And at least Linux additionally actually flushes stuff on every close,
346even when the file itself is still open.
347
348Sounds enourmously inefficient and complicated? Yes... please show me how
349to nuke perl's fd out of existence...
350
351=cut
352
353sub aio_close($;$) {
354 aio_block {
355 my ($fh, $cb) = @_;
356
357 my $pri = aioreq_pri;
358 my $grp = aio_group $cb;
359
360 my $fd = fileno $fh;
361
362 defined $fd or Carp::croak "aio_close called with fd-less filehandle";
363
364 # if the dups fail we will simply get EBADF
365 my $fd2 = _dup $fd;
366 aioreq_pri $pri;
367 add $grp _aio_close $fd2, sub {
368 my $fd2 = _dup $fd;
369 close $fh;
370 aioreq_pri $pri;
371 add $grp _aio_close $fd2, sub {
372 $grp->result ($_[0]);
373 };
374 };
375
376 $grp
377 }
378}
379
264 380
265=item aio_read $fh,$offset,$length, $data,$dataoffset, $callback->($retval) 381=item aio_read $fh,$offset,$length, $data,$dataoffset, $callback->($retval)
266 382
267=item aio_write $fh,$offset,$length, $data,$dataoffset, $callback->($retval) 383=item aio_write $fh,$offset,$length, $data,$dataoffset, $callback->($retval)
268 384
269Reads or writes C<length> bytes from the specified C<fh> and C<offset> 385Reads or writes C<$length> bytes from the specified C<$fh> and C<$offset>
270into the scalar given by C<data> and offset C<dataoffset> and calls the 386into the scalar given by C<$data> and offset C<$dataoffset> and calls the
271callback without the actual number of bytes read (or -1 on error, just 387callback without the actual number of bytes read (or -1 on error, just
272like the syscall). 388like the syscall).
273 389
390If C<$offset> is undefined, then the current file descriptor offset will
391be used (and updated), otherwise the file descriptor offset will not be
392changed by these calls.
393
394If C<$length> is undefined in C<aio_write>, use the remaining length of C<$data>.
395
396If C<$dataoffset> is less than zero, it will be counted from the end of
397C<$data>.
398
274The C<$data> scalar I<MUST NOT> be modified in any way while the request 399The C<$data> scalar I<MUST NOT> be modified in any way while the request
275is outstanding. Modifying it can result in segfaults or WW3 (if the 400is outstanding. Modifying it can result in segfaults or World War III (if
276necessary/optional hardware is installed). 401the necessary/optional hardware is installed).
277 402
278Example: Read 15 bytes at offset 7 into scalar C<$buffer>, starting at 403Example: Read 15 bytes at offset 7 into scalar C<$buffer>, starting at
279offset C<0> within the scalar: 404offset C<0> within the scalar:
280 405
281 aio_read $fh, 7, 15, $buffer, 0, sub { 406 aio_read $fh, 7, 15, $buffer, 0, sub {
282 $_[0] > 0 or die "read error: $!"; 407 $_[0] > 0 or die "read error: $!";
283 print "read $_[0] bytes: <$buffer>\n"; 408 print "read $_[0] bytes: <$buffer>\n";
284 }; 409 };
410
285 411
286=item aio_sendfile $out_fh, $in_fh, $in_offset, $length, $callback->($retval) 412=item aio_sendfile $out_fh, $in_fh, $in_offset, $length, $callback->($retval)
287 413
288Tries to copy C<$length> bytes from C<$in_fh> to C<$out_fh>. It starts 414Tries to copy C<$length> bytes from C<$in_fh> to C<$out_fh>. It starts
289reading at byte offset C<$in_offset>, and starts writing at the current 415reading at byte offset C<$in_offset>, and starts writing at the current
303C<$in_fh> than are written, and there is no way to find out how many 429C<$in_fh> than are written, and there is no way to find out how many
304bytes have been read from C<aio_sendfile> alone, as C<aio_sendfile> only 430bytes have been read from C<aio_sendfile> alone, as C<aio_sendfile> only
305provides the number of bytes written to C<$out_fh>. Only if the result 431provides the number of bytes written to C<$out_fh>. Only if the result
306value equals C<$length> one can assume that C<$length> bytes have been 432value equals C<$length> one can assume that C<$length> bytes have been
307read. 433read.
434
308 435
309=item aio_readahead $fh,$offset,$length, $callback->($retval) 436=item aio_readahead $fh,$offset,$length, $callback->($retval)
310 437
311C<aio_readahead> populates the page cache with data from a file so that 438C<aio_readahead> populates the page cache with data from a file so that
312subsequent reads from that file will not block on disk I/O. The C<$offset> 439subsequent reads from that file will not block on disk I/O. The C<$offset>
318file. The current file offset of the file is left unchanged. 445file. The current file offset of the file is left unchanged.
319 446
320If that syscall doesn't exist (likely if your OS isn't Linux) it will be 447If that syscall doesn't exist (likely if your OS isn't Linux) it will be
321emulated by simply reading the data, which would have a similar effect. 448emulated by simply reading the data, which would have a similar effect.
322 449
450
323=item aio_stat $fh_or_path, $callback->($status) 451=item aio_stat $fh_or_path, $callback->($status)
324 452
325=item aio_lstat $fh, $callback->($status) 453=item aio_lstat $fh, $callback->($status)
326 454
327Works like perl's C<stat> or C<lstat> in void context. The callback will 455Works like perl's C<stat> or C<lstat> in void context. The callback will
340 aio_stat "/etc/passwd", sub { 468 aio_stat "/etc/passwd", sub {
341 $_[0] and die "stat failed: $!"; 469 $_[0] and die "stat failed: $!";
342 print "size is ", -s _, "\n"; 470 print "size is ", -s _, "\n";
343 }; 471 };
344 472
473
474=item aio_utime $fh_or_path, $atime, $mtime, $callback->($status)
475
476Works like perl's C<utime> function (including the special case of $atime
477and $mtime being undef). Fractional times are supported if the underlying
478syscalls support them.
479
480When called with a pathname, uses utimes(2) if available, otherwise
481utime(2). If called on a file descriptor, uses futimes(2) if available,
482otherwise returns ENOSYS, so this is not portable.
483
484Examples:
485
486 # set atime and mtime to current time (basically touch(1)):
487 aio_utime "path", undef, undef;
488 # set atime to current time and mtime to beginning of the epoch:
489 aio_utime "path", time, undef; # undef==0
490
491
492=item aio_chown $fh_or_path, $uid, $gid, $callback->($status)
493
494Works like perl's C<chown> function, except that C<undef> for either $uid
495or $gid is being interpreted as "do not change" (but -1 can also be used).
496
497Examples:
498
499 # same as "chown root path" in the shell:
500 aio_chown "path", 0, -1;
501 # same as above:
502 aio_chown "path", 0, undef;
503
504
505=item aio_truncate $fh_or_path, $offset, $callback->($status)
506
507Works like truncate(2) or ftruncate(2).
508
509
510=item aio_chmod $fh_or_path, $mode, $callback->($status)
511
512Works like perl's C<chmod> function.
513
514
345=item aio_unlink $pathname, $callback->($status) 515=item aio_unlink $pathname, $callback->($status)
346 516
347Asynchronously unlink (delete) a file and call the callback with the 517Asynchronously unlink (delete) a file and call the callback with the
348result code. 518result code.
349 519
520
350=item aio_mknod $path, $mode, $dev, $callback->($status) 521=item aio_mknod $path, $mode, $dev, $callback->($status)
351 522
523[EXPERIMENTAL]
524
352Asynchronously create a device node (or fifo). See mknod(2). 525Asynchronously create a device node (or fifo). See mknod(2).
353 526
354The only portable (POSIX) way of calling this function is: 527The only (POSIX-) portable way of calling this function is:
355 528
356 aio_mknod $path, IO::AIO::S_IFIFO | $mode, 0, sub { ... 529 aio_mknod $path, IO::AIO::S_IFIFO | $mode, 0, sub { ...
530
357 531
358=item aio_link $srcpath, $dstpath, $callback->($status) 532=item aio_link $srcpath, $dstpath, $callback->($status)
359 533
360Asynchronously create a new link to the existing object at C<$srcpath> at 534Asynchronously create a new link to the existing object at C<$srcpath> at
361the path C<$dstpath> and call the callback with the result code. 535the path C<$dstpath> and call the callback with the result code.
362 536
537
363=item aio_symlink $srcpath, $dstpath, $callback->($status) 538=item aio_symlink $srcpath, $dstpath, $callback->($status)
364 539
365Asynchronously create a new symbolic link to the existing object at C<$srcpath> at 540Asynchronously create a new symbolic link to the existing object at C<$srcpath> at
366the path C<$dstpath> and call the callback with the result code. 541the path C<$dstpath> and call the callback with the result code.
367 542
543
544=item aio_readlink $path, $callback->($link)
545
546Asynchronously read the symlink specified by C<$path> and pass it to
547the callback. If an error occurs, nothing or undef gets passed to the
548callback.
549
550
368=item aio_rename $srcpath, $dstpath, $callback->($status) 551=item aio_rename $srcpath, $dstpath, $callback->($status)
369 552
370Asynchronously rename the object at C<$srcpath> to C<$dstpath>, just as 553Asynchronously rename the object at C<$srcpath> to C<$dstpath>, just as
371rename(2) and call the callback with the result code. 554rename(2) and call the callback with the result code.
372 555
556
557=item aio_mkdir $pathname, $mode, $callback->($status)
558
559Asynchronously mkdir (create) a directory and call the callback with
560the result code. C<$mode> will be modified by the umask at the time the
561request is executed, so do not change your umask.
562
563
373=item aio_rmdir $pathname, $callback->($status) 564=item aio_rmdir $pathname, $callback->($status)
374 565
375Asynchronously rmdir (delete) a directory and call the callback with the 566Asynchronously rmdir (delete) a directory and call the callback with the
376result code. 567result code.
568
377 569
378=item aio_readdir $pathname, $callback->($entries) 570=item aio_readdir $pathname, $callback->($entries)
379 571
380Unlike the POSIX call of the same name, C<aio_readdir> reads an entire 572Unlike the POSIX call of the same name, C<aio_readdir> reads an entire
381directory (i.e. opendir + readdir + closedir). The entries will not be 573directory (i.e. opendir + readdir + closedir). The entries will not be
382sorted, and will B<NOT> include the C<.> and C<..> entries. 574sorted, and will B<NOT> include the C<.> and C<..> entries.
383 575
384The callback a single argument which is either C<undef> or an array-ref 576The callback a single argument which is either C<undef> or an array-ref
385with the filenames. 577with the filenames.
578
579
580=item aio_load $path, $data, $callback->($status)
581
582This is a composite request that tries to fully load the given file into
583memory. Status is the same as with aio_read.
584
585=cut
586
587sub aio_load($$;$) {
588 aio_block {
589 my ($path, undef, $cb) = @_;
590 my $data = \$_[1];
591
592 my $pri = aioreq_pri;
593 my $grp = aio_group $cb;
594
595 aioreq_pri $pri;
596 add $grp aio_open $path, O_RDONLY, 0, sub {
597 my $fh = shift
598 or return $grp->result (-1);
599
600 aioreq_pri $pri;
601 add $grp aio_read $fh, 0, (-s $fh), $$data, 0, sub {
602 $grp->result ($_[0]);
603 };
604 };
605
606 $grp
607 }
608}
386 609
387=item aio_copy $srcpath, $dstpath, $callback->($status) 610=item aio_copy $srcpath, $dstpath, $callback->($status)
388 611
389Try to copy the I<file> (directories not supported as either source or 612Try to copy the I<file> (directories not supported as either source or
390destination) from C<$srcpath> to C<$dstpath> and call the callback with 613destination) from C<$srcpath> to C<$dstpath> and call the callback with
400errors are being ignored. 623errors are being ignored.
401 624
402=cut 625=cut
403 626
404sub aio_copy($$;$) { 627sub aio_copy($$;$) {
628 aio_block {
405 my ($src, $dst, $cb) = @_; 629 my ($src, $dst, $cb) = @_;
406 630
407 my $pri = aioreq_pri; 631 my $pri = aioreq_pri;
408 my $grp = aio_group $cb; 632 my $grp = aio_group $cb;
409 633
410 aioreq_pri $pri; 634 aioreq_pri $pri;
411 add $grp aio_open $src, O_RDONLY, 0, sub { 635 add $grp aio_open $src, O_RDONLY, 0, sub {
412 if (my $src_fh = $_[0]) { 636 if (my $src_fh = $_[0]) {
413 my @stat = stat $src_fh; 637 my @stat = stat $src_fh;
414 638
415 aioreq_pri $pri; 639 aioreq_pri $pri;
416 add $grp aio_open $dst, O_CREAT | O_WRONLY | O_TRUNC, 0200, sub { 640 add $grp aio_open $dst, O_CREAT | O_WRONLY | O_TRUNC, 0200, sub {
417 if (my $dst_fh = $_[0]) { 641 if (my $dst_fh = $_[0]) {
418 aioreq_pri $pri; 642 aioreq_pri $pri;
419 add $grp aio_sendfile $dst_fh, $src_fh, 0, $stat[7], sub { 643 add $grp aio_sendfile $dst_fh, $src_fh, 0, $stat[7], sub {
420 if ($_[0] == $stat[7]) { 644 if ($_[0] == $stat[7]) {
421 $grp->result (0); 645 $grp->result (0);
422 close $src_fh; 646 close $src_fh;
423 647
424 # those should not normally block. should. should. 648 # those should not normally block. should. should.
425 utime $stat[8], $stat[9], $dst; 649 utime $stat[8], $stat[9], $dst;
426 chmod $stat[2] & 07777, $dst_fh; 650 chmod $stat[2] & 07777, $dst_fh;
427 chown $stat[4], $stat[5], $dst_fh; 651 chown $stat[4], $stat[5], $dst_fh;
428 close $dst_fh; 652 close $dst_fh;
429 } else { 653 } else {
430 $grp->result (-1); 654 $grp->result (-1);
431 close $src_fh; 655 close $src_fh;
432 close $dst_fh; 656 close $dst_fh;
433 657
434 aioreq $pri; 658 aioreq $pri;
435 add $grp aio_unlink $dst; 659 add $grp aio_unlink $dst;
660 }
436 } 661 };
662 } else {
663 $grp->result (-1);
437 }; 664 }
438 } else {
439 $grp->result (-1);
440 } 665 },
666
667 } else {
668 $grp->result (-1);
441 }, 669 }
442
443 } else {
444 $grp->result (-1);
445 } 670 };
671
672 $grp
446 }; 673 }
447
448 $grp
449} 674}
450 675
451=item aio_move $srcpath, $dstpath, $callback->($status) 676=item aio_move $srcpath, $dstpath, $callback->($status)
452 677
453Try to move the I<file> (directories not supported as either source or 678Try to move the I<file> (directories not supported as either source or
459that is successful, unlinking the C<$srcpath>. 684that is successful, unlinking the C<$srcpath>.
460 685
461=cut 686=cut
462 687
463sub aio_move($$;$) { 688sub aio_move($$;$) {
689 aio_block {
464 my ($src, $dst, $cb) = @_; 690 my ($src, $dst, $cb) = @_;
465 691
466 my $pri = aioreq_pri; 692 my $pri = aioreq_pri;
467 my $grp = aio_group $cb; 693 my $grp = aio_group $cb;
468 694
469 aioreq_pri $pri; 695 aioreq_pri $pri;
470 add $grp aio_rename $src, $dst, sub { 696 add $grp aio_rename $src, $dst, sub {
471 if ($_[0] && $! == EXDEV) { 697 if ($_[0] && $! == EXDEV) {
472 aioreq_pri $pri; 698 aioreq_pri $pri;
473 add $grp aio_copy $src, $dst, sub { 699 add $grp aio_copy $src, $dst, sub {
700 $grp->result ($_[0]);
701
702 if (!$_[0]) {
703 aioreq_pri $pri;
704 add $grp aio_unlink $src;
705 }
706 };
707 } else {
474 $grp->result ($_[0]); 708 $grp->result ($_[0]);
475
476 if (!$_[0]) {
477 aioreq_pri $pri;
478 add $grp aio_unlink $src;
479 }
480 }; 709 }
481 } else {
482 $grp->result ($_[0]);
483 } 710 };
711
712 $grp
484 }; 713 }
485
486 $grp
487} 714}
488 715
489=item aio_scandir $path, $maxreq, $callback->($dirs, $nondirs) 716=item aio_scandir $path, $maxreq, $callback->($dirs, $nondirs)
490 717
491Scans a directory (similar to C<aio_readdir>) but additionally tries to 718Scans a directory (similar to C<aio_readdir>) but additionally tries to
538as those tend to return 0 or 1 as link counts, which disables the 765as those tend to return 0 or 1 as link counts, which disables the
539directory counting heuristic. 766directory counting heuristic.
540 767
541=cut 768=cut
542 769
543sub aio_scandir($$$) { 770sub aio_scandir($$;$) {
771 aio_block {
544 my ($path, $maxreq, $cb) = @_; 772 my ($path, $maxreq, $cb) = @_;
545 773
546 my $pri = aioreq_pri; 774 my $pri = aioreq_pri;
547 775
548 my $grp = aio_group $cb; 776 my $grp = aio_group $cb;
549 777
550 $maxreq = 4 if $maxreq <= 0; 778 $maxreq = 4 if $maxreq <= 0;
551 779
552 # stat once 780 # stat once
553 aioreq_pri $pri;
554 add $grp aio_stat $path, sub {
555 return $grp->result () if $_[0];
556 my $now = time;
557 my $hash1 = join ":", (stat _)[0,1,3,7,9];
558
559 # read the directory entries
560 aioreq_pri $pri; 781 aioreq_pri $pri;
561 add $grp aio_readdir $path, sub { 782 add $grp aio_stat $path, sub {
562 my $entries = shift
563 or return $grp->result (); 783 return $grp->result () if $_[0];
784 my $now = time;
785 my $hash1 = join ":", (stat _)[0,1,3,7,9];
564 786
565 # stat the dir another time 787 # read the directory entries
566 aioreq_pri $pri; 788 aioreq_pri $pri;
789 add $grp aio_readdir $path, sub {
790 my $entries = shift
791 or return $grp->result ();
792
793 # stat the dir another time
794 aioreq_pri $pri;
567 add $grp aio_stat $path, sub { 795 add $grp aio_stat $path, sub {
568 my $hash2 = join ":", (stat _)[0,1,3,7,9]; 796 my $hash2 = join ":", (stat _)[0,1,3,7,9];
569 797
570 my $ndirs; 798 my $ndirs;
571 799
572 # take the slow route if anything looks fishy 800 # take the slow route if anything looks fishy
573 if ($hash1 ne $hash2 or (stat _)[9] == $now) { 801 if ($hash1 ne $hash2 or (stat _)[9] == $now) {
574 $ndirs = -1; 802 $ndirs = -1;
575 } else { 803 } else {
576 # if nlink == 2, we are finished 804 # if nlink == 2, we are finished
577 # on non-posix-fs's, we rely on nlink < 2 805 # on non-posix-fs's, we rely on nlink < 2
578 $ndirs = (stat _)[3] - 2 806 $ndirs = (stat _)[3] - 2
579 or return $grp->result ([], $entries); 807 or return $grp->result ([], $entries);
580 } 808 }
581 809
582 # sort into likely dirs and likely nondirs 810 # sort into likely dirs and likely nondirs
583 # dirs == files without ".", short entries first 811 # dirs == files without ".", short entries first
584 $entries = [map $_->[0], 812 $entries = [map $_->[0],
585 sort { $b->[1] cmp $a->[1] } 813 sort { $b->[1] cmp $a->[1] }
586 map [$_, sprintf "%s%04d", (/.\./ ? "1" : "0"), length], 814 map [$_, sprintf "%s%04d", (/.\./ ? "1" : "0"), length],
587 @$entries]; 815 @$entries];
588 816
589 my (@dirs, @nondirs); 817 my (@dirs, @nondirs);
590 818
591 my $statgrp = add $grp aio_group sub { 819 my $statgrp = add $grp aio_group sub {
592 $grp->result (\@dirs, \@nondirs); 820 $grp->result (\@dirs, \@nondirs);
593 }; 821 };
594 822
595 limit $statgrp $maxreq; 823 limit $statgrp $maxreq;
596 feed $statgrp sub { 824 feed $statgrp sub {
597 return unless @$entries; 825 return unless @$entries;
598 my $entry = pop @$entries; 826 my $entry = pop @$entries;
599 827
600 aioreq_pri $pri; 828 aioreq_pri $pri;
601 add $statgrp aio_stat "$path/$entry/.", sub { 829 add $statgrp aio_stat "$path/$entry/.", sub {
602 if ($_[0] < 0) { 830 if ($_[0] < 0) {
603 push @nondirs, $entry; 831 push @nondirs, $entry;
604 } else { 832 } else {
605 # need to check for real directory 833 # need to check for real directory
606 aioreq_pri $pri; 834 aioreq_pri $pri;
607 add $statgrp aio_lstat "$path/$entry", sub { 835 add $statgrp aio_lstat "$path/$entry", sub {
608 if (-d _) { 836 if (-d _) {
609 push @dirs, $entry; 837 push @dirs, $entry;
610 838
611 unless (--$ndirs) { 839 unless (--$ndirs) {
612 push @nondirs, @$entries; 840 push @nondirs, @$entries;
613 feed $statgrp; 841 feed $statgrp;
842 }
843 } else {
844 push @nondirs, $entry;
614 } 845 }
615 } else {
616 push @nondirs, $entry;
617 } 846 }
618 } 847 }
619 } 848 };
620 }; 849 };
621 }; 850 };
622 }; 851 };
623 }; 852 };
853
854 $grp
624 }; 855 }
856}
625 857
858=item aio_rmtree $path, $callback->($status)
859
860Delete a directory tree starting (and including) C<$path>, return the
861status of the final C<rmdir> only. This is a composite request that
862uses C<aio_scandir> to recurse into and rmdir directories, and unlink
863everything else.
864
865=cut
866
867sub aio_rmtree;
868sub aio_rmtree($;$) {
869 aio_block {
870 my ($path, $cb) = @_;
871
872 my $pri = aioreq_pri;
873 my $grp = aio_group $cb;
874
875 aioreq_pri $pri;
876 add $grp aio_scandir $path, 0, sub {
877 my ($dirs, $nondirs) = @_;
878
879 my $dirgrp = aio_group sub {
880 add $grp aio_rmdir $path, sub {
881 $grp->result ($_[0]);
882 };
883 };
884
885 (aioreq_pri $pri), add $dirgrp aio_rmtree "$path/$_" for @$dirs;
886 (aioreq_pri $pri), add $dirgrp aio_unlink "$path/$_" for @$nondirs;
887
888 add $grp $dirgrp;
889 };
890
626 $grp 891 $grp
892 }
627} 893}
628 894
629=item aio_fsync $fh, $callback->($status) 895=item aio_fsync $fh, $callback->($status)
630 896
631Asynchronously call fsync on the given filehandle and call the callback 897Asynchronously call fsync on the given filehandle and call the callback
845 1111
846=back 1112=back
847 1113
848=head2 SUPPORT FUNCTIONS 1114=head2 SUPPORT FUNCTIONS
849 1115
1116=head3 EVENT PROCESSING AND EVENT LOOP INTEGRATION
1117
850=over 4 1118=over 4
851 1119
852=item $fileno = IO::AIO::poll_fileno 1120=item $fileno = IO::AIO::poll_fileno
853 1121
854Return the I<request result pipe file descriptor>. This filehandle must be 1122Return the I<request result pipe file descriptor>. This filehandle must be
858 1126
859See C<poll_cb> for an example. 1127See C<poll_cb> for an example.
860 1128
861=item IO::AIO::poll_cb 1129=item IO::AIO::poll_cb
862 1130
863Process all outstanding events on the result pipe. You have to call this 1131Process some outstanding events on the result pipe. You have to call this
864regularly. Returns the number of events processed. Returns immediately 1132regularly. Returns the number of events processed. Returns immediately
865when no events are outstanding. 1133when no events are outstanding. The amount of events processed depends on
1134the settings of C<IO::AIO::max_poll_req> and C<IO::AIO::max_poll_time>.
866 1135
867If not all requests were processed for whatever reason, the filehandle 1136If not all requests were processed for whatever reason, the filehandle
868will still be ready when C<poll_cb> returns. 1137will still be ready when C<poll_cb> returns.
869 1138
870Example: Install an Event watcher that automatically calls 1139Example: Install an Event watcher that automatically calls
872 1141
873 Event->io (fd => IO::AIO::poll_fileno, 1142 Event->io (fd => IO::AIO::poll_fileno,
874 poll => 'r', async => 1, 1143 poll => 'r', async => 1,
875 cb => \&IO::AIO::poll_cb); 1144 cb => \&IO::AIO::poll_cb);
876 1145
877=item IO::AIO::poll_some $max_requests 1146=item IO::AIO::max_poll_reqs $nreqs
878 1147
879Similar to C<poll_cb>, but only processes up to C<$max_requests> requests 1148=item IO::AIO::max_poll_time $seconds
880at a time.
881 1149
882Useful if you want to ensure some level of interactiveness when perl is 1150These set the maximum number of requests (default C<0>, meaning infinity)
883not fast enough to process all requests in time. 1151that are being processed by C<IO::AIO::poll_cb> in one call, respectively
1152the maximum amount of time (default C<0>, meaning infinity) spent in
1153C<IO::AIO::poll_cb> to process requests (more correctly the mininum amount
1154of time C<poll_cb> is allowed to use).
1155
1156Setting C<max_poll_time> to a non-zero value creates an overhead of one
1157syscall per request processed, which is not normally a problem unless your
1158callbacks are really really fast or your OS is really really slow (I am
1159not mentioning Solaris here). Using C<max_poll_reqs> incurs no overhead.
1160
1161Setting these is useful if you want to ensure some level of
1162interactiveness when perl is not fast enough to process all requests in
1163time.
1164
1165For interactive programs, values such as C<0.01> to C<0.1> should be fine.
884 1166
885Example: Install an Event watcher that automatically calls 1167Example: Install an Event watcher that automatically calls
886IO::AIO::poll_some with low priority, to ensure that other parts of the 1168IO::AIO::poll_cb with low priority, to ensure that other parts of the
887program get the CPU sometimes even under high AIO load. 1169program get the CPU sometimes even under high AIO load.
888 1170
1171 # try not to spend much more than 0.1s in poll_cb
1172 IO::AIO::max_poll_time 0.1;
1173
1174 # use a low priority so other tasks have priority
889 Event->io (fd => IO::AIO::poll_fileno, 1175 Event->io (fd => IO::AIO::poll_fileno,
890 poll => 'r', nice => 1, 1176 poll => 'r', nice => 1,
891 cb => sub { IO::AIO::poll_some 256 }); 1177 cb => &IO::AIO::poll_cb);
892 1178
893=item IO::AIO::poll_wait 1179=item IO::AIO::poll_wait
894 1180
1181If there are any outstanding requests and none of them in the result
895Wait till the result filehandle becomes ready for reading (simply does a 1182phase, wait till the result filehandle becomes ready for reading (simply
896C<select> on the filehandle. This is useful if you want to synchronously wait 1183does a C<select> on the filehandle. This is useful if you want to
897for some requests to finish). 1184synchronously wait for some requests to finish).
898 1185
899See C<nreqs> for an example. 1186See C<nreqs> for an example.
900 1187
1188=item IO::AIO::poll
1189
1190Waits until some requests have been handled.
1191
1192Returns the number of requests processed, but is otherwise strictly
1193equivalent to:
1194
1195 IO::AIO::poll_wait, IO::AIO::poll_cb
1196
901=item IO::AIO::nreqs 1197=item IO::AIO::flush
902 1198
903Returns the number of requests currently in the ready, execute or pending 1199Wait till all outstanding AIO requests have been handled.
904states (i.e. for which their callback has not been invoked yet).
905 1200
906Example: wait till there are no outstanding requests anymore: 1201Strictly equivalent to:
907 1202
908 IO::AIO::poll_wait, IO::AIO::poll_cb 1203 IO::AIO::poll_wait, IO::AIO::poll_cb
909 while IO::AIO::nreqs; 1204 while IO::AIO::nreqs;
910 1205
911=item IO::AIO::nready 1206=back
912 1207
913Returns the number of requests currently in the ready state (not yet 1208=head3 CONTROLLING THE NUMBER OF THREADS
914executed).
915 1209
916=item IO::AIO::npending 1210=over
917
918Returns the number of requests currently in the pending state (executed,
919but not yet processed by poll_cb).
920
921=item IO::AIO::flush
922
923Wait till all outstanding AIO requests have been handled.
924
925Strictly equivalent to:
926
927 IO::AIO::poll_wait, IO::AIO::poll_cb
928 while IO::AIO::nreqs;
929
930=item IO::AIO::poll
931
932Waits until some requests have been handled.
933
934Strictly equivalent to:
935
936 IO::AIO::poll_wait, IO::AIO::poll_cb
937 if IO::AIO::nreqs;
938 1211
939=item IO::AIO::min_parallel $nthreads 1212=item IO::AIO::min_parallel $nthreads
940 1213
941Set the minimum number of AIO threads to C<$nthreads>. The current 1214Set the minimum number of AIO threads to C<$nthreads>. The current
942default is C<8>, which means eight asynchronous operations can execute 1215default is C<8>, which means eight asynchronous operations can execute
943concurrently at any one time (the number of outstanding requests, 1216concurrently at any one time (the number of outstanding requests,
944however, is unlimited). 1217however, is unlimited).
945 1218
946IO::AIO starts threads only on demand, when an AIO request is queued and 1219IO::AIO starts threads only on demand, when an AIO request is queued and
947no free thread exists. 1220no free thread exists. Please note that queueing up a hundred requests can
1221create demand for a hundred threads, even if it turns out that everything
1222is in the cache and could have been processed faster by a single thread.
948 1223
949It is recommended to keep the number of threads relatively low, as some 1224It is recommended to keep the number of threads relatively low, as some
950Linux kernel versions will scale negatively with the number of threads 1225Linux kernel versions will scale negatively with the number of threads
951(higher parallelity => MUCH higher latency). With current Linux 2.6 1226(higher parallelity => MUCH higher latency). With current Linux 2.6
952versions, 4-32 threads should be fine. 1227versions, 4-32 threads should be fine.
966This module automatically runs C<max_parallel 0> at program end, to ensure 1241This module automatically runs C<max_parallel 0> at program end, to ensure
967that all threads are killed and that there are no outstanding requests. 1242that all threads are killed and that there are no outstanding requests.
968 1243
969Under normal circumstances you don't need to call this function. 1244Under normal circumstances you don't need to call this function.
970 1245
1246=item IO::AIO::max_idle $nthreads
1247
1248Limit the number of threads (default: 4) that are allowed to idle (i.e.,
1249threads that did not get a request to process within 10 seconds). That
1250means if a thread becomes idle while C<$nthreads> other threads are also
1251idle, it will free its resources and exit.
1252
1253This is useful when you allow a large number of threads (e.g. 100 or 1000)
1254to allow for extremely high load situations, but want to free resources
1255under normal circumstances (1000 threads can easily consume 30MB of RAM).
1256
1257The default is probably ok in most situations, especially if thread
1258creation is fast. If thread creation is very slow on your system you might
1259want to use larger values.
1260
971=item $oldmaxreqs = IO::AIO::max_outstanding $maxreqs 1261=item $oldmaxreqs = IO::AIO::max_outstanding $maxreqs
972 1262
973This is a very bad function to use in interactive programs because it 1263This is a very bad function to use in interactive programs because it
974blocks, and a bad way to reduce concurrency because it is inexact: Better 1264blocks, and a bad way to reduce concurrency because it is inexact: Better
975use an C<aio_group> together with a feed callback. 1265use an C<aio_group> together with a feed callback.
976 1266
977Sets the maximum number of outstanding requests to C<$nreqs>. If you 1267Sets the maximum number of outstanding requests to C<$nreqs>. If you
978to queue up more than this number of requests, the next call to the 1268do queue up more than this number of requests, the next call to the
979C<poll_cb> (and C<poll_some> and other functions calling C<poll_cb>) 1269C<poll_cb> (and C<poll_some> and other functions calling C<poll_cb>)
980function will block until the limit is no longer exceeded. 1270function will block until the limit is no longer exceeded.
981 1271
982The default value is very large, so there is no practical limit on the 1272The default value is very large, so there is no practical limit on the
983number of outstanding requests. 1273number of outstanding requests.
986C<max_oustsanding> is mainly useful in simple scripts (with low values) or 1276C<max_oustsanding> is mainly useful in simple scripts (with low values) or
987as a stop gap to shield against fatal memory overflow (with large values). 1277as a stop gap to shield against fatal memory overflow (with large values).
988 1278
989=back 1279=back
990 1280
1281=head3 STATISTICAL INFORMATION
1282
1283=over
1284
1285=item IO::AIO::nreqs
1286
1287Returns the number of requests currently in the ready, execute or pending
1288states (i.e. for which their callback has not been invoked yet).
1289
1290Example: wait till there are no outstanding requests anymore:
1291
1292 IO::AIO::poll_wait, IO::AIO::poll_cb
1293 while IO::AIO::nreqs;
1294
1295=item IO::AIO::nready
1296
1297Returns the number of requests currently in the ready state (not yet
1298executed).
1299
1300=item IO::AIO::npending
1301
1302Returns the number of requests currently in the pending state (executed,
1303but not yet processed by poll_cb).
1304
1305=back
1306
991=cut 1307=cut
992 1308
993# support function to convert a fd into a perl filehandle
994sub _fd2fh {
995 return undef if $_[0] < 0;
996
997 # try to generate nice filehandles
998 my $sym = "IO::AIO::fd#$_[0]";
999 local *$sym;
1000
1001 open *$sym, "+<&=$_[0]" # usually works under any unix
1002 or open *$sym, "<&=$_[0]" # cygwin needs this
1003 or open *$sym, ">&=$_[0]" # or this
1004 or return undef;
1005
1006 *$sym
1007}
1008
1009min_parallel 8; 1309min_parallel 8;
1010 1310
1011END { 1311END { flush }
1012 min_parallel 1;
1013 flush;
1014};
1015 1312
10161; 13131;
1017 1314
1018=head2 FORK BEHAVIOUR 1315=head2 FORK BEHAVIOUR
1019 1316
1039bytes of memory. In addition, stat requests need a stat buffer (possibly 1336bytes of memory. In addition, stat requests need a stat buffer (possibly
1040a few hundred bytes), readdir requires a result buffer and so on. Perl 1337a few hundred bytes), readdir requires a result buffer and so on. Perl
1041scalars and other data passed into aio requests will also be locked and 1338scalars and other data passed into aio requests will also be locked and
1042will consume memory till the request has entered the done state. 1339will consume memory till the request has entered the done state.
1043 1340
1044This is now awfully much, so queuing lots of requests is not usually a 1341This is not awfully much, so queuing lots of requests is not usually a
1045problem. 1342problem.
1046 1343
1047Per-thread usage: 1344Per-thread usage:
1048 1345
1049In the execution phase, some aio requests require more memory for 1346In the execution phase, some aio requests require more memory for

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines