U @f+&@sdddlZddlZddlmZddlmZddgZdZGdddZGdddZ Gd d d e Z dS) N)queue) greenlets GreenPool GreenPileTc@szeZdZdZdddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZdS)rz4The GreenPool class is a pool of green threads. c Csz t|}Wn>tk rJ}z dt|t|}t|W5d}~XYnX|dkrjdt|}t|||_t|_ t ||_ t |_dS)Nz-GreenPool() expect size :: int, actual: {} {}rz(GreenPool() expect size >= 0, actual: {})int ValueErrorformattypestr TypeErrorreprsizesetcoroutines_runningeventlet SemaphoresemEventno_coros_running)selfremsgrL/var/www/html/chatgpt/venv/lib/python3.8/site-packages/eventlet/greenpool.py__init__s  zGreenPool.__init__cCs$||j}|jj|7_||_dS)a Change the max number of greenthreads doing work at any given time. If resize is called when there are more than *new_size* greenthreads already working on tasks, they will be allowed to complete but no new tasks will be allowed to get launched until enough greenthreads finish their tasks to drop the overall quantity below *new_size*. Until then, the return value of free() will be negative. N)rrcounter)rnew_sizeZ size_deltarrrresizes zGreenPool.resizecCs t|jS)zd Returns the number of greenthreads that are currently executing functions in the GreenPool.)lenrrrrrrunning+szGreenPool.runningcCs|jjS)z Returns the number of greenthreads available for use. If zero or less, the next call to :meth:`spawn` or :meth:`spawn_n` will block the calling greenthread until a slot becomes available.)rrr rrrfree0szGreenPool.freecOst}|jr:||jkr:tj|}|||||S|jtj |f||}|jsft |_ |j || |j|S)a Run the *function* with its arguments in its own green thread. Returns the :class:`GreenThread ` object that is running the function, which can be used to retrieve the results. If the pool is currently at capacity, ``spawn`` will block until one of the running greenthreads completes its task and frees up a slot. This function is reentrant; *function* can call ``spawn`` on the same pool without risk of deadlocking the whole thing. )r getcurrentrlockedrZ greenthreadZ GreenThreadmainacquirespawnrraddlink _spawn_done)rfunctionargskwargscurrentgtrrrr'7s     zGreenPool.spawnc CsvdzHz|||Wn4tttjfk r0YntrBt YnXW5|dkr\dSt}||XdSN) rr#r*KeyboardInterrupt SystemExitgreenletZ GreenletExitDEBUG traceback print_exc)rfuncr,r-cororrr _spawn_n_implTszGreenPool._spawn_n_implcOslt}|jr.||jkr.||||dn:|jt|j|||d}|js\t|_ |j |dS)zCreate a greenthread to run the *function*, the same as :meth:`spawn`. The difference is that :meth:`spawn_n` returns None; the results of *function* are not retrievable. NT) rr#rr$rr9r&spawn_nrrr()rr+r,r-r.grrrr:ds  zGreenPool.spawn_ncCs,t|jkstd|r(|jdS)z>Waits until all greenthreads in the pool are finished working.zWCalling waitall() from within one of the GreenPool's greenthreads will never terminate.N)rr#rAssertionErrorr!rwaitr rrrwaitallws zGreenPool.waitallcCs<|j|dk r|j||jj|jkr8|jddSr0)rreleaserremovebalancerrsend)rr8rrrr*s   zGreenPool._spawn_donecCs|jjdkr|jj SdSdS)zz#GreenPool.starmap..)GreenMaprrr:rG)rr+iterablerFrrrstarmaps  zGreenPool.starmapcGs||t|S)aThis is the same as :func:`itertools.imap`, and has the same concurrency and memory behavior as :meth:`starmap`. It's quite convenient for, e.g., farming out jobs from a file:: def worker(line): return do_something(line) pool = GreenPool() for result in pool.imap(worker, open("filename", 'r')): print(result) )rMzip)rr+ iterablesrrrimaps zGreenPool.imapN)r)__name__ __module__ __qualname____doc__rrr!r"r'r9r:r>r*rCrGrMrPrrrrr s   c@s>eZdZdZdddZddZddZd d ZeZd d Z d S)raGreenPile is an abstraction representing a bunch of I/O-related tasks. Construct a GreenPile with an existing GreenPool object. The GreenPile will then use that pool's concurrency as it processes its jobs. There can be many GreenPiles associated with a single GreenPool. A GreenPile can also be constructed standalone, not associated with any GreenPool. To do this, construct it with an integer size parameter instead of a GreenPool. It is not advisable to iterate over a GreenPile in a different greenthread than the one which is calling spawn. The iterator will exit early in that situation. rcCs0t|tr||_n t||_t|_d|_dS)Nr) isinstancerpoolr LightQueuewaitersrrZ size_or_poolrrrrs    zGreenPile.__init__cOsT|jd7_z$|jj|f||}|j|Wn|jd8_YnXdS)znRuns *func* in its own green thread, with the result available by iterating over the GreenPile object.N)rrVr'rXput)rr7r,kwr/rrrr'szGreenPile.spawncCs|Sr0rr rrr__iter__szGreenPile.__iter__cCs|jdkrt|S)zWait for the next result, suspending the current greenthread until it is available. Raises StopIteration when there are no more results.r)r StopIteration_nextr rrrnexts zGreenPile.nextc Cs(z|jWS|jd8_XdS)NrZ)rrXgetr=r rrrr_szGreenPile._nextN)r) rQrRrSrTrr'r]r`__next__r_rrrrrs  cs0eZdZfddZddZddZeZZS)rKcs"t|tj|jjd|_dS)N)maxsize)superrrrWrVrrXrY __class__rrrs zGreenMap.__init__cCs|dddS)NcSstSr0)r^rrrrrIrJz(GreenMap.done_spawning..)r'r rrrrDszGreenMap.done_spawningcCs |}t|tr|n|SdSr0)r_rUr^)rvalrrrr`s z GreenMap.next)rQrRrSrrDr`rb __classcell__rrrerrKs rK) r5rrZeventlet.supportrr3__all__r4rrrKrrrrs  /8