Log some times to identify very slow parts

3 files changed, 6 insertions, 26 deletions

diff --git a/service/pixelated/adapter/mailstore/leap_mailstore.py b/service/pixelated/adapter/mailstore/leap_mailstore.py
index 98758d1f..6cbbe10a 100644
--- a/service/pixelated/adapter/mailstore/leap_mailstore.py
+++ b/service/pixelated/adapter/mailstore/leap_mailstore.py
@@ -14,10 +14,7 @@
 # You should have received a copy of the GNU Affero General Public License
 # along with Pixelated. If not, see <http://www.gnu.org/licenses/>.
 import re
-import logging
-import time
 from email.header import decode_header
-from functools import wraps
 from uuid import uuid4
 
 from leap.mail.adaptors.soledad import SoledadMailAdaptor
@@ -28,12 +25,10 @@ from twisted.internet.defer import FirstError, DeferredList
 from pixelated.adapter.mailstore.body_parser import BodyParser
 from pixelated.adapter.mailstore.mailstore import MailStore, underscore_uuid
 from pixelated.adapter.model.mail import Mail, InputMail
+from pixelated.support import log_time_deferred
 from pixelated.support.functional import to_unicode
 
 
-log = logging.getLogger(__name__)
-
-
 class AttachmentInfo(object):
     def __init__(self, ident, name, encoding=None, ctype='application/octet-stream', size=0):
         self.ident = ident
@@ -193,25 +188,6 @@ def _extract_filename_from_name_header_part(header_value):
     return filename
 
 
-def log_time_deferred(f):
-
-    def log_time(result, start):
-        log.info('after callback: Needed %f ms to execute %s' % ((time.clock() - start), f))
-        return result
-
-    @wraps(f)
-    def wrapper(*args, **kwds):
-        start = time.clock()
-        result = f(*args, **kwds)
-        if isinstance(result, defer.Deferred):
-            result.addCallback(log_time, start=start)
-        else:
-            log.warn('No Deferred returned, perhaps need to re-order annotations?')
-        return result
-
-    return wrapper
-
-
 class LeapMailStore(MailStore):
     __slots__ = ('soledad')
 
diff --git a/service/pixelated/adapter/search/__init__.py b/service/pixelated/adapter/search/__init__.py
index 35087101..e137b392 100644
--- a/service/pixelated/adapter/search/__init__.py
+++ b/service/pixelated/adapter/search/__init__.py
@@ -13,7 +13,7 @@
 #
 # You should have received a copy of the GNU Affero General Public License
 # along with Pixelated. If not, see <http://www.gnu.org/licenses/>.
-
+from pixelated.support import log_time
 from pixelated.support.encrypted_file_storage import EncryptedFileStorage
 
 import os
@@ -172,6 +172,7 @@ class SearchEngine(object):
             results = searcher.search(query, **options)
         return results
 
+    @log_time
     def search(self, query, window=25, page=1, all_mails=False):
         query = self.prepare_query(query)
         return self._search_all_mails(query) if all_mails else self._paginated_search_mails(query, window, page)
diff --git a/service/pixelated/adapter/services/mail_service.py b/service/pixelated/adapter/services/mail_service.py
index bfe45cad..2da2cca6 100644
--- a/service/pixelated/adapter/services/mail_service.py
+++ b/service/pixelated/adapter/services/mail_service.py
@@ -25,6 +25,8 @@ from pixelated.adapter.model.status import Status
 from pixelated.adapter.services.tag_service import extract_reserved_tags
 from leap.mail.adaptors.soledad import SoledadMailAdaptor
 
+from pixelated.support import log_time_deferred
+
 
 class MailService(object):
 
@@ -43,6 +45,7 @@ class MailService(object):
     def save_attachment(self, content, content_type):
         return self.attchment_store.add_attachment(content, content_type)
 
+    @log_time_deferred
     @defer.inlineCallbacks
     def mails(self, query, window_size, page):
         mail_ids, total = self.search_engine.search(query, window_size, page)