How to detect exceptions in concurrent.futures in Python3?

前端 未结 2 1526
情书的邮戳
情书的邮戳 2020-12-17 17:43

I have just moved on to python3 as a result of its concurrent futures module. I was wondering if I could get it to detect errors. I want to use concurrent futures to paralle

相关标签:
2条回答
  • 2020-12-17 18:22

    Way too late to the party, but maybe it'll help someone else...

    I'm pretty sure the original question was not really answered. Folks got hung up on the fact that user5327424 was using a keyboard interrupt to raise an exception when the point was that the exception (however it was caused) was not raised. For example:

    import concurrent.futures
    
    
    def main():
        numbers = range(10)
    
        with concurrent.futures.ThreadPoolExecutor() as executor:
            results = {executor.submit(raise_my_exception, number): number for number in numbers}
    
    
    def raise_my_exception(number):
        print('Proof that this function is getting called. %s' % number)
        raise Exception('This never sees the light of day...')
    
    
    main()
    

    When the example code above is executed, you will see the text inside the print statement displayed on the screen, but you will never see the exception. This is because the results of each thread are held in the results object. You need to iterate that object to get to your exceptions. The following example shows how to access the results.

    import concurrent.futures
    
    
    def main():
        numbers = range(10)
    
        with concurrent.futures.ThreadPoolExecutor() as executor:
            results = {executor.submit(raise_my_exception, number): number for number in numbers}
    
        for result in results:
            # This will cause the exception to be raised (but only the first one)
            print(result.result())
    
    
    def raise_my_exception(number):
        print('Proof that this function is getting called. %s' % number)
        raise Exception('This will be raised once the results are iterated.')
    
    
    main()
    

    I'm not sure I like this behavior or not, but it does allow the threads to fully execute, regardless of the exceptions encountered inside the individual threads.

    0 讨论(0)
  • Here's a solution. I'm not sure you like it, but I can't think of any other. I've modified your code to make it work.

    from concurrent.futures import ThreadPoolExecutor
    import time
    
    quit = False
    
    def pri():
        print("Hello World!!!")
    
    def start():
        while quit is not True:
            time.sleep(1)
            pri()
    
    try:
        pool = ThreadPoolExecutor(max_workers=3)
        pool.submit(start)
    
        while quit is not True:
            print("hei")
            time.sleep(1)
    except KeyboardInterrupt:
        quit = True
    

    Here are the points:

    1. When you use with ThreadPoolExecutor(max_workers=3) as exe, it waits until all tasks have been done. Have a look at Doc

      If wait is True then this method will not return until all the pending futures are done executing and the resources associated with the executor have been freed. If wait is False then this method will return immediately and the resources associated with the executor will be freed when all pending futures are done executing. Regardless of the value of wait, the entire Python program will not exit until all pending futures are done executing.

      You can avoid having to call this method explicitly if you use the with statement, which will shutdown the Executor (waiting as if Executor.shutdown() were called with wait set to True)

      It's like calling join() on a thread.
      That's why I replaced it with:

      pool = ThreadPoolExecutor(max_workers=3)
      pool.submit(start)
      
    2. Main thread must be doing "work" to be able to catch a Ctrl+C. So you can't just leave main thread there and exit, the simplest way is to run an infinite loop

    3. Now that you have a loop running in main thread, when you hit CTRL+C, program will enter the except KeyboardInterrupt block and set quit=True. Then your worker thread can exit.

    Strictly speaking, this is only a workaround. It seems to me it's impossible to have another way for this.

    Edit
    I'm not sure what's bothering you, but you can catch exception in another thread without problem:

    import socket
    import time
    from concurrent.futures import ThreadPoolExecutor 
    s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
    
    def con():
        try:
            raise socket.gaierror
            main()
        except socket.gaierror:
            print("gaierror occurred")
            err()
    
    def err():
        print("err invoked")
        time.sleep(1)
        con()
    
    def main():
        s.send("[+] Hello")
    
    with ThreadPoolExecutor(3) as exe:
        exe.submit(con)
    

    Output

    gaierror occurred
    err invoked
    gaierror occurred
    err invoked
    gaierror occurred
    err invoked
    gaierror occurred
    ...
    
    0 讨论(0)
提交回复
热议问题