o }Mf@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)argskwargsbehaviorh/var/www/html/analyze/labelStudio/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 r2sz_LoggingPool.submitcOs&|jjt|g|Rd|ddiS)Ntimeout)rmapr get)rfunc iterablesrr r r r!5s z_LoggingPool.mapTcCs|jj|ddS)Nrr)rrr r r r:rz_LoggingPool.shutdownN)T) __name__ __module__ __qualname____doc__rrrrr!rr 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. )rrThreadPoolExecutor) max_workersr r r pool>s r+) r( concurrentrlogging getLoggerr%rr objectrr+r r r r s