o ˷e@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.cs.z|i|WStytdw)Nz6Unexpected exception from %s executed in logging pool!) Exception_LOGGER exception)argskwargsbehavior]/var/www/ideatree/venv/lib/python3.10/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)self backing_poolr r r __init__(s z_LoggingPool.__init__cCs|Srr )rr r r __enter__+sz_LoggingPool.__enter__cCs|jjdddS)NTwaitrshutdown)rexc_typeexc_valexc_tbr r r __exit__.z_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 r!4s  z_LoggingPool.mapTcCs|jj|ddS)Nrr)rrr r r r9rz_LoggingPool.shutdownN)T) __name__ __module__ __qualname____doc__rrrrr!rr 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