1
2
3
4
5
6
7
8 import time
9 import logging
10 import threading
11 import traceback
12 import exceptions
13
14 import omero
15
16 from omero_ext.functional import wraps
17
18 perf_log = logging.getLogger("omero.perf")
19
21 """ Decorator for (optionally) printing performance statistics """
22 def handler(*args, **kwargs):
23
24
25
26 lvl = perf_log.getEffectiveLevel()
27 if lvl > logging.DEBUG:
28 return func(*args, **kwargs)
29
30 try:
31 self = args[0]
32 mod = self.__class__.__module__
33 cls = self.__class__.__name__
34 tag = "%s.%s.%s" % (mod, cls, func.func_name)
35 except:
36 tag = func.func_name
37 start = time.time()
38 try:
39 rv = func(*args, **kwargs)
40 return rv
41 finally:
42 stop = time.time()
43 diff = stop - start
44 startMillis = int(start * 1000)
45 timeMillis = int(diff * 1000)
46 perf_log.debug("start[%d] time[%d] tag[%s]", startMillis, timeMillis, tag)
47 handler = wraps(func)(handler)
48 return handler
49
50
52 """ Decorator for catching any uncaught exception and converting it to an InternalException """
53 log = logging.getLogger("omero.remote")
54 def exc_handler(*args, **kwargs):
55 try:
56 self = args[0]
57 log.info(" Meth: %s.%s", self.__class__.__name__, func.func_name)
58 start = time.time()
59 rv = func(*args, **kwargs)
60 stop = time.time()
61 log.info(" Rslt: %20.20s in %ss", rv, (stop-start))
62 return rv
63 except exceptions.Exception, e:
64 stop = time.time()
65 log.info(" Excp: %20.20s in %ss.", e, (stop-start))
66 if isinstance(e, omero.ServerError):
67 raise
68 else:
69 log.warn("%s raised a non-ServerError (%s): %s", func, type(e), e)
70 msg = traceback.format_exc()
71 raise omero.InternalException(msg, None, "Internal exception")
72 exc_handler = wraps(func)(exc_handler)
73 return exc_handler
74
76 """ Decorator for using the self._lock argument of the calling instance """
77 def with_lock(*args, **kwargs):
78 self = args[0]
79 self._lock.acquire()
80 try:
81 return func(*args, **kwargs)
82 finally:
83 self._lock.release()
84 with_lock = wraps(func)(with_lock)
85 return with_lock
86