OLD | NEW |
1 # Copyright 2013 The Chromium Authors. All rights reserved. | 1 # Copyright 2013 The Chromium Authors. All rights reserved. |
2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
4 | 4 |
5 """A utility to run functions with timeouts and retries.""" | 5 """A utility to run functions with timeouts and retries.""" |
| 6 # pylint: disable=W0702 |
6 | 7 |
7 import functools | |
8 import threading | 8 import threading |
9 | 9 |
10 import reraiser_thread | 10 from pylib.utils import reraiser_thread |
11 import watchdog_timer | 11 from pylib.utils import watchdog_timer |
12 | 12 |
13 | 13 |
14 def Run(func, timeout, retries, args=[], kwargs={}): | 14 def Run(func, timeout, retries, args=None, kwargs=None): |
15 """Runs the passed function in a separate thread with timeouts and retries. | 15 """Runs the passed function in a separate thread with timeouts and retries. |
16 | 16 |
17 Args: | 17 Args: |
18 func: the function to be wrapped. | 18 func: the function to be wrapped. |
19 timeout: the timeout in seconds for each try. | 19 timeout: the timeout in seconds for each try. |
20 retries: the number of retries. | 20 retries: the number of retries. |
21 args: list of positional args to pass to |func|. | 21 args: list of positional args to pass to |func|. |
22 kwargs: dictionary of keyword args to pass to |func|. | 22 kwargs: dictionary of keyword args to pass to |func|. |
23 | 23 |
24 Returns: | 24 Returns: |
25 The return value of func(*args, **kwargs). | 25 The return value of func(*args, **kwargs). |
26 """ | 26 """ |
| 27 if not args: |
| 28 args = [] |
| 29 if not kwargs: |
| 30 kwargs = {} |
| 31 |
27 # The return value uses a list because Python variables are references, not | 32 # The return value uses a list because Python variables are references, not |
28 # values. Closures make a copy of the reference, so updating the closure's | 33 # values. Closures make a copy of the reference, so updating the closure's |
29 # reference wouldn't update where the original reference pointed. | 34 # reference wouldn't update where the original reference pointed. |
30 ret = [None] | 35 ret = [None] |
31 def RunOnTimeoutThread(): | 36 def RunOnTimeoutThread(): |
32 ret[0] = func(*args, **kwargs) | 37 ret[0] = func(*args, **kwargs) |
33 | 38 |
34 while True: | 39 while True: |
35 try: | 40 try: |
36 name = 'TimeoutThread-for-%s' % threading.current_thread().name | 41 name = 'TimeoutThread-for-%s' % threading.current_thread().name |
37 thread_group = reraiser_thread.ReraiserThreadGroup( | 42 thread_group = reraiser_thread.ReraiserThreadGroup( |
38 [reraiser_thread.ReraiserThread(RunOnTimeoutThread, name=name)]) | 43 [reraiser_thread.ReraiserThread(RunOnTimeoutThread, name=name)]) |
39 thread_group.StartAll() | 44 thread_group.StartAll() |
40 thread_group.JoinAll(watchdog_timer.WatchdogTimer(timeout)) | 45 thread_group.JoinAll(watchdog_timer.WatchdogTimer(timeout)) |
41 return ret[0] | 46 return ret[0] |
42 except: | 47 except: |
43 if retries <= 0: | 48 if retries <= 0: |
44 raise | 49 raise |
45 retries -= 1 | 50 retries -= 1 |
OLD | NEW |