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

Comparing IO-AIO/AIO.xs (file contents):
Revision 1.74 by root, Wed Oct 25 17:57:30 2006 UTC vs.
Revision 1.84 by root, Sat Oct 28 01:24:19 2006 UTC

1/* solaris */ 1/* solaris */
2#define _POSIX_PTHREAD_SEMANTICS 1 2#define _POSIX_PTHREAD_SEMANTICS 1
3 3
4#if __linux 4#if __linux && !defined(_GNU_SOURCE)
5# define _GNU_SOURCE 5# define _GNU_SOURCE
6#endif 6#endif
7 7
8/* just in case */
8#define _REENTRANT 1 9#define _REENTRANT 1
9 10
10#include <errno.h> 11#include <errno.h>
11 12
12#include "EXTERN.h" 13#include "EXTERN.h"
42# else 43# else
43# error sendfile support requested but not available 44# error sendfile support requested but not available
44# endif 45# endif
45#endif 46#endif
46 47
48/* number of seconds after which idle threads exit */
49#define IDLE_TIMEOUT 10
50
47/* used for struct dirent, AIX doesn't provide it */ 51/* used for struct dirent, AIX doesn't provide it */
48#ifndef NAME_MAX 52#ifndef NAME_MAX
49# define NAME_MAX 4096 53# define NAME_MAX 4096
54#endif
55
56#ifndef PTHREAD_STACK_MIN
57/* care for broken platforms, e.g. windows */
58# define PTHREAD_STACK_MIN 16384
50#endif 59#endif
51 60
52#if __ia64 61#if __ia64
53# define STACKSIZE 65536 62# define STACKSIZE 65536
54#elif __i386 || __x86_64 /* 16k is unreasonably high :( */ 63#elif __i386 || __x86_64 /* 16k is unreasonably high :( */
55# define STACKSIZE PTHREAD_STACK_MIN 64# define STACKSIZE PTHREAD_STACK_MIN
56#else 65#else
57# define STACKSIZE 16384 66# define STACKSIZE 16384
67#endif
68
69/* wether word reads are potentially non-atomic.
70 * this is conservatice, likely most arches this runs
71 * on have atomic word read/writes.
72 */
73#ifndef WORDREAD_UNSAFE
74# if __i386 || __x86_64
75# define WORDREAD_UNSAFE 0
76# else
77# define WORDREAD_UNSAFE 1
78# endif
58#endif 79#endif
59 80
60/* buffer size for various temporary buffers */ 81/* buffer size for various temporary buffers */
61#define AIO_BUFSIZE 65536 82#define AIO_BUFSIZE 65536
62 83
74 REQ_READ, REQ_WRITE, REQ_READAHEAD, 95 REQ_READ, REQ_WRITE, REQ_READAHEAD,
75 REQ_SENDFILE, 96 REQ_SENDFILE,
76 REQ_STAT, REQ_LSTAT, REQ_FSTAT, 97 REQ_STAT, REQ_LSTAT, REQ_FSTAT,
77 REQ_FSYNC, REQ_FDATASYNC, 98 REQ_FSYNC, REQ_FDATASYNC,
78 REQ_UNLINK, REQ_RMDIR, REQ_RENAME, 99 REQ_UNLINK, REQ_RMDIR, REQ_RENAME,
79 REQ_READDIR, 100 REQ_MKNOD, REQ_READDIR,
80 REQ_LINK, REQ_SYMLINK, 101 REQ_LINK, REQ_SYMLINK,
81 REQ_GROUP, REQ_NOP, 102 REQ_GROUP, REQ_NOP,
82 REQ_BUSY, 103 REQ_BUSY,
83}; 104};
84 105
126 NUM_PRI = PRI_MAX + PRI_BIAS + 1, 147 NUM_PRI = PRI_MAX + PRI_BIAS + 1,
127}; 148};
128 149
129static int next_pri = DEFAULT_PRI + PRI_BIAS; 150static int next_pri = DEFAULT_PRI + PRI_BIAS;
130 151
131static int started, wanted; 152static unsigned int started, idle, wanted;
132static volatile int nreqs;
133static int respipe [2];
134 153
135#if __linux && defined (PTHREAD_ADAPTIVE_MUTEX_INITIALIZER_NP) 154#if __linux && defined (PTHREAD_ADAPTIVE_MUTEX_INITIALIZER_NP)
136# define AIO_MUTEX_INIT PTHREAD_ADAPTIVE_MUTEX_INITIALIZER_NP 155# define AIO_MUTEX_INIT PTHREAD_ADAPTIVE_MUTEX_INITIALIZER_NP
137#else 156#else
138# define AIO_MUTEX_INIT PTHREAD_MUTEX_INITIALIZER 157# define AIO_MUTEX_INIT PTHREAD_MUTEX_INITIALIZER
139#endif 158#endif
140 159
141#define LOCK(mutex) pthread_mutex_lock (&(mutex)) 160#define LOCK(mutex) pthread_mutex_lock (&(mutex))
142#define UNLOCK(mutex) pthread_mutex_unlock (&(mutex)) 161#define UNLOCK(mutex) pthread_mutex_unlock (&(mutex))
143 162
144/* worker threasd management */ 163/* worker threads management */
145static pthread_mutex_t wrklock = AIO_MUTEX_INIT; 164static pthread_mutex_t wrklock = AIO_MUTEX_INIT;
146 165
147typedef struct worker { 166typedef struct worker {
148 /* locked by wrklock */ 167 /* locked by wrklock */
149 struct worker *prev, *next; 168 struct worker *prev, *next;
179 wrk->prev->next = wrk->next; 198 wrk->prev->next = wrk->next;
180 199
181 free (wrk); 200 free (wrk);
182} 201}
183 202
203static volatile unsigned int nreqs, nready, npending;
204static volatile unsigned int max_idle = 4;
205static volatile unsigned int max_outstanding = 0xffffffff;
206static int respipe [2];
207
184static pthread_mutex_t reslock = AIO_MUTEX_INIT; 208static pthread_mutex_t reslock = AIO_MUTEX_INIT;
185static pthread_mutex_t reqlock = AIO_MUTEX_INIT; 209static pthread_mutex_t reqlock = AIO_MUTEX_INIT;
186static pthread_cond_t reqwait = PTHREAD_COND_INITIALIZER; 210static pthread_cond_t reqwait = PTHREAD_COND_INITIALIZER;
211
212#if WORDREAD_UNSAFE
213
214static unsigned int get_nready ()
215{
216 unsigned int retval;
217
218 LOCK (reqlock);
219 retval = nready;
220 UNLOCK (reqlock);
221
222 return retval;
223}
224
225static unsigned int get_npending ()
226{
227 unsigned int retval;
228
229 LOCK (reslock);
230 retval = npending;
231 UNLOCK (reslock);
232
233 return retval;
234}
235
236#else
237
238# define get_nready() nready
239# define get_npending() npending
240
241#endif
187 242
188/* 243/*
189 * a somewhat faster data structure might be nice, but 244 * a somewhat faster data structure might be nice, but
190 * with 8 priorities this actually needs <20 insns 245 * with 8 priorities this actually needs <20 insns
191 * per shift, the most expensive operation. 246 * per shift, the most expensive operation.
237 } 292 }
238 293
239 abort (); 294 abort ();
240} 295}
241 296
242static int poll_cb (); 297static int poll_cb (int max);
243static void req_invoke (aio_req req); 298static void req_invoke (aio_req req);
244static void req_free (aio_req req); 299static void req_free (aio_req req);
245static void req_cancel (aio_req req); 300static void req_cancel (aio_req req);
246 301
247/* must be called at most once */ 302/* must be called at most once */
312 req_invoke (grp); 367 req_invoke (grp);
313 req_free (grp); 368 req_free (grp);
314 } 369 }
315} 370}
316 371
317static void poll_wait ()
318{
319 fd_set rfd;
320
321 while (nreqs)
322 {
323 int size;
324#if !(__i386 || __x86_64) /* safe without sempahore on these archs */
325 LOCK (reslock);
326#endif
327 size = res_queue.size;
328#if !(__i386 || __x86_64) /* safe without sempahore on these archs */
329 UNLOCK (reslock);
330#endif
331
332 if (size)
333 return;
334
335 FD_ZERO(&rfd);
336 FD_SET(respipe [0], &rfd);
337
338 select (respipe [0] + 1, &rfd, 0, 0, 0);
339 }
340}
341
342static void req_invoke (aio_req req) 372static void req_invoke (aio_req req)
343{ 373{
344 dSP; 374 dSP;
345 375
346 if (!(req->flags & FLAG_CANCELLED) && SvOK (req->callback)) 376 if (!(req->flags & FLAG_CANCELLED) && SvOK (req->callback))
347 { 377 {
348 errno = req->errorno;
349
350 ENTER; 378 ENTER;
351 SAVETMPS; 379 SAVETMPS;
352 PUSHMARK (SP); 380 PUSHMARK (SP);
353 EXTEND (SP, 1); 381 EXTEND (SP, 1);
354 382
419 default: 447 default:
420 PUSHs (sv_2mortal (newSViv (req->result))); 448 PUSHs (sv_2mortal (newSViv (req->result)));
421 break; 449 break;
422 } 450 }
423 451
452 errno = req->errorno;
424 453
425 PUTBACK; 454 PUTBACK;
426 call_sv (req->callback, G_VOID | G_EVAL); 455 call_sv (req->callback, G_VOID | G_EVAL);
427 SPAGAIN; 456 SPAGAIN;
428 457
490 req->flags |= FLAG_CANCELLED; 519 req->flags |= FLAG_CANCELLED;
491 520
492 req_cancel_subs (req); 521 req_cancel_subs (req);
493} 522}
494 523
524static void *aio_proc(void *arg);
525
526static void start_thread (void)
527{
528 sigset_t fullsigset, oldsigset;
529 pthread_attr_t attr;
530
531 worker *wrk = calloc (1, sizeof (worker));
532
533 if (!wrk)
534 croak ("unable to allocate worker thread data");
535
536 pthread_attr_init (&attr);
537 pthread_attr_setstacksize (&attr, STACKSIZE);
538 pthread_attr_setdetachstate (&attr, PTHREAD_CREATE_DETACHED);
539#ifdef PTHREAD_SCOPE_PROCESS
540 pthread_attr_setscope (&attr, PTHREAD_SCOPE_PROCESS);
541#endif
542
543 sigfillset (&fullsigset);
544
545 LOCK (wrklock);
546 sigprocmask (SIG_SETMASK, &fullsigset, &oldsigset);
547
548 if (pthread_create (&wrk->tid, &attr, aio_proc, (void *)wrk) == 0)
549 {
550 wrk->prev = &wrk_first;
551 wrk->next = wrk_first.next;
552 wrk_first.next->prev = wrk;
553 wrk_first.next = wrk;
554 ++started;
555 }
556 else
557 free (wrk);
558
559 sigprocmask (SIG_SETMASK, &oldsigset, 0);
560 UNLOCK (wrklock);
561}
562
563static void maybe_start_thread ()
564{
565 if (started >= wanted)
566 return;
567
568 /* todo: maybe use idle here, but might be less exact */
569 if ((int)nready <= (int)started - (int)(nreqs - get_nready () - get_npending ()))
570 return;
571
572 start_thread ();
573}
574
575static void req_send (aio_req req)
576{
577 ++nreqs;
578
579 LOCK (reqlock);
580 ++nready;
581 reqq_push (&req_queue, req);
582 pthread_cond_signal (&reqwait);
583 UNLOCK (reqlock);
584
585 maybe_start_thread ();
586}
587
588static void end_thread (void)
589{
590 aio_req req;
591
592 Newz (0, req, 1, aio_cb);
593
594 req->type = REQ_QUIT;
595 req->pri = PRI_MAX + PRI_BIAS;
596
597 LOCK (reqlock);
598 reqq_push (&req_queue, req);
599 pthread_cond_signal (&reqwait);
600 UNLOCK (reqlock);
601
602 LOCK (wrklock);
603 --started;
604 UNLOCK (wrklock);
605}
606
607static void min_parallel (int nthreads)
608{
609 if (wanted < nthreads)
610 wanted = nthreads;
611}
612
613static void max_parallel (int nthreads)
614{
615 if (wanted > nthreads)
616 wanted = nthreads;
617
618 while (started > wanted)
619 end_thread ();
620}
621
622static void poll_wait ()
623{
624 fd_set rfd;
625
626 while (nreqs)
627 {
628 int size;
629 if (WORDREAD_UNSAFE) LOCK (reslock);
630 size = res_queue.size;
631 if (WORDREAD_UNSAFE) UNLOCK (reslock);
632
633 if (size)
634 return;
635
636 maybe_start_thread ();
637
638 FD_ZERO(&rfd);
639 FD_SET(respipe [0], &rfd);
640
641 select (respipe [0] + 1, &rfd, 0, 0, 0);
642 }
643}
644
495static int poll_cb () 645static int poll_cb (int max)
496{ 646{
497 dSP; 647 dSP;
498 int count = 0; 648 int count = 0;
499 int do_croak = 0; 649 int do_croak = 0;
500 aio_req req; 650 aio_req req;
501 651
502 for (;;) 652 for (;;)
503 { 653 {
504 LOCK (reslock); 654 while (max <= 0 || count < max)
505 req = reqq_shift (&res_queue);
506
507 if (req)
508 { 655 {
656 maybe_start_thread ();
657
658 LOCK (reslock);
659 req = reqq_shift (&res_queue);
660
509 if (!res_queue.size) 661 if (req)
510 { 662 {
663 --npending;
664
665 if (!res_queue.size)
666 {
511 /* read any signals sent by the worker threads */ 667 /* read any signals sent by the worker threads */
512 char buf [32]; 668 char buf [32];
513 while (read (respipe [0], buf, 32) == 32) 669 while (read (respipe [0], buf, 32) == 32)
670 ;
514 ; 671 }
515 } 672 }
673
674 UNLOCK (reslock);
675
676 if (!req)
677 break;
678
679 --nreqs;
680
681 if (req->type == REQ_GROUP && req->length)
682 {
683 req->fd = 1; /* mark request as delayed */
684 continue;
685 }
686 else
687 {
688 if (req->type == REQ_READ)
689 SvCUR_set (req->data, req->dataoffset + (req->result > 0 ? req->result : 0));
690
691 if (req->data2ptr && (req->type == REQ_READ || req->type == REQ_WRITE))
692 SvREADONLY_off (req->data);
693
694 if (req->statdata)
695 {
696 PL_laststype = req->type == REQ_LSTAT ? OP_LSTAT : OP_STAT;
697 PL_laststatval = req->result;
698 PL_statcache = *(req->statdata);
699 }
700
701 req_invoke (req);
702
703 count++;
704 }
705
706 req_free (req);
516 } 707 }
517 708
518 UNLOCK (reslock); 709 if (nreqs <= max_outstanding)
519
520 if (!req)
521 break; 710 break;
522 711
523 --nreqs; 712 poll_wait ();
524 713
525 if (req->type == REQ_QUIT) 714 max = 0;
526 started--;
527 else if (req->type == REQ_GROUP && req->length)
528 {
529 req->fd = 1; /* mark request as delayed */
530 continue;
531 }
532 else
533 {
534 if (req->type == REQ_READ)
535 SvCUR_set (req->data, req->dataoffset + (req->result > 0 ? req->result : 0));
536
537 if (req->data2ptr && (req->type == REQ_READ || req->type == REQ_WRITE))
538 SvREADONLY_off (req->data);
539
540 if (req->statdata)
541 {
542 PL_laststype = req->type == REQ_LSTAT ? OP_LSTAT : OP_STAT;
543 PL_laststatval = req->result;
544 PL_statcache = *(req->statdata);
545 }
546
547 req_invoke (req);
548
549 count++;
550 }
551
552 req_free (req);
553 } 715 }
554 716
555 return count; 717 return count;
556}
557
558static void *aio_proc(void *arg);
559
560static void start_thread (void)
561{
562 sigset_t fullsigset, oldsigset;
563 pthread_attr_t attr;
564
565 worker *wrk = calloc (1, sizeof (worker));
566
567 if (!wrk)
568 croak ("unable to allocate worker thread data");
569
570 pthread_attr_init (&attr);
571 pthread_attr_setstacksize (&attr, STACKSIZE);
572 pthread_attr_setdetachstate (&attr, PTHREAD_CREATE_DETACHED);
573
574 sigfillset (&fullsigset);
575
576 LOCK (wrklock);
577 sigprocmask (SIG_SETMASK, &fullsigset, &oldsigset);
578
579 if (pthread_create (&wrk->tid, &attr, aio_proc, (void *)wrk) == 0)
580 {
581 wrk->prev = &wrk_first;
582 wrk->next = wrk_first.next;
583 wrk_first.next->prev = wrk;
584 wrk_first.next = wrk;
585 started++;
586 }
587 else
588 free (wrk);
589
590 sigprocmask (SIG_SETMASK, &oldsigset, 0);
591 UNLOCK (wrklock);
592}
593
594static void req_send (aio_req req)
595{
596 while (started < wanted && nreqs >= started)
597 start_thread ();
598
599 ++nreqs;
600
601 LOCK (reqlock);
602 reqq_push (&req_queue, req);
603 pthread_cond_signal (&reqwait);
604 UNLOCK (reqlock);
605}
606
607static void end_thread (void)
608{
609 aio_req req;
610
611 Newz (0, req, 1, aio_cb);
612
613 req->type = REQ_QUIT;
614 req->pri = PRI_MAX + PRI_BIAS;
615
616 req_send (req);
617}
618
619static void min_parallel (int nthreads)
620{
621 if (wanted < nthreads)
622 wanted = nthreads;
623}
624
625static void max_parallel (int nthreads)
626{
627 int cur = started;
628
629 if (wanted > nthreads)
630 wanted = nthreads;
631
632 while (cur > wanted)
633 {
634 end_thread ();
635 cur--;
636 }
637
638 while (started > wanted)
639 {
640 poll_wait ();
641 poll_cb ();
642 }
643} 718}
644 719
645static void create_pipe () 720static void create_pipe ()
646{ 721{
647 if (pipe (respipe)) 722 if (pipe (respipe))
702#if !HAVE_FDATASYNC 777#if !HAVE_FDATASYNC
703# define fdatasync fsync 778# define fdatasync fsync
704#endif 779#endif
705 780
706#if !HAVE_READAHEAD 781#if !HAVE_READAHEAD
707# define readahead aio_readahead 782# define readahead(fd,offset,count) aio_readahead (fd, offset, count, self)
708 783
709static ssize_t readahead (int fd, off_t offset, size_t count) 784static ssize_t aio_readahead (int fd, off_t offset, size_t count, worker *self)
710{ 785{
711 dBUF; 786 dBUF;
712 787
713 while (count > 0) 788 while (count > 0)
714 { 789 {
719 count -= len; 794 count -= len;
720 } 795 }
721 796
722 errno = 0; 797 errno = 0;
723} 798}
799
724#endif 800#endif
725 801
726#if !HAVE_READDIR_R 802#if !HAVE_READDIR_R
727# define readdir_r aio_readdir_r 803# define readdir_r aio_readdir_r
728 804
773 { 849 {
774 off_t sbytes; 850 off_t sbytes;
775 res = sendfile (ifd, ofd, offset, count, 0, &sbytes, 0); 851 res = sendfile (ifd, ofd, offset, count, 0, &sbytes, 0);
776 852
777 if (res < 0 && sbytes) 853 if (res < 0 && sbytes)
778 /* maybe only on EAGAIN only: as usual, the manpage leaves you guessing */ 854 /* maybe only on EAGAIN: as usual, the manpage leaves you guessing */
779 res = sbytes; 855 res = sbytes;
780 } 856 }
781 857
782# elif __hpux 858# elif __hpux
783 res = sendfile (ofd, ifd, offset, count, 0, 0); 859 res = sendfile (ofd, ifd, offset, count, 0, 0);
863 int errorno; 939 int errorno;
864 940
865 LOCK (wrklock); 941 LOCK (wrklock);
866 self->dirp = dirp = opendir (req->dataptr); 942 self->dirp = dirp = opendir (req->dataptr);
867 self->dbuf = u = malloc (sizeof (*u)); 943 self->dbuf = u = malloc (sizeof (*u));
944 req->data2ptr = names = malloc (memlen);
868 UNLOCK (wrklock); 945 UNLOCK (wrklock);
869
870 req->data2ptr = names = malloc (memlen);
871 946
872 if (dirp && u && names) 947 if (dirp && u && names)
873 for (;;) 948 for (;;)
874 { 949 {
875 errno = 0; 950 errno = 0;
911/*****************************************************************************/ 986/*****************************************************************************/
912 987
913static void *aio_proc (void *thr_arg) 988static void *aio_proc (void *thr_arg)
914{ 989{
915 aio_req req; 990 aio_req req;
916 int type; 991 struct timespec ts;
917 worker *self = (worker *)thr_arg; 992 worker *self = (worker *)thr_arg;
918 993
919 do 994 /* try to distribute timeouts somewhat evenly */
995 ts.tv_nsec = (((unsigned long)self + (unsigned long)ts.tv_sec) & 1023UL)
996 * (1000000000UL / 1024UL);
997
998 for (;;)
920 { 999 {
1000 ts.tv_sec = time (0) + IDLE_TIMEOUT;
1001
921 LOCK (reqlock); 1002 LOCK (reqlock);
922 1003
923 for (;;) 1004 for (;;)
924 { 1005 {
925 self->req = req = reqq_shift (&req_queue); 1006 self->req = req = reqq_shift (&req_queue);
926 1007
927 if (req) 1008 if (req)
928 break; 1009 break;
929 1010
1011 ++idle;
1012
1013 if (pthread_cond_timedwait (&reqwait, &reqlock, &ts)
1014 == ETIMEDOUT)
1015 {
1016 if (idle > max_idle)
1017 {
1018 --idle;
1019 UNLOCK (reqlock);
1020 LOCK (wrklock);
1021 --started;
1022 UNLOCK (wrklock);
1023 goto quit;
1024 }
1025
1026 /* we are allowed to idle, so do so without any timeout */
930 pthread_cond_wait (&reqwait, &reqlock); 1027 pthread_cond_wait (&reqwait, &reqlock);
1028 ts.tv_sec = time (0) + IDLE_TIMEOUT;
1029 }
1030
1031 --idle;
931 } 1032 }
1033
1034 --nready;
932 1035
933 UNLOCK (reqlock); 1036 UNLOCK (reqlock);
934 1037
935 errno = 0; /* strictly unnecessary */ 1038 errno = 0; /* strictly unnecessary */
936 type = req->type; /* remember type for QUIT check */
937 1039
938 if (!(req->flags & FLAG_CANCELLED)) 1040 if (!(req->flags & FLAG_CANCELLED))
939 switch (type) 1041 switch (req->type)
940 { 1042 {
941 case REQ_READ: req->result = pread (req->fd, req->dataptr, req->length, req->offset); break; 1043 case REQ_READ: req->result = pread (req->fd, req->dataptr, req->length, req->offset); break;
942 case REQ_WRITE: req->result = pwrite (req->fd, req->dataptr, req->length, req->offset); break; 1044 case REQ_WRITE: req->result = pwrite (req->fd, req->dataptr, req->length, req->offset); break;
943 1045
944 case REQ_READAHEAD: req->result = readahead (req->fd, req->offset, req->length); break; 1046 case REQ_READAHEAD: req->result = readahead (req->fd, req->offset, req->length); break;
953 case REQ_UNLINK: req->result = unlink (req->dataptr); break; 1055 case REQ_UNLINK: req->result = unlink (req->dataptr); break;
954 case REQ_RMDIR: req->result = rmdir (req->dataptr); break; 1056 case REQ_RMDIR: req->result = rmdir (req->dataptr); break;
955 case REQ_RENAME: req->result = rename (req->data2ptr, req->dataptr); break; 1057 case REQ_RENAME: req->result = rename (req->data2ptr, req->dataptr); break;
956 case REQ_LINK: req->result = link (req->data2ptr, req->dataptr); break; 1058 case REQ_LINK: req->result = link (req->data2ptr, req->dataptr); break;
957 case REQ_SYMLINK: req->result = symlink (req->data2ptr, req->dataptr); break; 1059 case REQ_SYMLINK: req->result = symlink (req->data2ptr, req->dataptr); break;
1060 case REQ_MKNOD: req->result = mknod (req->data2ptr, req->mode, (dev_t)req->offset); break;
958 1061
959 case REQ_FDATASYNC: req->result = fdatasync (req->fd); break; 1062 case REQ_FDATASYNC: req->result = fdatasync (req->fd); break;
960 case REQ_FSYNC: req->result = fsync (req->fd); break; 1063 case REQ_FSYNC: req->result = fsync (req->fd); break;
961 case REQ_READDIR: scandir_ (req, self); break; 1064 case REQ_READDIR: scandir_ (req, self); break;
962 1065
970 req->result = select (0, 0, 0, 0, &tv); 1073 req->result = select (0, 0, 0, 0, &tv);
971 } 1074 }
972 1075
973 case REQ_GROUP: 1076 case REQ_GROUP:
974 case REQ_NOP: 1077 case REQ_NOP:
1078 break;
1079
975 case REQ_QUIT: 1080 case REQ_QUIT:
976 break; 1081 goto quit;
977 1082
978 default: 1083 default:
979 req->result = ENOSYS; 1084 req->result = ENOSYS;
980 break; 1085 break;
981 } 1086 }
982 1087
983 req->errorno = errno; 1088 req->errorno = errno;
984 1089
985 LOCK (reslock); 1090 LOCK (reslock);
986 1091
1092 ++npending;
1093
987 if (!reqq_push (&res_queue, req)) 1094 if (!reqq_push (&res_queue, req))
988 /* write a dummy byte to the pipe so fh becomes ready */ 1095 /* write a dummy byte to the pipe so fh becomes ready */
989 write (respipe [1], &respipe, 1); 1096 write (respipe [1], &respipe, 1);
990 1097
991 self->req = 0; 1098 self->req = 0;
992 worker_clear (self); 1099 worker_clear (self);
993 1100
994 UNLOCK (reslock); 1101 UNLOCK (reslock);
995 } 1102 }
996 while (type != REQ_QUIT);
997 1103
1104quit:
998 LOCK (wrklock); 1105 LOCK (wrklock);
999 worker_free (self); 1106 worker_free (self);
1000 UNLOCK (wrklock); 1107 UNLOCK (wrklock);
1001 1108
1002 return 0; 1109 return 0;
1049 1156
1050 worker_clear (wrk); 1157 worker_clear (wrk);
1051 worker_free (wrk); 1158 worker_free (wrk);
1052 } 1159 }
1053 1160
1054 started = 0; 1161 started = 0;
1162 idle = 0;
1055 nreqs = 0; 1163 nreqs = 0;
1164 nready = 0;
1165 npending = 0;
1056 1166
1057 close (respipe [0]); 1167 close (respipe [0]);
1058 close (respipe [1]); 1168 close (respipe [1]);
1059 create_pipe (); 1169 create_pipe ();
1060 1170
1087PROTOTYPES: ENABLE 1197PROTOTYPES: ENABLE
1088 1198
1089BOOT: 1199BOOT:
1090{ 1200{
1091 HV *stash = gv_stashpv ("IO::AIO", 1); 1201 HV *stash = gv_stashpv ("IO::AIO", 1);
1202
1092 newCONSTSUB (stash, "EXDEV", newSViv (EXDEV)); 1203 newCONSTSUB (stash, "EXDEV", newSViv (EXDEV));
1093 newCONSTSUB (stash, "O_RDONLY", newSViv (O_RDONLY)); 1204 newCONSTSUB (stash, "O_RDONLY", newSViv (O_RDONLY));
1094 newCONSTSUB (stash, "O_WRONLY", newSViv (O_WRONLY)); 1205 newCONSTSUB (stash, "O_WRONLY", newSViv (O_WRONLY));
1206 newCONSTSUB (stash, "O_CREAT", newSViv (O_CREAT));
1207 newCONSTSUB (stash, "O_TRUNC", newSViv (O_TRUNC));
1208 newCONSTSUB (stash, "S_IFIFO", newSViv (S_IFIFO));
1095 1209
1096 create_pipe (); 1210 create_pipe ();
1097 pthread_atfork (atfork_prepare, atfork_parent, atfork_child); 1211 pthread_atfork (atfork_prepare, atfork_parent, atfork_child);
1098} 1212}
1099 1213
1100void 1214void
1101min_parallel (nthreads) 1215min_parallel (int nthreads)
1102 int nthreads
1103 PROTOTYPE: $ 1216 PROTOTYPE: $
1104 1217
1105void 1218void
1106max_parallel (nthreads) 1219max_parallel (int nthreads)
1107 int nthreads
1108 PROTOTYPE: $ 1220 PROTOTYPE: $
1221
1222int
1223max_outstanding (int maxreqs)
1224 PROTOTYPE: $
1225 CODE:
1226 RETVAL = max_outstanding;
1227 max_outstanding = maxreqs;
1228 OUTPUT:
1229 RETVAL
1109 1230
1110void 1231void
1111aio_open (pathname,flags,mode,callback=&PL_sv_undef) 1232aio_open (pathname,flags,mode,callback=&PL_sv_undef)
1112 SV * pathname 1233 SV * pathname
1113 int flags 1234 int flags
1328 1449
1329 REQ_SEND; 1450 REQ_SEND;
1330} 1451}
1331 1452
1332void 1453void
1454aio_mknod (pathname,mode,dev,callback=&PL_sv_undef)
1455 SV * pathname
1456 SV * callback
1457 UV mode
1458 UV dev
1459 PPCODE:
1460{
1461 dREQ;
1462
1463 req->type = REQ_MKNOD;
1464 req->data = newSVsv (pathname);
1465 req->dataptr = SvPVbyte_nolen (req->data);
1466 req->mode = (mode_t)mode;
1467 req->offset = dev;
1468
1469 REQ_SEND;
1470}
1471
1472void
1333aio_busy (delay,callback=&PL_sv_undef) 1473aio_busy (delay,callback=&PL_sv_undef)
1334 double delay 1474 double delay
1335 SV * callback 1475 SV * callback
1336 PPCODE: 1476 PPCODE:
1337{ 1477{
1368 req->type = REQ_NOP; 1508 req->type = REQ_NOP;
1369 1509
1370 REQ_SEND; 1510 REQ_SEND;
1371} 1511}
1372 1512
1373void 1513int
1374aioreq_pri (int pri = DEFAULT_PRI) 1514aioreq_pri (int pri = 0)
1375 CODE: 1515 PROTOTYPE: ;$
1516 CODE:
1517 RETVAL = next_pri - PRI_BIAS;
1518 if (items > 0)
1519 {
1376 if (pri < PRI_MIN) pri = PRI_MIN; 1520 if (pri < PRI_MIN) pri = PRI_MIN;
1377 if (pri > PRI_MAX) pri = PRI_MAX; 1521 if (pri > PRI_MAX) pri = PRI_MAX;
1378 next_pri = pri + PRI_BIAS; 1522 next_pri = pri + PRI_BIAS;
1523 }
1524 OUTPUT:
1525 RETVAL
1379 1526
1380void 1527void
1381aioreq_nice (int nice = 0) 1528aioreq_nice (int nice = 0)
1382 CODE: 1529 CODE:
1383 nice = next_pri - nice; 1530 nice = next_pri - nice;
1384 if (nice < PRI_MIN) nice = PRI_MIN; 1531 if (nice < PRI_MIN) nice = PRI_MIN;
1385 if (nice > PRI_MAX) nice = PRI_MAX; 1532 if (nice > PRI_MAX) nice = PRI_MAX;
1386 next_pri = nice + PRI_BIAS; 1533 next_pri = nice + PRI_BIAS;
1387 1534
1388void 1535void
1389flush () 1536flush ()
1390 PROTOTYPE: 1537 PROTOTYPE:
1391 CODE: 1538 CODE:
1392 while (nreqs) 1539 while (nreqs)
1393 { 1540 {
1394 poll_wait (); 1541 poll_wait ();
1395 poll_cb (); 1542 poll_cb (0);
1396 } 1543 }
1397 1544
1398void 1545void
1399poll() 1546poll()
1400 PROTOTYPE: 1547 PROTOTYPE:
1401 CODE: 1548 CODE:
1402 if (nreqs) 1549 if (nreqs)
1403 { 1550 {
1404 poll_wait (); 1551 poll_wait ();
1405 poll_cb (); 1552 poll_cb (0);
1406 } 1553 }
1407 1554
1408int 1555int
1409poll_fileno() 1556poll_fileno()
1410 PROTOTYPE: 1557 PROTOTYPE:
1415 1562
1416int 1563int
1417poll_cb(...) 1564poll_cb(...)
1418 PROTOTYPE: 1565 PROTOTYPE:
1419 CODE: 1566 CODE:
1420 RETVAL = poll_cb (); 1567 RETVAL = poll_cb (0);
1568 OUTPUT:
1569 RETVAL
1570
1571int
1572poll_some(int max = 0)
1573 PROTOTYPE: $
1574 CODE:
1575 RETVAL = poll_cb (max);
1421 OUTPUT: 1576 OUTPUT:
1422 RETVAL 1577 RETVAL
1423 1578
1424void 1579void
1425poll_wait() 1580poll_wait()
1434 CODE: 1589 CODE:
1435 RETVAL = nreqs; 1590 RETVAL = nreqs;
1436 OUTPUT: 1591 OUTPUT:
1437 RETVAL 1592 RETVAL
1438 1593
1594int
1595nready()
1596 PROTOTYPE:
1597 CODE:
1598 RETVAL = get_nready ();
1599 OUTPUT:
1600 RETVAL
1601
1602int
1603npending()
1604 PROTOTYPE:
1605 CODE:
1606 RETVAL = get_npending ();
1607 OUTPUT:
1608 RETVAL
1609
1439PROTOTYPES: DISABLE 1610PROTOTYPES: DISABLE
1440 1611
1441MODULE = IO::AIO PACKAGE = IO::AIO::REQ 1612MODULE = IO::AIO PACKAGE = IO::AIO::REQ
1442 1613
1443void 1614void
1494void 1665void
1495result (aio_req grp, ...) 1666result (aio_req grp, ...)
1496 CODE: 1667 CODE:
1497{ 1668{
1498 int i; 1669 int i;
1670 AV *av;
1671
1672 grp->errorno = errno;
1673
1499 AV *av = newAV (); 1674 av = newAV ();
1500 1675
1501 for (i = 1; i < items; ++i ) 1676 for (i = 1; i < items; ++i )
1502 av_push (av, newSVsv (ST (i))); 1677 av_push (av, newSVsv (ST (i)));
1503 1678
1504 SvREFCNT_dec (grp->data); 1679 SvREFCNT_dec (grp->data);
1505 grp->data = (SV *)av; 1680 grp->data = (SV *)av;
1506} 1681}
1682
1683void
1684errno (aio_req grp, int errorno = errno)
1685 CODE:
1686 grp->errorno = errorno;
1507 1687
1508void 1688void
1509limit (aio_req grp, int limit) 1689limit (aio_req grp, int limit)
1510 CODE: 1690 CODE:
1511 grp->fd2 = limit; 1691 grp->fd2 = limit;

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines