a !f@sFdZddlmZddlZeeZddZGdddeZ dd Z dS) zFA thread pool that logs exceptions raised by tasks executed within it.)futuresNcsfdd}|S)z:Wraps an arbitrary callable behavior in exception-logging.cs6z|i|WSty0tdYn0dS)Nz6Unexpected exception from %s executed in logging pool!) Exception_LOGGER exception)argskwargsbehaviorg/var/www/html/python-backend/venv/lib/python3.9/site-packages/grpc/framework/foundation/logging_pool.py _wrappings z_wrap.._wrappingr )r r r rr _wraps r c@sBeZdZdZddZddZddZdd Zd d Zdd dZ dS) _LoggingPoolzGAn exception-logging futures.ThreadPoolExecutor-compatible thread pool.cCs ||_dSN) _backing_pool)selfZ backing_poolr r r __init__(sz_LoggingPool.__init__cCs|Srr )rr r r __enter__+sz_LoggingPool.__enter__cCs|jjdddS)NTwaitrshutdown)rexc_typeexc_valexc_tbr r r __exit__.sz_LoggingPool.__exit__cOs|jjt|g|Ri|Sr)rsubmitr )rfnrrr r r r1sz_LoggingPool.submitcOs&|jjt|g|Rd|ddiS)Ntimeout)rmapr get)rfunc iterablesrr r r r4s  z_LoggingPool.mapTcCs|jj|ddS)Nrr)rrr r r r9sz_LoggingPool.shutdownN)T) __name__ __module__ __qualname____doc__rrrrrrr r r r r%srcCstt|S)a%Creates a thread pool that logs exceptions raised by the tasks within it. Args: max_workers: The maximum number of worker threads to allow the pool. Returns: A futures.ThreadPoolExecutor-compatible thread pool that logs exceptions raised by the tasks executed within it. )rrThreadPoolExecutor) max_workersr r r pool=s r)) r& concurrentrlogging getLoggerr#rr objectrr)r r r r s