Skip to content

Commit 0c9980a

Browse files
authored
Added minimal logging
1 parent 0a32064 commit 0c9980a

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

retrying.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
14-
14+
import logging
1515
import random
1616
import six
1717
import sys
@@ -83,6 +83,7 @@ def __init__(
8383
wait_jitter_max=None,
8484
before_attempts=None,
8585
after_attempts=None,
86+
logger=None,
8687
):
8788

8889
self._stop_max_attempt_number = (
@@ -110,6 +111,7 @@ def __init__(
110111
self._wait_jitter_max = 0 if wait_jitter_max is None else wait_jitter_max
111112
self._before_attempts = before_attempts
112113
self._after_attempts = after_attempts
114+
self._logger = logging.getLogger(__name__) is logger is None else logger
113115

114116
# TODO add chaining of stop behaviors
115117
# stop behavior
@@ -256,6 +258,7 @@ def call(self, fn, *args, **kwargs):
256258
if not self.should_reject(attempt):
257259
return attempt.get(self._wrap_exception)
258260

261+
self._logger.warn(attempt)
259262
if self._after_attempts:
260263
self._after_attempts(attempt_number)
261264

@@ -268,6 +271,7 @@ def call(self, fn, *args, **kwargs):
268271
raise RetryError(attempt)
269272
else:
270273
sleep = self.wait(attempt_number, delay_since_first_attempt_ms)
274+
self._logger.info(f"Retrying in {sleep / 1000.0:.2f} seconds.")
271275
if self._wait_jitter_max:
272276
jitter = random.random() * self._wait_jitter_max
273277
sleep = sleep + max(0, jitter)

0 commit comments

Comments
 (0)