… | |
… | |
1995 | PerlIOBuf_get_cnt, |
1995 | PerlIOBuf_get_cnt, |
1996 | PerlIOBuf_set_ptrcnt, |
1996 | PerlIOBuf_set_ptrcnt, |
1997 | }; |
1997 | }; |
1998 | |
1998 | |
1999 | /*****************************************************************************/ |
1999 | /*****************************************************************************/ |
|
|
2000 | /* Coro::Semaphore & Coro::Signal */ |
|
|
2001 | |
|
|
2002 | static SV * |
|
|
2003 | coro_waitarray_new (pTHX_ int count) |
|
|
2004 | { |
|
|
2005 | /* a semaphore contains a counter IV in $sem->[0] and any waiters after that */ |
|
|
2006 | AV *av = newAV (); |
|
|
2007 | SV **ary; |
|
|
2008 | |
|
|
2009 | /* unfortunately, building manually saves memory */ |
|
|
2010 | Newx (ary, 2, SV *); |
|
|
2011 | AvALLOC (av) = ary; |
|
|
2012 | AvARRAY (av) = ary; |
|
|
2013 | AvMAX (av) = 1; |
|
|
2014 | AvFILLp (av) = 0; |
|
|
2015 | ary [0] = newSViv (count); |
|
|
2016 | |
|
|
2017 | return newRV_noinc ((SV *)av); |
|
|
2018 | } |
|
|
2019 | |
2000 | /* Coro::Semaphore */ |
2020 | /* semaphore */ |
2001 | |
2021 | |
2002 | static void |
2022 | static void |
2003 | coro_semaphore_adjust (pTHX_ AV *av, IV adjust) |
2023 | coro_semaphore_adjust (pTHX_ AV *av, IV adjust) |
2004 | { |
2024 | { |
2005 | SV *count_sv = AvARRAY (av)[0]; |
2025 | SV *count_sv = AvARRAY (av)[0]; |
… | |
… | |
2035 | /* call $sem->adjust (0) to possibly wake up some other waiters */ |
2055 | /* call $sem->adjust (0) to possibly wake up some other waiters */ |
2036 | coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0); |
2056 | coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0); |
2037 | } |
2057 | } |
2038 | |
2058 | |
2039 | static int |
2059 | static int |
2040 | slf_check_semaphore_down (pTHX_ struct CoroSLF *frame) |
2060 | slf_check_semaphore_down_or_wait (pTHX_ struct CoroSLF *frame, int acquire) |
2041 | { |
2061 | { |
2042 | AV *av = (AV *)frame->data; |
2062 | AV *av = (AV *)frame->data; |
2043 | SV *count_sv = AvARRAY (av)[0]; |
2063 | SV *count_sv = AvARRAY (av)[0]; |
2044 | |
2064 | |
2045 | /* if we are about to throw, don't actually acquire the lock, just throw */ |
2065 | /* if we are about to throw, don't actually acquire the lock, just throw */ |
2046 | if (CORO_THROW) |
2066 | if (CORO_THROW) |
2047 | return 0; |
2067 | return 0; |
2048 | else if (SvIVX (count_sv) > 0) |
2068 | else if (SvIVX (count_sv) > 0) |
2049 | { |
2069 | { |
2050 | SvSTATE_current->on_destroy = 0; |
2070 | SvSTATE_current->on_destroy = 0; |
|
|
2071 | |
|
|
2072 | if (acquire) |
2051 | SvIVX (count_sv) = SvIVX (count_sv) - 1; |
2073 | SvIVX (count_sv) = SvIVX (count_sv) - 1; |
|
|
2074 | else |
|
|
2075 | coro_semaphore_adjust (aTHX_ av, 0); |
|
|
2076 | |
2052 | return 0; |
2077 | return 0; |
2053 | } |
2078 | } |
2054 | else |
2079 | else |
2055 | { |
2080 | { |
2056 | int i; |
2081 | int i; |
… | |
… | |
2065 | av_push (av, SvREFCNT_inc (SvRV (coro_current))); |
2090 | av_push (av, SvREFCNT_inc (SvRV (coro_current))); |
2066 | return 1; |
2091 | return 1; |
2067 | } |
2092 | } |
2068 | } |
2093 | } |
2069 | |
2094 | |
2070 | static void |
2095 | static int |
|
|
2096 | slf_check_semaphore_down (pTHX_ struct CoroSLF *frame) |
|
|
2097 | { |
|
|
2098 | return slf_check_semaphore_down_or_wait (aTHX_ frame, 1); |
|
|
2099 | } |
|
|
2100 | |
|
|
2101 | static int |
|
|
2102 | slf_check_semaphore_wait (pTHX_ struct CoroSLF *frame) |
|
|
2103 | { |
|
|
2104 | return slf_check_semaphore_down_or_wait (aTHX_ frame, 0); |
|
|
2105 | } |
|
|
2106 | |
|
|
2107 | static void |
2071 | slf_init_semaphore_down (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items) |
2108 | slf_init_semaphore_down_or_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items) |
2072 | { |
2109 | { |
2073 | AV *av = (AV *)SvRV (arg [0]); |
2110 | AV *av = (AV *)SvRV (arg [0]); |
2074 | |
2111 | |
2075 | if (SvIVX (AvARRAY (av)[0]) > 0) |
2112 | if (SvIVX (AvARRAY (av)[0]) > 0) |
2076 | { |
2113 | { |
… | |
… | |
2086 | |
2123 | |
2087 | /* to avoid race conditions when a woken-up coro gets terminated */ |
2124 | /* to avoid race conditions when a woken-up coro gets terminated */ |
2088 | /* we arrange for a temporary on_destroy that calls adjust (0) */ |
2125 | /* we arrange for a temporary on_destroy that calls adjust (0) */ |
2089 | SvSTATE_current->on_destroy = coro_semaphore_on_destroy; |
2126 | SvSTATE_current->on_destroy = coro_semaphore_on_destroy; |
2090 | } |
2127 | } |
|
|
2128 | } |
2091 | |
2129 | |
|
|
2130 | static void |
|
|
2131 | slf_init_semaphore_down (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items) |
|
|
2132 | { |
|
|
2133 | slf_init_semaphore_down_or_wait (aTHX_ frame, cv, arg, items); |
2092 | frame->check = slf_check_semaphore_down; |
2134 | frame->check = slf_check_semaphore_down; |
|
|
2135 | } |
2093 | |
2136 | |
|
|
2137 | static void |
|
|
2138 | slf_init_semaphore_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items) |
|
|
2139 | { |
|
|
2140 | slf_init_semaphore_down_or_wait (aTHX_ frame, cv, arg, items); |
|
|
2141 | frame->check = slf_check_semaphore_wait; |
|
|
2142 | } |
|
|
2143 | |
|
|
2144 | /* signal */ |
|
|
2145 | |
|
|
2146 | static void |
|
|
2147 | coro_signal_wake (pTHX_ AV *av, int count) |
|
|
2148 | { |
|
|
2149 | SvIVX (AvARRAY (av)[0]) = 0; |
|
|
2150 | |
|
|
2151 | /* now signal count waiters */ |
|
|
2152 | while (count > 0 && AvFILLp (av) > 0) |
|
|
2153 | { |
|
|
2154 | SV *cb; |
|
|
2155 | |
|
|
2156 | /* swap first two elements so we can shift a waiter */ |
|
|
2157 | cb = AvARRAY (av)[0]; |
|
|
2158 | AvARRAY (av)[0] = AvARRAY (av)[1]; |
|
|
2159 | AvARRAY (av)[1] = cb; |
|
|
2160 | |
|
|
2161 | cb = av_shift (av); |
|
|
2162 | |
|
|
2163 | api_ready (cb); |
|
|
2164 | sv_setiv (cb, 0); /* signal waiter */ |
|
|
2165 | SvREFCNT_dec (cb); |
|
|
2166 | |
|
|
2167 | --count; |
|
|
2168 | } |
|
|
2169 | } |
|
|
2170 | |
|
|
2171 | static int |
|
|
2172 | slf_check_signal_wait (pTHX_ struct CoroSLF *frame) |
|
|
2173 | { |
|
|
2174 | /* if we are about to throw, also stop waiting */ |
|
|
2175 | return SvROK ((SV *)frame->data) && !CORO_THROW; |
|
|
2176 | } |
|
|
2177 | |
|
|
2178 | static void |
|
|
2179 | slf_init_signal_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items) |
|
|
2180 | { |
|
|
2181 | AV *av = (AV *)SvRV (arg [0]); |
|
|
2182 | |
|
|
2183 | if (SvIVX (AvARRAY (av)[0])) |
|
|
2184 | { |
|
|
2185 | SvIVX (AvARRAY (av)[0]) = 0; |
|
|
2186 | frame->prepare = prepare_nop; |
|
|
2187 | frame->check = slf_check_nop; |
|
|
2188 | } |
|
|
2189 | else |
|
|
2190 | { |
|
|
2191 | SV *waiter = newRV_inc (SvRV (coro_current)); /* owned by signal av */ |
|
|
2192 | |
|
|
2193 | av_push (av, waiter); |
|
|
2194 | |
|
|
2195 | frame->data = (void *)sv_2mortal (SvREFCNT_inc_NN (waiter)); /* owned by process */ |
|
|
2196 | frame->prepare = prepare_schedule; |
|
|
2197 | frame->check = slf_check_signal_wait; |
|
|
2198 | } |
2094 | } |
2199 | } |
2095 | |
2200 | |
2096 | /*****************************************************************************/ |
2201 | /*****************************************************************************/ |
2097 | /* gensub: simple closure generation utility */ |
2202 | /* gensub: simple closure generation utility */ |
2098 | |
2203 | |
… | |
… | |
2772 | |
2877 | |
2773 | |
2878 | |
2774 | MODULE = Coro::State PACKAGE = Coro::Semaphore |
2879 | MODULE = Coro::State PACKAGE = Coro::Semaphore |
2775 | |
2880 | |
2776 | SV * |
2881 | SV * |
2777 | new (SV *klass, SV *count_ = 0) |
2882 | new (SV *klass, SV *count = 0) |
2778 | CODE: |
2883 | CODE: |
2779 | { |
2884 | RETVAL = sv_bless ( |
2780 | /* a semaphore contains a counter IV in $sem->[0] and any waiters after that */ |
2885 | coro_waitarray_new (aTHX_ count && SvOK (count) ? SvIV (count) : 1), |
2781 | AV *av = newAV (); |
2886 | GvSTASH (CvGV (cv)) |
2782 | SV **ary; |
2887 | ); |
|
|
2888 | OUTPUT: |
|
|
2889 | RETVAL |
2783 | |
2890 | |
2784 | /* unfortunately, building manually saves memory */ |
2891 | # helper for Coro::Channel |
2785 | Newx (ary, 2, SV *); |
2892 | SV * |
2786 | AvALLOC (av) = ary; |
2893 | _alloc (int count) |
2787 | AvARRAY (av) = ary; |
2894 | CODE: |
2788 | AvMAX (av) = 1; |
2895 | RETVAL = coro_waitarray_new (aTHX_ count); |
2789 | AvFILLp (av) = 0; |
|
|
2790 | ary [0] = newSViv (count_ && SvOK (count_) ? SvIV (count_) : 1); |
|
|
2791 | |
|
|
2792 | RETVAL = sv_bless (newRV_noinc ((SV *)av), GvSTASH (CvGV (cv))); |
|
|
2793 | } |
|
|
2794 | OUTPUT: |
2896 | OUTPUT: |
2795 | RETVAL |
2897 | RETVAL |
2796 | |
2898 | |
2797 | SV * |
2899 | SV * |
2798 | count (SV *self) |
2900 | count (SV *self) |
… | |
… | |
2810 | |
2912 | |
2811 | void |
2913 | void |
2812 | down (SV *self) |
2914 | down (SV *self) |
2813 | CODE: |
2915 | CODE: |
2814 | CORO_EXECUTE_SLF_XS (slf_init_semaphore_down); |
2916 | CORO_EXECUTE_SLF_XS (slf_init_semaphore_down); |
|
|
2917 | |
|
|
2918 | void |
|
|
2919 | wait (SV *self) |
|
|
2920 | CODE: |
|
|
2921 | CORO_EXECUTE_SLF_XS (slf_init_semaphore_wait); |
2815 | |
2922 | |
2816 | void |
2923 | void |
2817 | try (SV *self) |
2924 | try (SV *self) |
2818 | PPCODE: |
2925 | PPCODE: |
2819 | { |
2926 | { |
… | |
… | |
2831 | XSRETURN_NO; |
2938 | XSRETURN_NO; |
2832 | } |
2939 | } |
2833 | |
2940 | |
2834 | void |
2941 | void |
2835 | waiters (SV *self) |
2942 | waiters (SV *self) |
2836 | CODE: |
2943 | PPCODE: |
2837 | { |
2944 | { |
2838 | AV *av = (AV *)SvRV (self); |
2945 | AV *av = (AV *)SvRV (self); |
|
|
2946 | int wcount = AvFILLp (av) + 1 - 1; |
2839 | |
2947 | |
2840 | if (GIMME_V == G_SCALAR) |
2948 | if (GIMME_V == G_SCALAR) |
2841 | XPUSHs (sv_2mortal (newSVsv (AvARRAY (av)[0]))); |
2949 | XPUSHs (sv_2mortal (newSViv (wcount))); |
2842 | else |
2950 | else |
2843 | { |
2951 | { |
2844 | int i; |
2952 | int i; |
2845 | EXTEND (SP, AvFILLp (av) + 1 - 1); |
2953 | EXTEND (SP, wcount); |
2846 | for (i = 1; i <= AvFILLp (av); ++i) |
2954 | for (i = 1; i <= wcount; ++i) |
2847 | PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i]))); |
2955 | PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i]))); |
2848 | } |
2956 | } |
2849 | } |
2957 | } |
|
|
2958 | |
|
|
2959 | MODULE = Coro::State PACKAGE = Coro::Signal |
|
|
2960 | |
|
|
2961 | SV * |
|
|
2962 | new (SV *klass) |
|
|
2963 | CODE: |
|
|
2964 | RETVAL = sv_bless ( |
|
|
2965 | coro_waitarray_new (aTHX_ 0), |
|
|
2966 | GvSTASH (CvGV (cv)) |
|
|
2967 | ); |
|
|
2968 | OUTPUT: |
|
|
2969 | RETVAL |
|
|
2970 | |
|
|
2971 | void |
|
|
2972 | wait (SV *self) |
|
|
2973 | CODE: |
|
|
2974 | CORO_EXECUTE_SLF_XS (slf_init_signal_wait); |
|
|
2975 | |
|
|
2976 | void |
|
|
2977 | broadcast (SV *self) |
|
|
2978 | CODE: |
|
|
2979 | { |
|
|
2980 | AV *av = (AV *)SvRV (self); |
|
|
2981 | coro_signal_wake (aTHX_ av, AvFILLp (av)); |
|
|
2982 | } |
|
|
2983 | |
|
|
2984 | void |
|
|
2985 | send (SV *self) |
|
|
2986 | CODE: |
|
|
2987 | { |
|
|
2988 | AV *av = (AV *)SvRV (self); |
|
|
2989 | |
|
|
2990 | if (AvFILLp (av)) |
|
|
2991 | coro_signal_wake (av, 1); |
|
|
2992 | else |
|
|
2993 | SvIVX (AvARRAY (av)[0]) = 1; /* remember the signal */ |
|
|
2994 | } |
|
|
2995 | |
|
|
2996 | IV |
|
|
2997 | awaited (SV *self) |
|
|
2998 | CODE: |
|
|
2999 | RETVAL = AvFILLp ((AV *)SvRV (self)) + 1 - 1; |
|
|
3000 | OUTPUT: |
|
|
3001 | RETVAL |
2850 | |
3002 | |
2851 | |
3003 | |
2852 | MODULE = Coro::State PACKAGE = Coro::AnyEvent |
3004 | MODULE = Coro::State PACKAGE = Coro::AnyEvent |
2853 | |
3005 | |
2854 | BOOT: |
3006 | BOOT: |