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.116 by root, Wed Oct 3 21:27:51 2007 UTC vs.
Revision 1.125 by root, Sat May 10 22:47:34 2008 UTC

… …
26 $req->cancel; # cancel request if still in queue 26 $req->cancel; # cancel request if still in queue
27 27
28 my $grp = aio_group sub { print "all stats done\n" }; 28 my $grp = aio_group sub { print "all stats done\n" };
29 add $grp aio_stat "..." for ...; 29 add $grp aio_stat "..." for ...;
30 30
31 # AnyEvent integration (EV, Event, Glib, Tk, POE, urxvt, pureperl...)
32 use AnyEvent::AIO;
33
31 # AnyEvent integration 34 # EV integration
32 open my $fh, "<&=" . IO::AIO::poll_fileno or die "$!"; 35 my $w = EV::io IO::AIO::poll_fileno, EV::READ, \&IO::AIO::poll_cb;
33 my $w = AnyEvent->io (fh => $fh, poll => 'r', cb => sub { IO::AIO::poll_cb });
34 36
35 # Event integration 37 # Event integration
36 Event->io (fd => IO::AIO::poll_fileno, 38 Event->io (fd => IO::AIO::poll_fileno,
37 poll => 'r', 39 poll => 'r',
38 cb => \&IO::AIO::poll_cb); 40 cb => \&IO::AIO::poll_cb);
… …
183 185
184=cut 186=cut
185 187
186package IO::AIO; 188package IO::AIO;
187 189
190use Carp ();
191
188no warnings; 192no warnings;
189use strict 'vars'; 193use strict 'vars';
190 194
191use base 'Exporter'; 195use base 'Exporter';
192 196
193BEGIN { 197BEGIN {
194 our $VERSION = '2.5'; 198 our $VERSION = '3.0';
195 199
196 our @AIO_REQ = qw(aio_sendfile aio_read aio_write aio_open aio_close aio_stat 200 our @AIO_REQ = qw(aio_sendfile aio_read aio_write aio_open aio_close
197 aio_lstat aio_unlink aio_rmdir aio_readdir aio_scandir aio_symlink 201 aio_stat aio_lstat aio_unlink aio_rmdir aio_readdir
198 aio_readlink aio_fsync aio_fdatasync aio_readahead aio_rename aio_link 202 aio_scandir aio_symlink aio_readlink aio_sync aio_fsync
203 aio_fdatasync aio_pathsync aio_readahead
204 aio_rename aio_link aio_move aio_copy aio_group
199 aio_move aio_copy aio_group aio_nop aio_mknod aio_load aio_rmtree aio_mkdir 205 aio_nop aio_mknod aio_load aio_rmtree aio_mkdir aio_chown
200 aio_chown aio_chmod aio_utime aio_truncate); 206 aio_chmod aio_utime aio_truncate);
207
201 our @EXPORT = (@AIO_REQ, qw(aioreq_pri aioreq_nice aio_block)); 208 our @EXPORT = (@AIO_REQ, qw(aioreq_pri aioreq_nice));
202 our @EXPORT_OK = qw(poll_fileno poll_cb poll_wait flush 209 our @EXPORT_OK = qw(poll_fileno poll_cb poll_wait flush
203 min_parallel max_parallel max_idle 210 min_parallel max_parallel max_idle
204 nreqs nready npending nthreads 211 nreqs nready npending nthreads
205 max_poll_time max_poll_reqs); 212 max_poll_time max_poll_reqs);
206 213
… …
313=item aio_close $fh, $callback->($status) 320=item aio_close $fh, $callback->($status)
314 321
315Asynchronously close a file and call the callback with the result 322Asynchronously close a file and call the callback with the result
316code. 323code.
317 324
318Unlike the other functions operating on files, this function uses the 325Unfortunately, you can't do this to perl. Perl I<insists> very strongly on
319PerlIO layer to close the filehandle. The reason is that the PerlIO API 326closing the file descriptor associated with the filehandle itself.
320insists on closing the underlying fd itself, no matter what, and doesn't
321allow modifications to the fd. Unfortunately, it is not clear that you can
322call PerlIO from different threads (actually, its quite clear that this
323won't work in some cases), so while it likely works perfectly with simple
324file handles (such as the ones created by C<aio_open>) it might fail in
325interesting ways for others.
326 327
327Having said that, aio_close tries to clean up the filehandle as much as 328Therefore, C<aio_close> will not close the filehandle - instead it will
328possible before handing it to an io thread, and generally does work. 329use dup2 to overwrite the file descriptor with the write-end of a pipe
330(the pipe fd will be created on demand and will be cached).
329 331
332Or in other words: the file descriptor will be closed, but it will not be
333free for reuse until the perl filehandle is closed.
334
335=cut
330 336
331=item aio_read $fh,$offset,$length, $data,$dataoffset, $callback->($retval) 337=item aio_read $fh,$offset,$length, $data,$dataoffset, $callback->($retval)
332 338
333=item aio_write $fh,$offset,$length, $data,$dataoffset, $callback->($retval) 339=item aio_write $fh,$offset,$length, $data,$dataoffset, $callback->($retval)
334 340
… …
533memory. Status is the same as with aio_read. 539memory. Status is the same as with aio_read.
534 540
535=cut 541=cut
536 542
537sub aio_load($$;$) { 543sub aio_load($$;$) {
538 aio_block {
539 my ($path, undef, $cb) = @_; 544 my ($path, undef, $cb) = @_;
540 my $data = \$_[1]; 545 my $data = \$_[1];
541 546
542 my $pri = aioreq_pri; 547 my $pri = aioreq_pri;
543 my $grp = aio_group $cb; 548 my $grp = aio_group $cb;
549
550 aioreq_pri $pri;
551 add $grp aio_open $path, O_RDONLY, 0, sub {
552 my $fh = shift
553 or return $grp->result (-1);
544 554
545 aioreq_pri $pri; 555 aioreq_pri $pri;
546 add $grp aio_open $path, O_RDONLY, 0, sub {
547 my $fh = shift
548 or return $grp->result (-1);
549
550 aioreq_pri $pri;
551 add $grp aio_read $fh, 0, (-s $fh), $$data, 0, sub { 556 add $grp aio_read $fh, 0, (-s $fh), $$data, 0, sub {
552 $grp->result ($_[0]); 557 $grp->result ($_[0]);
553 };
554 }; 558 };
555
556 $grp
557 } 559 };
560
561 $grp
558} 562}
559 563
560=item aio_copy $srcpath, $dstpath, $callback->($status) 564=item aio_copy $srcpath, $dstpath, $callback->($status)
561 565
562Try to copy the I<file> (directories not supported as either source or 566Try to copy the I<file> (directories not supported as either source or
… …
573errors are being ignored. 577errors are being ignored.
574 578
575=cut 579=cut
576 580
577sub aio_copy($$;$) { 581sub aio_copy($$;$) {
578 aio_block {
579 my ($src, $dst, $cb) = @_; 582 my ($src, $dst, $cb) = @_;
580 583
581 my $pri = aioreq_pri; 584 my $pri = aioreq_pri;
582 my $grp = aio_group $cb; 585 my $grp = aio_group $cb;
583 586
584 aioreq_pri $pri; 587 aioreq_pri $pri;
585 add $grp aio_open $src, O_RDONLY, 0, sub { 588 add $grp aio_open $src, O_RDONLY, 0, sub {
586 if (my $src_fh = $_[0]) { 589 if (my $src_fh = $_[0]) {
587 my @stat = stat $src_fh; 590 my @stat = stat $src_fh;
588 591
589 aioreq_pri $pri; 592 aioreq_pri $pri;
590 add $grp aio_open $dst, O_CREAT | O_WRONLY | O_TRUNC, 0200, sub { 593 add $grp aio_open $dst, O_CREAT | O_WRONLY | O_TRUNC, 0200, sub {
591 if (my $dst_fh = $_[0]) { 594 if (my $dst_fh = $_[0]) {
592 aioreq_pri $pri; 595 aioreq_pri $pri;
593 add $grp aio_sendfile $dst_fh, $src_fh, 0, $stat[7], sub { 596 add $grp aio_sendfile $dst_fh, $src_fh, 0, $stat[7], sub {
594 if ($_[0] == $stat[7]) { 597 if ($_[0] == $stat[7]) {
595 $grp->result (0); 598 $grp->result (0);
596 close $src_fh; 599 close $src_fh;
597 600
598 # those should not normally block. should. should. 601 # those should not normally block. should. should.
599 utime $stat[8], $stat[9], $dst; 602 utime $stat[8], $stat[9], $dst;
600 chmod $stat[2] & 07777, $dst_fh; 603 chmod $stat[2] & 07777, $dst_fh;
601 chown $stat[4], $stat[5], $dst_fh; 604 chown $stat[4], $stat[5], $dst_fh;
605
606 aioreq_pri $pri;
602 close $dst_fh; 607 add $grp aio_close $dst_fh;
603 } else { 608 } else {
604 $grp->result (-1); 609 $grp->result (-1);
605 close $src_fh; 610 close $src_fh;
606 close $dst_fh; 611 close $dst_fh;
607 612
608 aioreq $pri; 613 aioreq $pri;
609 add $grp aio_unlink $dst; 614 add $grp aio_unlink $dst;
610 }
611 }; 615 }
612 } else {
613 $grp->result (-1);
614 } 616 };
617 } else {
618 $grp->result (-1);
615 }, 619 }
616
617 } else {
618 $grp->result (-1);
619 } 620 },
621
622 } else {
623 $grp->result (-1);
620 }; 624 }
621
622 $grp
623 } 625 };
626
627 $grp
624} 628}
625 629
626=item aio_move $srcpath, $dstpath, $callback->($status) 630=item aio_move $srcpath, $dstpath, $callback->($status)
627 631
628Try to move the I<file> (directories not supported as either source or 632Try to move the I<file> (directories not supported as either source or
… …
634that is successful, unlinking the C<$srcpath>. 638that is successful, unlinking the C<$srcpath>.
635 639
636=cut 640=cut
637 641
638sub aio_move($$;$) { 642sub aio_move($$;$) {
639 aio_block {
640 my ($src, $dst, $cb) = @_; 643 my ($src, $dst, $cb) = @_;
641 644
642 my $pri = aioreq_pri; 645 my $pri = aioreq_pri;
643 my $grp = aio_group $cb; 646 my $grp = aio_group $cb;
644 647
645 aioreq_pri $pri; 648 aioreq_pri $pri;
646 add $grp aio_rename $src, $dst, sub { 649 add $grp aio_rename $src, $dst, sub {
647 if ($_[0] && $! == EXDEV) { 650 if ($_[0] && $! == EXDEV) {
648 aioreq_pri $pri; 651 aioreq_pri $pri;
649 add $grp aio_copy $src, $dst, sub { 652 add $grp aio_copy $src, $dst, sub {
650 $grp->result ($_[0]);
651
652 if (!$_[0]) {
653 aioreq_pri $pri;
654 add $grp aio_unlink $src;
655 }
656 };
657 } else {
658 $grp->result ($_[0]); 653 $grp->result ($_[0]);
654
655 if (!$_[0]) {
656 aioreq_pri $pri;
657 add $grp aio_unlink $src;
658 }
659 } 659 };
660 } else {
661 $grp->result ($_[0]);
660 }; 662 }
661
662 $grp
663 } 663 };
664
665 $grp
664} 666}
665 667
666=item aio_scandir $path, $maxreq, $callback->($dirs, $nondirs) 668=item aio_scandir $path, $maxreq, $callback->($dirs, $nondirs)
667 669
668Scans a directory (similar to C<aio_readdir>) but additionally tries to 670Scans a directory (similar to C<aio_readdir>) but additionally tries to
… …
716directory counting heuristic. 718directory counting heuristic.
717 719
718=cut 720=cut
719 721
720sub aio_scandir($$;$) { 722sub aio_scandir($$;$) {
721 aio_block {
722 my ($path, $maxreq, $cb) = @_; 723 my ($path, $maxreq, $cb) = @_;
723 724
724 my $pri = aioreq_pri; 725 my $pri = aioreq_pri;
725 726
726 my $grp = aio_group $cb; 727 my $grp = aio_group $cb;
727 728
728 $maxreq = 4 if $maxreq <= 0; 729 $maxreq = 4 if $maxreq <= 0;
729 730
730 # stat once 731 # stat once
732 aioreq_pri $pri;
733 add $grp aio_stat $path, sub {
734 return $grp->result () if $_[0];
735 my $now = time;
736 my $hash1 = join ":", (stat _)[0,1,3,7,9];
737
738 # read the directory entries
731 aioreq_pri $pri; 739 aioreq_pri $pri;
732 add $grp aio_stat $path, sub { 740 add $grp aio_readdir $path, sub {
741 my $entries = shift
733 return $grp->result () if $_[0]; 742 or return $grp->result ();
734 my $now = time;
735 my $hash1 = join ":", (stat _)[0,1,3,7,9];
736 743
737 # read the directory entries 744 # stat the dir another time
738 aioreq_pri $pri; 745 aioreq_pri $pri;
739 add $grp aio_readdir $path, sub {
740 my $entries = shift
741 or return $grp->result ();
742
743 # stat the dir another time
744 aioreq_pri $pri;
745 add $grp aio_stat $path, sub { 746 add $grp aio_stat $path, sub {
746 my $hash2 = join ":", (stat _)[0,1,3,7,9]; 747 my $hash2 = join ":", (stat _)[0,1,3,7,9];
747 748
748 my $ndirs; 749 my $ndirs;
749 750
750 # take the slow route if anything looks fishy 751 # take the slow route if anything looks fishy
751 if ($hash1 ne $hash2 or (stat _)[9] == $now) { 752 if ($hash1 ne $hash2 or (stat _)[9] == $now) {
752 $ndirs = -1; 753 $ndirs = -1;
753 } else { 754 } else {
754 # if nlink == 2, we are finished 755 # if nlink == 2, we are finished
755 # on non-posix-fs's, we rely on nlink < 2 756 # on non-posix-fs's, we rely on nlink < 2
756 $ndirs = (stat _)[3] - 2 757 $ndirs = (stat _)[3] - 2
757 or return $grp->result ([], $entries); 758 or return $grp->result ([], $entries);
758 } 759 }
759 760
760 # sort into likely dirs and likely nondirs 761 # sort into likely dirs and likely nondirs
761 # dirs == files without ".", short entries first 762 # dirs == files without ".", short entries first
762 $entries = [map $_->[0], 763 $entries = [map $_->[0],
763 sort { $b->[1] cmp $a->[1] } 764 sort { $b->[1] cmp $a->[1] }
764 map [$_, sprintf "%s%04d", (/.\./ ? "1" : "0"), length], 765 map [$_, sprintf "%s%04d", (/.\./ ? "1" : "0"), length],
765 @$entries]; 766 @$entries];
766 767
767 my (@dirs, @nondirs); 768 my (@dirs, @nondirs);
768 769
769 my $statgrp = add $grp aio_group sub { 770 my $statgrp = add $grp aio_group sub {
770 $grp->result (\@dirs, \@nondirs); 771 $grp->result (\@dirs, \@nondirs);
771 }; 772 };
772 773
773 limit $statgrp $maxreq; 774 limit $statgrp $maxreq;
774 feed $statgrp sub { 775 feed $statgrp sub {
775 return unless @$entries; 776 return unless @$entries;
776 my $entry = pop @$entries; 777 my $entry = pop @$entries;
777 778
778 aioreq_pri $pri; 779 aioreq_pri $pri;
779 add $statgrp aio_stat "$path/$entry/.", sub { 780 add $statgrp aio_stat "$path/$entry/.", sub {
780 if ($_[0] < 0) { 781 if ($_[0] < 0) {
781 push @nondirs, $entry; 782 push @nondirs, $entry;
782 } else { 783 } else {
783 # need to check for real directory 784 # need to check for real directory
784 aioreq_pri $pri; 785 aioreq_pri $pri;
785 add $statgrp aio_lstat "$path/$entry", sub { 786 add $statgrp aio_lstat "$path/$entry", sub {
786 if (-d _) { 787 if (-d _) {
787 push @dirs, $entry; 788 push @dirs, $entry;
788 789
789 unless (--$ndirs) { 790 unless (--$ndirs) {
790 push @nondirs, @$entries; 791 push @nondirs, @$entries;
791 feed $statgrp; 792 feed $statgrp;
792 }
793 } else {
794 push @nondirs, $entry;
795 } 793 }
794 } else {
795 push @nondirs, $entry;
796 } 796 }
797 } 797 }
798 }; 798 }
799 }; 799 };
800 }; 800 };
801 }; 801 };
802 }; 802 };
803
804 $grp
805 } 803 };
804
805 $grp
806} 806}
807 807
808=item aio_rmtree $path, $callback->($status) 808=item aio_rmtree $path, $callback->($status)
809 809
810Delete a directory tree starting (and including) C<$path>, return the 810Delete a directory tree starting (and including) C<$path>, return the
… …
814 814
815=cut 815=cut
816 816
817sub aio_rmtree; 817sub aio_rmtree;
818sub aio_rmtree($;$) { 818sub aio_rmtree($;$) {
819 aio_block {
820 my ($path, $cb) = @_; 819 my ($path, $cb) = @_;
821 820
822 my $pri = aioreq_pri; 821 my $pri = aioreq_pri;
823 my $grp = aio_group $cb; 822 my $grp = aio_group $cb;
824 823
825 aioreq_pri $pri; 824 aioreq_pri $pri;
826 add $grp aio_scandir $path, 0, sub { 825 add $grp aio_scandir $path, 0, sub {
827 my ($dirs, $nondirs) = @_; 826 my ($dirs, $nondirs) = @_;
828 827
829 my $dirgrp = aio_group sub { 828 my $dirgrp = aio_group sub {
830 add $grp aio_rmdir $path, sub { 829 add $grp aio_rmdir $path, sub {
831 $grp->result ($_[0]); 830 $grp->result ($_[0]);
832 };
833 }; 831 };
834
835 (aioreq_pri $pri), add $dirgrp aio_rmtree "$path/$_" for @$dirs;
836 (aioreq_pri $pri), add $dirgrp aio_unlink "$path/$_" for @$nondirs;
837
838 add $grp $dirgrp;
839 }; 832 };
840 833
841 $grp 834 (aioreq_pri $pri), add $dirgrp aio_rmtree "$path/$_" for @$dirs;
835 (aioreq_pri $pri), add $dirgrp aio_unlink "$path/$_" for @$nondirs;
836
837 add $grp $dirgrp;
842 } 838 };
839
840 $grp
843} 841}
842
843=item aio_sync $callback->($status)
844
845Asynchronously call sync and call the callback when finished.
844 846
845=item aio_fsync $fh, $callback->($status) 847=item aio_fsync $fh, $callback->($status)
846 848
847Asynchronously call fsync on the given filehandle and call the callback 849Asynchronously call fsync on the given filehandle and call the callback
848with the fsync result code. 850with the fsync result code.
… …
852Asynchronously call fdatasync on the given filehandle and call the 854Asynchronously call fdatasync on the given filehandle and call the
853callback with the fdatasync result code. 855callback with the fdatasync result code.
854 856
855If this call isn't available because your OS lacks it or it couldn't be 857If this call isn't available because your OS lacks it or it couldn't be
856detected, it will be emulated by calling C<fsync> instead. 858detected, it will be emulated by calling C<fsync> instead.
859
860=item aio_pathsync $path, $callback->($status)
861
862This request tries to open, fsync and close the given path. This is a
863composite request intended tosync directories after directory operations
864(E.g. rename). This might not work on all operating systems or have any
865specific effect, but usually it makes sure that directory changes get
866written to disc. It works for anything that can be opened for read-only,
867not just directories.
868
869Passes C<0> when everything went ok, and C<-1> on error.
870
871=cut
872
873sub aio_pathsync($;$) {
874 my ($path, $cb) = @_;
875
876 my $pri = aioreq_pri;
877 my $grp = aio_group $cb;
878
879 aioreq_pri $pri;
880 add $grp aio_open $path, O_RDONLY, 0, sub {
881 my ($fh) = @_;
882 if ($fh) {
883 aioreq_pri $pri;
884 add $grp aio_fsync $fh, sub {
885 $grp->result ($_[0]);
886
887 aioreq_pri $pri;
888 add $grp aio_close $fh;
889 };
890 } else {
891 $grp->result (-1);
892 }
893 };
894
895 $grp
896}
857 897
858=item aio_group $callback->(...) 898=item aio_group $callback->(...)
859 899
860This is a very special aio request: Instead of doing something, it is a 900This is a very special aio request: Instead of doing something, it is a
861container for other aio requests, which is useful if you want to bundle 901container for other aio requests, which is useful if you want to bundle
… …
998itself. Useful when you queued a lot of events but got a result early. 1038itself. Useful when you queued a lot of events but got a result early.
999 1039
1000=item $grp->result (...) 1040=item $grp->result (...)
1001 1041
1002Set the result value(s) that will be passed to the group callback when all 1042Set the result value(s) that will be passed to the group callback when all
1003subrequests have finished and set thre groups errno to the current value 1043subrequests have finished and set the groups errno to the current value
1004of errno (just like calling C<errno> without an error number). By default, 1044of errno (just like calling C<errno> without an error number). By default,
1005no argument will be passed and errno is zero. 1045no argument will be passed and errno is zero.
1006 1046
1007=item $grp->errno ([$errno]) 1047=item $grp->errno ([$errno])
1008 1048
… …
1206 1246
1207The default is probably ok in most situations, especially if thread 1247The default is probably ok in most situations, especially if thread
1208creation is fast. If thread creation is very slow on your system you might 1248creation is fast. If thread creation is very slow on your system you might
1209want to use larger values. 1249want to use larger values.
1210 1250
1211=item $oldmaxreqs = IO::AIO::max_outstanding $maxreqs 1251=item IO::AIO::max_outstanding $maxreqs
1212 1252
1213This is a very bad function to use in interactive programs because it 1253This is a very bad function to use in interactive programs because it
1214blocks, and a bad way to reduce concurrency because it is inexact: Better 1254blocks, and a bad way to reduce concurrency because it is inexact: Better
1215use an C<aio_group> together with a feed callback. 1255use an C<aio_group> together with a feed callback.
1216 1256
… …
1221 1261
1222The default value is very large, so there is no practical limit on the 1262The default value is very large, so there is no practical limit on the
1223number of outstanding requests. 1263number of outstanding requests.
1224 1264
1225You can still queue as many requests as you want. Therefore, 1265You can still queue as many requests as you want. Therefore,
1226C<max_oustsanding> is mainly useful in simple scripts (with low values) or 1266C<max_outstanding> is mainly useful in simple scripts (with low values) or
1227as a stop gap to shield against fatal memory overflow (with large values). 1267as a stop gap to shield against fatal memory overflow (with large values).
1228 1268
1229=back 1269=back
1230 1270
1231=head3 STATISTICAL INFORMATION 1271=head3 STATISTICAL INFORMATION
… …
1301 1341
1302Known bugs will be fixed in the next release. 1342Known bugs will be fixed in the next release.
1303 1343
1304=head1 SEE ALSO 1344=head1 SEE ALSO
1305 1345
1306L<Coro::AIO>. 1346L<AnyEvent::AIO> for easy integration into event loops, L<Coro::AIO> for a
1347more natural syntax.
1307 1348
1308=head1 AUTHOR 1349=head1 AUTHOR
1309 1350
1310 Marc Lehmann <schmorp@schmorp.de> 1351 Marc Lehmann <schmorp@schmorp.de>
1311 http://home.schmorp.de/ 1352 http://home.schmorp.de/

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines