summaryrefslogtreecommitdiff
path: root/src/leap/mail/imap
diff options
context:
space:
mode:
authorKali Kaneko <kali@leap.se>2015-09-30 22:44:57 -0400
committerKali Kaneko <kali@leap.se>2015-10-01 10:22:37 -0400
commit8fde125086c100434356c4413560871d9cbe8dfb (patch)
tree2fcad81612d64f30f0ba2430c0027fad57f57b94 /src/leap/mail/imap
parent20273e82dfb66dcf1d2dac2df217ba835a622c38 (diff)
[bug] fix slow appends
we were adding listeners for each mailbox instance, which was making appends particularly slow, since the method that gets current count and recent count is expensive and was being called way too many times.
Diffstat (limited to 'src/leap/mail/imap')
-rw-r--r--src/leap/mail/imap/mailbox.py39
-rw-r--r--src/leap/mail/imap/server.py6
2 files changed, 34 insertions, 11 deletions
diff --git a/src/leap/mail/imap/mailbox.py b/src/leap/mail/imap/mailbox.py
index c7accbb..bfc0bfc 100644
--- a/src/leap/mail/imap/mailbox.py
+++ b/src/leap/mail/imap/mailbox.py
@@ -77,6 +77,32 @@ INIT_FLAGS = (MessageFlags.SEEN_FLAG, MessageFlags.ANSWERED_FLAG,
MessageFlags.LIST_FLAG)
+def make_collection_listener(mailbox):
+ """
+ Wrap a mailbox in a class that can be hashed according to the mailbox name.
+
+ This means that dicts or sets will use this new equality rule, so we won't
+ collect multiple instances of the same mailbox in collections like the
+ MessageCollection set where we keep track of listeners.
+ """
+
+ class HashableMailbox(object):
+
+ def __init__(self, mbox):
+ self.mbox = mbox
+
+ def __hash__(self):
+ return hash(self.mbox.mbox_name)
+
+ def __eq__(self, other):
+ return self.mbox.mbox_name == other.mbox.mbox_name
+
+ def notify_new(self):
+ self.mbox.notify_new()
+
+ return HashableMailbox(mailbox)
+
+
class IMAPMailbox(object):
"""
A Soledad-backed IMAP mailbox.
@@ -118,7 +144,7 @@ class IMAPMailbox(object):
self.rw = rw
self._uidvalidity = None
self.collection = collection
- self.collection.addListener(self)
+ self.collection.addListener(make_collection_listener(self))
@property
def mbox_name(self):
@@ -383,6 +409,7 @@ class IMAPMailbox(object):
def notify_new(self, *args):
"""
Notify of new messages to all the listeners.
+
This will be called indirectly by the underlying collection, that will
notify this IMAPMailbox whenever there are changes in the number of
messages in the collection, since we have added ourselves to the
@@ -405,13 +432,15 @@ class IMAPMailbox(object):
def _get_notify_count(self):
"""
- Get message count and recent count for this mailbox
- Executed in a separate thread. Called from notify_new.
+ Get message count and recent count for this mailbox.
:return: a deferred that will fire with a tuple, with number of
messages and number of recent messages.
:rtype: Deferred
"""
+ # XXX this is way too expensive in cases like multiple APPENDS.
+ # We should have a way of keep a cache or do a self-increment for that
+ # kind of calls.
d_exists = defer.maybeDeferred(self.getMessageCount)
d_recent = defer.maybeDeferred(self.getRecentCount)
d_list = [d_exists, d_recent]
@@ -875,8 +904,8 @@ class IMAPMailbox(object):
uid when the copy succeed.
:rtype: Deferred
"""
- if PROFILE_CMD:
- do_profile_cmd(d, "COPY")
+ # if PROFILE_CMD:
+ # do_profile_cmd(d, "COPY")
# A better place for this would be the COPY/APPEND dispatcher
# in server.py, but qtreactor hangs when I do that, so this seems
diff --git a/src/leap/mail/imap/server.py b/src/leap/mail/imap/server.py
index 8f14936..99e7174 100644
--- a/src/leap/mail/imap/server.py
+++ b/src/leap/mail/imap/server.py
@@ -269,12 +269,6 @@ class LEAPIMAPServer(imap4.IMAP4Server):
select_FETCH = (do_FETCH, imap4.IMAP4Server.arg_seqset,
imap4.IMAP4Server.arg_fetchatt)
- def notifyNew(self, ignored=None):
- """
- Notify new messages to listeners.
- """
- reactor.callFromThread(self.mbox.notify_new)
-
def _cbSelectWork(self, mbox, cmdName, tag):
"""
Callback for selectWork