pep/slixmpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Update JID cache to do extra memoization and locking.

Browse source 
Passing cache_lock=True to JID() will insert the JID into the cache
and prevent it from being dropped from the cache.
This commit is contained in:
Lance Stout 2012-10-24 12:47:25 -07:00
parent 2a4e435228
commit d0666a5eb6
1 changed files with 37 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

60
sleekxmpp/jid.py
View file

@ -16,6 +16,7 @@ from __future__ import unicode_literals
import re import re
import socket import socket
import stringprep import stringprep
import threading
import encodings.idna import encodings.idna
from sleekxmpp.util import stringprep_profiles from sleekxmpp.util import stringprep_profiles
@ -65,6 +66,7 @@ JID_UNESCAPE_TRANSFORMATIONS = {'\\20': ' ',
'\\5c': '\\'} '\\5c': '\\'}
JID_CACHE = OrderedDict() JID_CACHE = OrderedDict()
JID_CACHE_LOCK = threading.Lock()
JID_CACHE_MAX_SIZE = 1024 JID_CACHE_MAX_SIZE = 1024
@ -416,37 +418,49 @@ class JID(object):
# pylint: disable=W0212 # pylint: disable=W0212
def __init__(self, jid=None, **kwargs): def __init__(self, jid=None, **kwargs):
self._jid = (None, None, None) jid_data = (jid, kwargs.get('local', None),
kwargs.get('domain', None),
kwargs.get('resource', None))
if jid is None or jid == '': locked = kwargs.get('cache_lock', False)
jid = ''
if not jid: if jid_data in JID_CACHE:
jid = (None, None, None) parsed_jid, locked = JID_CACHE[jid_data]
elif jid in JID_CACHE: self._jid = parsed_jid
jid = JID_CACHE[jid]
elif not isinstance(jid, JID):
jid = _parse_jid(jid)
else: else:
jid = jid._jid if jid is None:
jid = ''
local, domain, resource = jid if not jid:
parsed_jid = (None, None, None)
elif not isinstance(jid, JID):
parsed_jid = _parse_jid(jid)
else:
parsed_jid = jid._jid
local = kwargs.get('local', local) local, domain, resource = parsed_jid
domain = kwargs.get('domain', domain)
resource = kwargs.get('resource', resource)
if 'local' in kwargs: local = kwargs.get('local', local)
local = _escape_node(local) domain = kwargs.get('domain', domain)
if 'domain' in kwargs: resource = kwargs.get('resource', resource)
domain = _validate_domain(domain)
if 'resource' in kwargs:
resource = _validate_resource(resource)
self._jid = (local, domain, resource) if 'local' in kwargs:
JID_CACHE[_format_jid(*self._jid)] = self._jid local = _escape_node(local)
if 'domain' in kwargs:
domain = _validate_domain(domain)
if 'resource' in kwargs:
resource = _validate_resource(resource)
self._jid = (local, domain, resource)
JID_CACHE[jid_data] = (self._jid, locked)
if len(JID_CACHE) > JID_CACHE_MAX_SIZE: if len(JID_CACHE) > JID_CACHE_MAX_SIZE:
JID_CACHE.popitem(False) with JID_CACHE_LOCK:
key, item = JID_CACHE.popitem(False)
if item[1]:
# Need to reinsert locked JIDs
JID_CACHE[key] = item
def unescape(self): def unescape(self):
"""Return an unescaped JID object. """Return an unescaped JID object.