U "FZh@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.cs4z ||WStk r.tdYnXdS)Nz6Unexpected exception from %s executed in logging pool!) Exception_LOGGER exception)argskwargsbehaviorm/home/aprabhat/apps/x.techxrdev.in/venv/lib/python3.8/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|f||Sr)rsubmitr )rfnrrr r r r2sz_LoggingPool.submitcOs$|jjt|f|d|ddiS)Ntimeout)rmapr get)rfunc iterablesrr r r r5s z_LoggingPool.mapTcCs|jj|ddS)Nrr)rrr r r r:sz_LoggingPool.shutdownN)T) __name__ __module__ __qualname____doc__rrrrrrr r r r r&srcCstt|S)a1Creates 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. )rrZThreadPoolExecutor) max_workersr r r pool>s r() r&Z concurrentrlogging getLoggerr#rr objectrr(r r r r s