← Back to branch summary

~libra-core/libra/master

« back to all changes in this revision

Viewing changes to libra/openstack/common/excutils.py

  • Committer: Monty Taylor
  • Date: 2015-10-17 20:03:27 UTC
  • Revision ID: git-v1:c7082fa72ac73b23b48ce63fc82aa7da2d3e5d6a
Retire stackforge/libra

Show diffs side-by-side

added added

removed removed

Lines of Context:
libra/openstack/common/excutils.py
1
 
# vim: tabstop=4 shiftwidth=4 softtabstop=4
2
 
 
3
 
# Copyright 2011 OpenStack Foundation.
4
 
# Copyright 2012, Red Hat, Inc.
5
 
#
6
 
#    Licensed under the Apache License, Version 2.0 (the "License"); you may
7
 
#    not use this file except in compliance with the License. You may obtain
8
 
#    a copy of the License at
9
 
#
10
 
#         http://www.apache.org/licenses/LICENSE-2.0
11
 
#
12
 
#    Unless required by applicable law or agreed to in writing, software
13
 
#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
14
 
#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
15
 
#    License for the specific language governing permissions and limitations
16
 
#    under the License.
17
 
 
18
 
"""
19
 
Exception related utilities.
20
 
"""
21
 
 
22
 
import logging
23
 
import sys
24
 
import time
25
 
import traceback
26
 
 
27
 
import six
28
 
 
29
 
from libra.openstack.common.gettextutils import _  # noqa
30
 
 
31
 
 
32
 
class save_and_reraise_exception(object):
33
 
    """Save current exception, run some code and then re-raise.
34
 
 
35
 
    In some cases the exception context can be cleared, resulting in None
36
 
    being attempted to be re-raised after an exception handler is run. This
37
 
    can happen when eventlet switches greenthreads or when running an
38
 
    exception handler, code raises and catches an exception. In both
39
 
    cases the exception context will be cleared.
40
 
 
41
 
    To work around this, we save the exception state, run handler code, and
42
 
    then re-raise the original exception. If another exception occurs, the
43
 
    saved exception is logged and the new exception is re-raised.
44
 
 
45
 
    In some cases the caller may not want to re-raise the exception, and
46
 
    for those circumstances this context provides a reraise flag that
47
 
    can be used to suppress the exception.  For example:
48
 
 
49
 
    except Exception:
50
 
        with save_and_reraise_exception() as ctxt:
51
 
            decide_if_need_reraise()
52
 
            if not should_be_reraised:
53
 
                ctxt.reraise = False
54
 
    """
55
 
    def __init__(self):
56
 
        self.reraise = True
57
 
 
58
 
    def __enter__(self):
59
 
        self.type_, self.value, self.tb, = sys.exc_info()
60
 
        return self
61
 
 
62
 
    def __exit__(self, exc_type, exc_val, exc_tb):
63
 
        if exc_type is not None:
64
 
            logging.error(_('Original exception being dropped: %s'),
65
 
                          traceback.format_exception(self.type_,
66
 
                                                     self.value,
67
 
                                                     self.tb))
68
 
            return False
69
 
        if self.reraise:
70
 
            six.reraise(self.type_, self.value, self.tb)
71
 
 
72
 
 
73
 
def forever_retry_uncaught_exceptions(infunc):
74
 
    def inner_func(*args, **kwargs):
75
 
        last_log_time = 0
76
 
        last_exc_message = None
77
 
        exc_count = 0
78
 
        while True:
79
 
            try:
80
 
                return infunc(*args, **kwargs)
81
 
            except Exception as exc:
82
 
                this_exc_message = six.u(str(exc))
83
 
                if this_exc_message == last_exc_message:
84
 
                    exc_count += 1
85
 
                else:
86
 
                    exc_count = 1
87
 
                # Do not log any more frequently than once a minute unless
88
 
                # the exception message changes
89
 
                cur_time = int(time.time())
90
 
                if (cur_time - last_log_time > 60 or
91
 
                        this_exc_message != last_exc_message):
92
 
                    logging.exception(
93
 
                        _('Unexpected exception occurred %d time(s)... '
94
 
                          'retrying.') % exc_count)
95
 
                    last_log_time = cur_time
96
 
                    last_exc_message = this_exc_message
97
 
                    exc_count = 0
98
 
                # This should be a very rare event. In case it isn't, do
99
 
                # a sleep.
100
 
                time.sleep(1)
101
 
    return inner_func