使用python设置超时

Python code

   
   

    
    
@TraceFunc

    
    def
    
     WorkInTime(Func, Timeout 
    
    =
    
     0, 
    
    *
    
    args): 
    
    
    '''
    
    
    If function cann't finished in time then timeout
    
    
    '''
    
    

    
    
    def
    
     handler(signum, frame):    
        
    
    raise
    
     AssertionError

    
    
    try
    
    :
        signal.signal(signal.SIGALRM, handler)
        signal.alarm(Timeout)
        ret 
    
    =
    
     Func(
    
    *
    
    args)
        signal.alarm(0)
        
    
    return
    
     ret
    
    
    except
    
     AssertionError:
        logger.error(
    
    "
    
    %s conn't finished in %d seconds, timeout!
    
    "
    
     
    
    %
    
     (Func.
    
    __name__
    
    , Timeout))
        
    
    return
    
     
    
    -
    
    1